ekuiper-new/deploy/docker/conf_util_test.go

// Copyright 2021 EMQ Technologies Co., Ltd.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"fmt"
	"reflect"
	"testing"
)

func TestHandle(t *testing.T) {
	var tests = []struct {
		config map[string]interface{}
		skeys  []string
		val    string
		exp    map[string]interface{}
	}{
		{
			config: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
			skeys: []string{"default", "protocol"},
			val:   "ssl",
			exp: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "ssl",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
		},

		{
			config: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
			skeys: []string{"default", "optional", "CLIENTID"},
			val:   "client2",
			exp: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client2",
					},
				},
			},
		},

		{
			config: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
			skeys: []string{"default", "optional", "KEEPALIVE"},
			val:   "6000",
			exp: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId":  "client1",
						"KeepAlive": int64(6000),
					},
				},
			},
		},

		{
			config: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
			skeys: []string{"default", "optional", "RETAINED"},
			val:   "true",
			exp: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
						"Retained": true,
					},
				},
			},
		},

		{
			config: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
			skeys: []string{"default", "optional", "test"},
			val:   "3.14",
			exp: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
						"test":     3.14,
					},
				},
			},
		},

		{
			config: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
			},
			skeys: []string{"application_conf", "test"},
			val:   "ssl",
			exp: map[string]interface{}{
				"default": map[string]interface{}{
					"protocol": "tcp",
					"port":     5563,
					"optional": map[string]interface{}{
						"ClientId": "client1",
					},
				},
				"application_conf": map[string]interface{}{
					"test": "ssl",
				},
			},
		},
	}

	for i, tt := range tests {
		Handle("edgex", tt.config, tt.skeys, tt.val)
		if !reflect.DeepEqual(tt.exp, tt.config) {
			t.Errorf("%d \tresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.exp, tt.config)
		}
	}
}

func TestProcessEnv(t *testing.T) {
	fileMap["edgex"] = "test/edgex.yaml"
	var tests = []struct {
		vars []string
		file string
		expt map[string]interface{}
		out  string
	}{
		{
			vars: []string{
				"EDGEX__DEFAULT__TYPE=zmq",
				"EDGEX__DEFAULT__MESSAGETYPE=event",
				"EDGEX__DEFAULT__OPTIONAL__CLIENTID=clientid_0000",
				"EDGEX__DEFAULT__OPTIONAL__PASSWORD=should_not_print",
				"EDGEX__APPLICATION_CONF__PROTOCOL=ssl",
			},
			file: "edgex",
			expt: map[string]interface{}{
				"default": map[string]interface{}{
					"messageType": "event",
					"protocol":    "tcp",
					"type":        "zmq",
					"optional": map[string]interface{}{
						"ClientId": "clientid_0000",
						"Password": "should_not_print",
					},
				},
				"application_conf": map[string]interface{}{
					"protocol": "ssl",
				},
			},
			out: "application_conf:\n    protocol: ssl\ndefault:\n    messageType: event\n    optional:\n        ClientId: clientid_0000\n        Password: '*'\n    protocol: tcp\n    type: zmq\n",
		},
	}
	files := make(map[string]map[string]interface{})
	for i, tt := range tests {
		ProcessEnv(files, tt.vars)
		if !reflect.DeepEqual(tt.expt, files[tt.file]) {
			t.Errorf("%d \tresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.expt, files[tt.file])
		}
		for f, v := range files {
			p := toPrintableString(v)
			if !reflect.DeepEqual(tt.out, p) {
				t.Errorf("%d \tresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.out, p)
			}
			message := fmt.Sprintf("-------------------\nConf file %s: \n %s", f, p)
			fmt.Println(message)
		}
	}
}

func TestProcessEnvArrayValue(t *testing.T) {
	fileMap["mqtt_source"] = "test/mqtt_source.yaml"
	var tests = []struct {
		vars []string
		file string
		expt map[string]interface{}
	}{
		{
			vars: []string{
				"MQTT_SOURCE__DEFAULT__SERVERS=[tcp://10.211.55.12:1883,tcp://10.211.55.13:1883]",
				"MQTT_SOURCE__DEFAULT__TEST=[1,2]",
			},
			file: "mqtt_source",
			expt: map[string]interface{}{
				"default": map[string]interface{}{
					"servers": []interface{}{"tcp://10.211.55.12:1883", "tcp://10.211.55.13:1883"},
					"test":    []interface{}{int64(1), int64(2)},
				},
			},
		},
	}
	files := make(map[string]map[string]interface{})
	for i, tt := range tests {
		ProcessEnv(files, tt.vars)
		if !reflect.DeepEqual(tt.expt, files[tt.file]) {
			t.Errorf("%d \tresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, tt.expt, files[tt.file])
		}
	}
}