-
Notifications
You must be signed in to change notification settings - Fork 0
/
lib_test.go
84 lines (74 loc) · 2.32 KB
/
lib_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
81
82
83
84
package duface
import (
"github.com/guestin/mob/merrors"
"testing"
)
func Test_LibraryImpl_Create(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
err = libs.Create()
merrors.AssertError(err, "create library failed")
}
func Test_LibraryImpl_RegisterFace(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
imgData := readImage("me.jpeg")
r, err := libs.RegisterFace("su", imgData,
&RegExtParams{
ActionType: REPLACE,
})
merrors.AssertError(err, "register facefailed")
dump(t, r)
}
func Test_LibraryImpl_ListUsers(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
ulist, err := libs.ListUsers(0, 100)
merrors.AssertError(err, "delete user failed")
dump(t, ulist)
}
func Test_LibraryImpl_ListUserFaces(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
ll, err := libs.ListUserFaces("su")
merrors.AssertError(err, "list user face failed")
dump(t, ll)
}
func Test_LibraryImpl_Search(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
imgData := readImage("me.jpeg")
r, err := libs.Search(imgData, &SearchExtParams{
SearchExtGeneric: SearchExtGeneric{
GroupIdList: GroupIdList{"admin"},
},
})
merrors.AssertError(err, "search failed")
dump(t, r)
}
func Test_LibraryImpl_MultiSearch(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
imgData := readImage("me.jpeg")
r, err := libs.MultiSearch(imgData, nil)
merrors.AssertError(err, "search failed")
dump(t, r)
}
func Test_LibraryImpl_DeleteFace(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
err = libs.DeleteFace("su", "0820eded50c3f938b791bc11b9df61f7")
merrors.AssertError(err, "delete user failed")
}
func Test_LibraryImpl_DeleteUser(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
err = libs.DeleteUser("su")
merrors.AssertError(err, "delete user failed")
}
func Test_LibraryImpl_Drop(t *testing.T) {
libs, err := _testCli.NewLibrary("test")
merrors.AssertError(err, "new lib failed")
err = libs.Drop()
merrors.AssertError(err, "drop library failed")
}