2 Copyright (c) 2021 AT&T Intellectual Property.
3 Copyright (c) 2018-2021 Nokia.
5 Licensed under the Apache License, Version 2.0 (the "License");
6 you may not use this file except in compliance with the License.
7 You may obtain a copy of the License at
9 http://www.apache.org/licenses/LICENSE-2.0
11 Unless required by applicable law or agreed to in writing, software
12 distributed under the License is distributed on an "AS IS" BASIS,
13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 See the License for the specific language governing permissions and
15 limitations under the License.
19 * This source code is part of the near-RT RIC (RAN Intelligent Controller)
20 * platform project (RICP).
28 "gerrit.o-ran-sc.org/r/ric-plt/sdlgo/internal/cli"
29 "gerrit.o-ran-sc.org/r/ric-plt/sdlgo/internal/mocks"
30 "github.com/stretchr/testify/assert"
36 type keysMock struct {
37 sdlIface *mocks.MockSdlApi
44 func setupKeysCliMock(ns, pattern string, keys []string, err error) {
47 mKeys.pattern = pattern
52 func newMockSdlApi() cli.ISyncStorage {
53 mKeys.sdlIface = new(mocks.MockSdlApi)
54 mKeys.sdlIface.On("ListKeys", mKeys.ns, mKeys.pattern).Return(mKeys.keys, mKeys.err)
58 func TestKeysCmdShowHelp(t *testing.T) {
60 expNokErrZeroArgs := errors.New("accepts between 1 and 2 arg(s), received 0")
61 expNokErrThreeArgs := errors.New("accepts between 1 and 2 arg(s), received 3")
62 expHelp := "Usage:\n keys <namespace> [pattern|default '*'] [flags]"
68 {args: []string{"-h"}, expErr: expOkErr, expOut: expHelp},
69 {args: []string{"--help"}, expErr: expOkErr, expOut: expHelp},
70 {args: []string{}, expErr: expNokErrZeroArgs, expOut: expHelp},
71 {args: []string{"ns", "key", "extra-arg"}, expErr: expNokErrThreeArgs, expOut: expHelp},
74 for _, test := range tests {
75 buf := new(bytes.Buffer)
76 cmd := cli.NewKeysCmdForTest(newMockSdlApi)
79 cmd.SetArgs(test.args)
81 result := buf.String()
83 assert.Equal(t, test.expErr, err)
84 assert.Contains(t, result, test.expOut)
88 func TestKeysCmdSuccess(t *testing.T) {
90 argsNsPattern := []string{"testns", "*"}
91 argsNs := []string{"testns"}
98 {args: argsNs, outListKeys: []string{"key13"}, expOut: "key13", expErr: expOkErr},
99 {args: argsNsPattern, outListKeys: []string{"key1"}, expOut: "key1", expErr: expOkErr},
100 {args: argsNsPattern, outListKeys: []string{"key1", "key2"}, expOut: "key1\nkey2", expErr: expOkErr},
101 {args: argsNsPattern, outListKeys: []string{"key2", "key3", "key1"}, expOut: "key1\nkey2\nkey3", expErr: expOkErr},
102 {args: argsNsPattern, outListKeys: []string{}, expOut: "", expErr: expOkErr},
105 for _, test := range tests {
106 buf := new(bytes.Buffer)
107 if len(test.args) > 1 {
108 setupKeysCliMock(test.args[0], test.args[1], test.outListKeys, test.expErr)
110 setupKeysCliMock(test.args[0], "*", test.outListKeys, test.expErr)
112 cmd := cli.NewKeysCmdForTest(newMockSdlApi)
115 cmd.SetArgs(test.args)
117 result := buf.String()
119 assert.Equal(t, test.expErr, err)
120 assert.Contains(t, result, test.expOut)
124 func TestKeysCmdFails(t *testing.T) {
125 argsNs := []string{"testns"}
126 expNokErr := errors.New("Boom!")
127 expNokOut := "Usage:\n keys <namespace> [pattern|default '*'] [flags]"
129 buf := new(bytes.Buffer)
130 setupKeysCliMock("testns", "*", []string{"very wrong"}, expNokErr)
131 cmd := cli.NewKeysCmdForTest(newMockSdlApi)
136 result := buf.String()
138 assert.Equal(t, expNokErr, err)
139 assert.Contains(t, result, expNokOut)