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 kong
import (
"sort"
"testing"
"github.com/kong/go-kong/kong/custom"
"github.com/stretchr/testify/assert"
)
func TestCustomEntityService(T *testing.T) {
T.Skip()
assert := assert.New(T)
client, err := NewTestClient(nil, nil)
assert.Nil(err)
assert.NotNil(client)
// fixture consumer
consumer, err := client.Consumers.Create(defaultCtx,
&Consumer{Username: String("foo")})
assert.Nil(err)
assert.NotNil(consumer)
// create a key associated with the consumer
k1 := custom.NewEntityObject("key-auth")
k1.AddRelation("consumer_id", *consumer.ID)
e1, err := client.CustomEntities.Create(defaultCtx, k1)
assert.NotNil(e1)
assert.Nil(err)
// look up the key
se := custom.NewEntityObject("key-auth")
se.AddRelation("consumer_id", *consumer.ID)
se.SetObject(map[string]interface{}{"id": e1.Object()["id"]})
gotE, err := client.CustomEntities.Get(defaultCtx, se)
assert.NotNil(gotE)
assert.Equal(e1.Object()["key"], gotE.Object()["key"])
assert.Nil(err)
gotE.Object()["key"] = "my-secret"
e1, err = client.CustomEntities.Update(defaultCtx, gotE)
assert.NotNil(e1)
assert.Nil(err)
assert.Equal("my-secret", e1.Object()["key"])
// PUT request
k2 := custom.NewEntityObject("key-auth")
id := "fc3898d9-4b4d-4491-a834-8358646e2d20"
k2.SetObject(map[string]interface{}{
"id": id,
"key": "super-secret",
})
k2.AddRelation("consumer_id", *consumer.ID)
e2, err := client.CustomEntities.Create(defaultCtx, k2)
assert.NotNil(e2)
assert.Nil(err)
assert.Equal("super-secret", e2.Object()["key"])
assert.Equal(id, e2.Object()["id"])
se = custom.NewEntityObject("key-auth")
se.AddRelation("consumer_id", *consumer.ID)
keyAuths, _, err := client.CustomEntities.List(defaultCtx, nil, se)
assert.Nil(err)
assert.Equal(2, len(keyAuths))
// list endpoint
keyAuths, err = client.CustomEntities.ListAll(defaultCtx, se)
assert.Nil(err)
assert.Equal(2, len(keyAuths))
expectedKeys := []string{e1.Object()["key"].(string),
e2.Object()["key"].(string)}
actualKeys := []string{keyAuths[0].Object()["key"].(string),
keyAuths[1].Object()["key"].(string)}
sort.Strings(expectedKeys)
sort.Strings(actualKeys)
assert.Equal(expectedKeys, actualKeys)
assert.Nil(client.CustomEntities.Delete(defaultCtx, e1))
assert.Nil(client.CustomEntities.Delete(defaultCtx, e2))
// delete fixture consumer
assert.Nil(client.Consumers.Delete(defaultCtx, consumer.ID))
}
|
