diff options
Diffstat (limited to 'src/orchestrator/pkg/appcontext/appcontext_test.go')
-rw-r--r-- | src/orchestrator/pkg/appcontext/appcontext_test.go | 283 |
1 files changed, 283 insertions, 0 deletions
diff --git a/src/orchestrator/pkg/appcontext/appcontext_test.go b/src/orchestrator/pkg/appcontext/appcontext_test.go new file mode 100644 index 00000000..a6f3e94e --- /dev/null +++ b/src/orchestrator/pkg/appcontext/appcontext_test.go @@ -0,0 +1,283 @@ +/* + * Copyright 2020 Intel Corporation, Inc + * + * 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 appcontext + +import ( + "fmt" + "testing" + "strings" + pkgerrors "github.com/pkg/errors" +) + +// Mock run time context +type MockRunTimeContext struct { + Items map[string]interface{} + Err error +} + +func (c *MockRunTimeContext) RtcCreate() (interface{}, error) { + var key string = "/context/9345674458787728/" + + if c.Items == nil { + c.Items = make(map[string]interface{}) + } + c.Items[key] = "9345674458787728" + return interface{}(key), c.Err + +} + +func (c *MockRunTimeContext) RtcGet() (interface{}, error) { + var key string = "/context/9345674458787728/" + return key, c.Err +} + +func (c *MockRunTimeContext) RtcAddLevel(handle interface{}, level string, value string) (interface{}, error) { + str := fmt.Sprintf("%v", handle) + level + "/" + value + "/" + c.Items[str] = value + return nil, c.Err + +} + +func (c *MockRunTimeContext) RtcAddResource(handle interface{}, resname string, value interface{}) (interface{}, error) { + str := fmt.Sprintf("%v", handle) + "resource" + "/" + resname + "/" + c.Items[str] = value + return nil, c.Err + +} + +func (c *MockRunTimeContext) RtcAddInstruction(handle interface{}, level string, insttype string, value interface{}) (interface{}, error) { + str := fmt.Sprintf("%v", handle) + level + "/" + insttype + "/" + c.Items[str] = value + return nil, c.Err +} + +func (c *MockRunTimeContext) RtcDeletePair(handle interface{}) (error) { + str := fmt.Sprintf("%v", handle) + delete(c.Items, str) + return c.Err +} + +func (c *MockRunTimeContext) RtcDeletePrefix(handle interface{}) (error) { + for k, _ := range c.Items { + delete(c.Items, k) + } + return c.Err +} + +func (c *MockRunTimeContext) RtcGetHandles(handle interface{}) ([]interface{}, error) { + var keys []interface{} + + for k, _ := range c.Items { + keys = append(keys, string(k)) + } + return keys, c.Err +} + +func (c *MockRunTimeContext) RtcGetValue(handle interface{}, value interface{}) (error) { + key := fmt.Sprintf("%v", handle) + var s *string + s = value.(*string) + for kvKey, kvValue := range c.Items { + if kvKey == key { + *s = kvValue.(string) + return c.Err + } + } + return c.Err +} + +func (c *MockRunTimeContext) RtcUpdateValue(handle interface{}, value interface{}) (error) { + key := fmt.Sprintf("%v", handle) + c.Items[key] = value + return c.Err +} + +func TestCreateCompositeApp(t *testing.T) { + var ac = AppContext{} + testCases := []struct { + label string + mockRtcontext *MockRunTimeContext + expectedError string + }{ + { + label: "Success case", + mockRtcontext: &MockRunTimeContext{}, + }, + { + label: "Create returns error case", + mockRtcontext: &MockRunTimeContext{Err: pkgerrors.Errorf("Error creating run time context:")}, + expectedError: "Error creating run time context:", + }, + } + + for _, testCase := range testCases { + t.Run(testCase.label, func(t *testing.T) { + ac.rtc = testCase.mockRtcontext + _, err := ac.CreateCompositeApp() + if err != nil { + if !strings.Contains(string(err.Error()), testCase.expectedError) { + t.Fatalf("Method returned an error (%s)", err) + } + } + + }) + } +} + +func TestGetCompositeApp(t *testing.T) { + var ac = AppContext{} + testCases := []struct { + label string + mockRtcontext *MockRunTimeContext + expectedError string + }{ + { + label: "Success case", + mockRtcontext: &MockRunTimeContext{}, + }, + { + label: "Get returns error case", + mockRtcontext: &MockRunTimeContext{Err: pkgerrors.Errorf("Error getting run time context:")}, + expectedError: "Error getting run time context:", + }, + } + + for _, testCase := range testCases { + t.Run(testCase.label, func(t *testing.T) { + ac.rtc = testCase.mockRtcontext + _, err := ac.GetCompositeApp() + if err != nil { + if !strings.Contains(string(err.Error()), testCase.expectedError) { + t.Fatalf("Method returned an error (%s)", err) + } + } + + }) + } +} + +func TestDeleteCompositeApp(t *testing.T) { + var ac = AppContext{} + testCases := []struct { + label string + mockRtcontext *MockRunTimeContext + expectedError string + }{ + { + label: "Success case", + mockRtcontext: &MockRunTimeContext{}, + }, + { + label: "Delete returns error case", + mockRtcontext: &MockRunTimeContext{Err: pkgerrors.Errorf("Error deleting run time context:")}, + expectedError: "Error deleting run time context:", + }, + } + + for _, testCase := range testCases { + t.Run(testCase.label, func(t *testing.T) { + ac.rtc = testCase.mockRtcontext + err := ac.DeleteCompositeApp() + if err != nil { + if !strings.Contains(string(err.Error()), testCase.expectedError) { + t.Fatalf("Method returned an error (%s)", err) + } + } + + }) + } +} + +func TestAddApp(t *testing.T) { + var ac = AppContext{} + testCases := []struct { + label string + mockRtcontext *MockRunTimeContext + key interface{} + expectedError string + }{ + { + label: "Success case", + mockRtcontext: &MockRunTimeContext{}, + key: "/context/9345674458787728/", + }, + { + label: "Delete returns error case", + mockRtcontext: &MockRunTimeContext{Err: pkgerrors.Errorf("Error adding app to run time context:")}, + key: "/context/9345674458787728/", + expectedError: "Error adding app to run time context:", + }, + } + + for _, testCase := range testCases { + t.Run(testCase.label, func(t *testing.T) { + ac.rtc = testCase.mockRtcontext + _, err := ac.CreateCompositeApp() + _, err = ac.AddApp(testCase.key, "testapp1") + if err != nil { + if !strings.Contains(string(err.Error()), testCase.expectedError) { + t.Fatalf("Method returned an error (%s)", err) + } + } + + }) + } +} + +func TestGetAppHandle(t *testing.T) { + var ac = AppContext{} + testCases := []struct { + label string + mockRtcontext *MockRunTimeContext + key interface{} + appname string + expectedError string + }{ + { + label: "Success case", + mockRtcontext: &MockRunTimeContext{}, + key: "/context/9345674458787728/", + appname: "testapp1", + }, + { + label: "Invalid app name case", + mockRtcontext: &MockRunTimeContext{}, + key: "/context/9345674458787728/", + appname: "", + }, + { + label: "Delete returns error case", + mockRtcontext: &MockRunTimeContext{Err: pkgerrors.Errorf("Error getting app handle from run time context:")}, + key: "/context/9345674458787728/", + appname: "testapp1", + expectedError: "Error getting app handle from run time context:", + }, + } + + for _, testCase := range testCases { + t.Run(testCase.label, func(t *testing.T) { + ac.rtc = testCase.mockRtcontext + _, err := ac.GetAppHandle(testCase.appname) + if err != nil { + if !strings.Contains(string(err.Error()), testCase.expectedError) { + t.Fatalf("Method returned an error (%s)", err) + } + } + + }) + } +} |