mock_provider.go

package zigbee

import (
	"context"
	"github.com/stretchr/testify/mock"
)

var _ Provider = (*MockProvider)(nil)

type MockProvider struct {
	mock.Mock
}

func (m *MockProvider) RequestNodeLeave(ctx context.Context, networkAddress IEEEAddress) error {
	args := m.Called(ctx, networkAddress)
	return args.Error(0)
}

func (m *MockProvider) ForceNodeLeave(ctx context.Context, networkAddress IEEEAddress) error {
	args := m.Called(ctx, networkAddress)
	return args.Error(0)
}

func (m *MockProvider) PermitJoin(ctx context.Context, allRouters bool) error {
	args := m.Called(ctx, allRouters)
	return args.Error(0)
}

func (m *MockProvider) DenyJoin(ctx context.Context) error {
	args := m.Called(ctx)
	return args.Error(0)
}

func (m *MockProvider) AdapterNode() Node {
	args := m.Called()
	return args.Get(0).(Node)
}

func (m *MockProvider) QueryNodeDescription(ctx context.Context, networkAddress IEEEAddress) (NodeDescription, error) {
	args := m.Called(ctx, networkAddress)
	return args.Get(0).(NodeDescription), args.Error(1)
}

func (m *MockProvider) QueryNodeEndpoints(ctx context.Context, networkAddress IEEEAddress) ([]Endpoint, error) {
	args := m.Called(ctx, networkAddress)
	return args.Get(0).([]Endpoint), args.Error(1)
}

func (m *MockProvider) QueryNodeEndpointDescription(ctx context.Context, networkAddress IEEEAddress, endpoint Endpoint) (EndpointDescription, error) {
	args := m.Called(ctx, networkAddress, endpoint)
	return args.Get(0).(EndpointDescription), args.Error(1)
}

func (m *MockProvider) BindNodeToController(ctx context.Context, nodeAddress IEEEAddress, sourceEndpoint Endpoint, destinationEndpoint Endpoint, cluster ClusterID) error {
	args := m.Called(ctx, nodeAddress, sourceEndpoint, destinationEndpoint, cluster)
	return args.Error(0)
}

func (m *MockProvider) UnbindNodeFromController(ctx context.Context, nodeAddress IEEEAddress, sourceEndpoint Endpoint, destinationEndpoint Endpoint, cluster ClusterID) error {
	args := m.Called(ctx, nodeAddress, sourceEndpoint, destinationEndpoint, cluster)
	return args.Error(0)
}

func (m *MockProvider) SendApplicationMessageToNode(ctx context.Context, destinationAddress IEEEAddress, message ApplicationMessage, requireAck bool) error {
	args := m.Called(ctx, destinationAddress, message, requireAck)
	return args.Error(0)
}

func (m *MockProvider) ReadEvent(ctx context.Context) (interface{}, error) {
	args := m.Called(ctx)
	return args.Get(0), args.Error(1)
}

func (m *MockProvider) RegisterAdapterEndpoint(ctx context.Context, endpoint Endpoint, appProfileId ProfileID, appDeviceId uint16, appDeviceVersion uint8, inClusters []ClusterID, outClusters []ClusterID) error {
	args := m.Called(ctx, endpoint, appProfileId, appDeviceId, appDeviceVersion, inClusters, outClusters)
	return args.Error(0)
}