/
helper.go
48 lines (40 loc) · 1.29 KB
/
helper.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
/* _ _
*__ _____ __ ___ ___ __ _| |_ ___
*\ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
* \ V V / __/ (_| |\ V /| | (_| | || __/
* \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
*
* Copyright © 2016 - 2019 Weaviate. All rights reserved.
* LICENSE: https://github.com/creativesoftwarefdn/weaviate/blob/develop/LICENSE.md
* DESIGN & CONCEPT: Bob van Luijt (@bobvanluijt)
* CONTACT: hello@creativesoftwarefdn.org
*/
package test
import (
"testing"
"github.com/creativesoftwarefdn/weaviate/test/acceptance/helper"
)
// Helper function to get all the names of Thing classes.
func GetThingClassNames(t *testing.T) []string {
resp, err := helper.Client(t).Schema.WeaviateSchemaDump(nil)
var names []string
// Extract all names
helper.AssertRequestOk(t, resp, err, func() {
for _, class := range resp.Payload.Things.Classes {
names = append(names, class.Class)
}
})
return names
}
// Helper function to get all the names of Action classes.
func GetActionClassNames(t *testing.T) []string {
resp, err := helper.Client(t).Schema.WeaviateSchemaDump(nil)
var names []string
// Extract all names
helper.AssertRequestOk(t, resp, err, func() {
for _, class := range resp.Payload.Actions.Classes {
names = append(names, class.Class)
}
})
return names
}