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  }