-
Notifications
You must be signed in to change notification settings - Fork 1
/
registry_test.go
80 lines (59 loc) · 2.39 KB
/
registry_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package zcl
import (
"github.com/shimmeringbee/zigbee"
"github.com/stretchr/testify/assert"
"reflect"
"testing"
)
func Test_CommandRegistryGlobal(t *testing.T) {
t.Run("getting a global command that does not exist results in an error", func(t *testing.T) {
type ThisCommand struct{}
expectedIdentifier := CommandIdentifier(1)
cr := NewCommandRegistry()
_, err := cr.GetGlobalCommandIdentifier(&ThisCommand{})
assert.Error(t, err)
_, err = cr.GetGlobalCommand(expectedIdentifier)
assert.Error(t, err)
})
t.Run("registering a global command can be retrieved", func(t *testing.T) {
type ThisCommand struct{}
expectedIdentifier := CommandIdentifier(1)
expectedType := reflect.TypeOf(&ThisCommand{})
cr := NewCommandRegistry()
cr.RegisterGlobal(expectedIdentifier, &ThisCommand{})
actualIdentifier, err := cr.GetGlobalCommandIdentifier(&ThisCommand{})
assert.NoError(t, err)
assert.Equal(t, expectedIdentifier, actualIdentifier)
cmd, err := cr.GetGlobalCommand(expectedIdentifier)
actualType := reflect.TypeOf(cmd)
assert.NoError(t, err)
assert.Equal(t, expectedType, actualType)
})
t.Run("getting a local command that does not exist results in an error", func(t *testing.T) {
type ThisCommand struct{}
expectedIdentifier := CommandIdentifier(1)
clusterId := zigbee.ClusterID(0x1020)
manufacturer := zigbee.ManufacturerCode(0x3040)
cr := NewCommandRegistry()
_, err := cr.GetLocalCommandIdentifier(clusterId, manufacturer, ClientToServer, &ThisCommand{})
assert.Error(t, err)
_, err = cr.GetGlobalCommand(expectedIdentifier)
assert.Error(t, err)
})
t.Run("registering a local command can be retrieved", func(t *testing.T) {
type ThisCommand struct{}
expectedIdentifier := CommandIdentifier(1)
expectedType := reflect.TypeOf(&ThisCommand{})
clusterId := zigbee.ClusterID(0x1020)
manufacturer := zigbee.ManufacturerCode(0x3040)
cr := NewCommandRegistry()
cr.RegisterLocal(clusterId, manufacturer, ClientToServer, expectedIdentifier, &ThisCommand{})
actualIdentifier, err := cr.GetLocalCommandIdentifier(clusterId, manufacturer, ClientToServer, &ThisCommand{})
assert.NoError(t, err)
assert.Equal(t, expectedIdentifier, actualIdentifier)
cmd, err := cr.GetLocalCommand(clusterId, manufacturer, ClientToServer, expectedIdentifier)
actualType := reflect.TypeOf(cmd)
assert.NoError(t, err)
assert.Equal(t, expectedType, actualType)
})
}