krakend-http-cache/config_test.go at master · onliner/krakend-http-cache · GitHub

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
85
86
87
88
89
90
91
92
package main

import (
	"encoding/json"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewClientConfig(t *testing.T) {
	raw := []byte(`{
		"onliner/krakend-http-cache": {
			"ttl": 1,
			"connection": "memory"
		}
	}`)
	var input map[string]interface{}
	err := json.Unmarshal(raw, &input)
	assert.Nil(t, err)

	cnf, err := NewClientConfig(input)

	assert.Nil(t, err)
	assert.EqualValues(t, 1, cnf.Ttl)
	assert.Equal(t, "memory", cnf.Conn)
	assert.Nil(t, cnf.Headers)
}

func TestClientConfigHeadersNormalize(t *testing.T) {
	raw := []byte(`{
		"onliner/krakend-http-cache": {
			"ttl": 1,
			"connection": "memory",
			"headers": ["x-custom-header", "Content-type", "X-custom-header"]
		}
	}`)
	var input map[string]interface{}
	err := json.Unmarshal(raw, &input)
	assert.Nil(t, err)

	cnf, err := NewClientConfig(input)

	assert.Nil(t, err)
	assert.EqualValues(t, 1, cnf.Ttl)
	assert.Equal(t, "memory", cnf.Conn)
	assert.Equal(t, []string{"Content-Type", "X-Custom-Header"}, cnf.Headers)
}

func TestNewSrvConfig(t *testing.T) {
	raw := []byte(`{
		"onliner/krakend-http-cache": {
			"connections": {
				"inmemory": {
					"driver": "memory"
				},
				"redis": {
					"driver": "redis",
					"options": {
						"addr": "127.0.0.1:6379",
						"pass": "123qwe",
						"db": 1
					}
				}
			}
		}
	}`)

	var input map[string]interface{}
	err := json.Unmarshal(raw, &input)
	assert.Nil(t, err)

	cnf, err := NewSrvConfig(input)

	assert.Nil(t, err)

	expected := map[string]CacheCnf{
		"inmemory": {
			Driver("memory"),
			nil,
		},
		"redis": {
			Driver("redis"),
			map[string]interface{}{
				"addr": "127.0.0.1:6379",
				"pass": "123qwe",
				"db":   float64(1),
			},
		},
	}

	assert.Equal(t, expected, cnf.Conns)
}