Merge "Add UTs for plugins module"

2 files changed, 104 insertions, 5 deletions

diff --git a/src/k8splugin/krd/plugins.go b/src/k8splugin/krd/plugins.go
index 41b83226..9ccb04fa 100644
--- a/src/k8splugin/krd/plugins.go
+++ b/src/k8splugin/krd/plugins.go
@@ -38,21 +38,25 @@ type ResourceData struct {
 
 // DecodeYAML reads a YAMl file to extract the Kubernetes object definition
 var DecodeYAML = func(path string) (runtime.Object, error) {
-	if _, err := os.Stat(path); os.IsNotExist(err) {
-		return nil, pkgerrors.New("File " + path + " not found")
+	if _, err := os.Stat(path); err != nil {
+		if os.IsNotExist(err) {
+			return nil, pkgerrors.New("File " + path + " not found")
+		} else {
+			return nil, pkgerrors.Wrap(err, "Stat file error")
+		}
 	}
 
-	log.Println("Reading deployment YAML")
+	log.Println("Reading YAML file")
 	rawBytes, err := ioutil.ReadFile(path)
 	if err != nil {
-		return nil, pkgerrors.Wrap(err, "Deployment YAML file read error")
+		return nil, pkgerrors.Wrap(err, "Read YAML file error")
 	}
 
 	log.Println("Decoding deployment YAML")
 	decode := scheme.Codecs.UniversalDeserializer().Decode
 	obj, _, err := decode(rawBytes, nil, nil)
 	if err != nil {
-		return nil, pkgerrors.Wrap(err, "Deserialize deployment error")
+		return nil, pkgerrors.Wrap(err, "Deserialize YAML error")
 	}
 
 	return obj, nil
diff --git a/src/k8splugin/krd/plugins_test.go b/src/k8splugin/krd/plugins_test.go
new file mode 100644
index 00000000..81d2784e
--- /dev/null
+++ b/src/k8splugin/krd/plugins_test.go
@@ -0,0 +1,95 @@
+// +build unit
+
+/*
+Copyright 2018 Intel Corporation.
+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 krd
+
+import (
+	"strings"
+	"testing"
+
+	appsV1 "k8s.io/api/apps/v1"
+	coreV1 "k8s.io/api/core/v1"
+	metaV1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+)
+
+func TestDecodeYAML(t *testing.T) {
+	testCases := []struct {
+		label          string
+		input          string
+		expectedResult runtime.Object
+		expectedError  string
+	}{
+		{
+			label:         "Fail to read non-existing YAML file",
+			input:         "unexisting-file.yaml",
+			expectedError: "not found",
+		},
+		{
+			label:         "Fail to read invalid YAML format",
+			input:         "./plugins_test.go",
+			expectedError: "mapping values are not allowed in this contex",
+		},
+		{
+			label: "Successfully read YAML file",
+			input: "../mock_files/mock_yamls/deployment.yaml",
+			expectedResult: &appsV1.Deployment{
+				ObjectMeta: metaV1.ObjectMeta{
+					Name: "mock-deployment",
+				},
+				Spec: appsV1.DeploymentSpec{
+					Template: coreV1.PodTemplateSpec{
+						ObjectMeta: metaV1.ObjectMeta{
+							Labels: map[string]string{"app": "sise"},
+						},
+						Spec: coreV1.PodSpec{
+							Containers: []coreV1.Container{
+								coreV1.Container{
+									Name:  "sise",
+									Image: "mhausenblas/simpleservice:0.5.0",
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	for _, testCase := range testCases {
+		t.Run(testCase.label, func(t *testing.T) {
+			result, err := DecodeYAML(testCase.input)
+			if err != nil {
+				if testCase.expectedError == "" {
+					t.Fatalf("Decode YAML method return an un-expected (%s)", err)
+				}
+				if !strings.Contains(string(err.Error()), testCase.expectedError) {
+					t.Fatalf("Decode YAML method returned an error (%s)", err)
+				}
+			} else {
+				if testCase.expectedError != "" && testCase.expectedResult == nil {
+					t.Fatalf("Decode YAML method was expecting \"%s\" error message", testCase.expectedError)
+				}
+				if result == nil {
+					t.Fatal("Decode YAML method returned nil result")
+				}
+				// if !reflect.DeepEqual(testCase.expectedResult, result) {
+
+				// 	t.Fatalf("Decode YAML method returned: \n%v\n and it was expected: \n%v", result, testCase.expectedResult)
+				// }
+			}
+		})
+	}
+}