blob: c2efd58c1af9375dad6c9e362f9cbdc701067998 [file] [log] [blame]
khenaidoo59ce9dd2019-11-11 13:05:32 -05001/*
2 * Copyright 2019-present Open Networking Foundation
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package mocks
18
19import (
khenaidooc7005fc2019-11-18 19:23:57 -050020 "fmt"
serkant.uluderyab38671c2019-11-01 09:35:38 -070021 "github.com/opencord/voltha-lib-go/v3/pkg/db/kvstore"
khenaidooc7005fc2019-11-18 19:23:57 -050022 "github.com/phayes/freeport"
khenaidoo59ce9dd2019-11-11 13:05:32 -050023 "github.com/stretchr/testify/assert"
khenaidoo59ce9dd2019-11-11 13:05:32 -050024 "os"
25 "testing"
26)
27
28var etcdServer *EtcdServer
29var client *kvstore.EtcdClient
30
31func setup() {
khenaidooc7005fc2019-11-18 19:23:57 -050032 clientPort, err := freeport.GetFreePort()
33 if err != nil {
khenaidoob332f9b2020-01-16 16:25:26 -050034 logger.Fatal(err)
khenaidooc7005fc2019-11-18 19:23:57 -050035 }
36 peerPort, err := freeport.GetFreePort()
37 if err != nil {
khenaidoob332f9b2020-01-16 16:25:26 -050038 logger.Fatal(err)
khenaidooc7005fc2019-11-18 19:23:57 -050039 }
40 etcdServer = StartEtcdServer(MKConfig("voltha.mock.test", clientPort, peerPort, "voltha.lib.mocks.etcd", "error"))
khenaidoo59ce9dd2019-11-11 13:05:32 -050041 if etcdServer == nil {
khenaidoob332f9b2020-01-16 16:25:26 -050042 logger.Fatal("Embedded server failed to start")
khenaidoo59ce9dd2019-11-11 13:05:32 -050043 }
khenaidooc7005fc2019-11-18 19:23:57 -050044 clientAddr := fmt.Sprintf("localhost:%d", clientPort)
45 client, err = kvstore.NewEtcdClient(clientAddr, 10)
khenaidoo59ce9dd2019-11-11 13:05:32 -050046 if err != nil || client == nil {
47 etcdServer.Stop()
khenaidoob332f9b2020-01-16 16:25:26 -050048 logger.Fatal("Failed to create an Etcd client")
khenaidoo59ce9dd2019-11-11 13:05:32 -050049 }
50}
51
52func TestEtcdServerRW(t *testing.T) {
53 key := "myKey-1"
54 value := "myVal-1"
55 err := client.Put(key, value, 10)
56 assert.Nil(t, err)
57 kvp, err := client.Get(key, 10)
58 assert.Nil(t, err)
59 assert.NotNil(t, kvp)
60 assert.Equal(t, kvp.Key, key)
61 val, err := kvstore.ToString(kvp.Value)
62 assert.Nil(t, err)
63 assert.Equal(t, value, val)
64}
65
66func TestEtcdServerReserve(t *testing.T) {
67 assert.NotNil(t, client)
68 txId := "tnxId-1"
69 val, err := client.Reserve("transactions", txId, 10)
70 assert.Nil(t, err)
71 assert.NotNil(t, val)
72 assert.Equal(t, val, txId)
73}
74
75func shutdown() {
76 if client != nil {
77 client.Close()
78 }
79 if etcdServer != nil {
80 etcdServer.Stop()
81 }
82}
83
84func TestMain(m *testing.M) {
85 setup()
86 code := m.Run()
87 shutdown()
88 os.Exit(code)
89}