github.com/xmidt-org/webpa-common@v1.11.9/device/devicegate/mocks_test.go (about) 1 package devicegate 2 3 import ( 4 "github.com/stretchr/testify/mock" 5 "github.com/xmidt-org/webpa-common/device" 6 ) 7 8 type mockDeviceGate struct { 9 mock.Mock 10 } 11 12 func (m *mockDeviceGate) VisitAll(visit func(string, Set) bool) int { 13 args := m.Called(visit) 14 return args.Int(0) 15 } 16 17 func (m *mockDeviceGate) GetFilter(key string) (Set, bool) { 18 args := m.Called(key) 19 set, _ := args.Get(0).(Set) 20 return set, args.Bool(1) 21 } 22 23 func (m *mockDeviceGate) SetFilter(key string, values []interface{}) (Set, bool) { 24 args := m.Called(key, values) 25 set, _ := args.Get(0).(Set) 26 return set, args.Bool(1) 27 } 28 29 func (m *mockDeviceGate) DeleteFilter(key string) bool { 30 args := m.Called(key) 31 return args.Bool(0) 32 } 33 34 func (m *mockDeviceGate) GetAllowedFilters() (Set, bool) { 35 args := m.Called() 36 set, _ := args.Get(0).(Set) 37 return set, args.Bool(1) 38 } 39 40 func (m *mockDeviceGate) AllowConnection(d device.Interface) (bool, device.MatchResult) { 41 args := m.Called(d) 42 result, _ := args.Get(1).(device.MatchResult) 43 return args.Bool(0), result 44 } 45 46 func (m *mockDeviceGate) MarshalJSON() ([]byte, error) { 47 args := m.Called() 48 json, _ := args.Get(0).([]byte) 49 return json, args.Error(1) 50 }