/* * 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 api import ( "bytes" "encoding/json" "io" "net/http" "net/http/httptest" "reflect" "testing" moduleLib "github.com/onap/multicloud-k8s/src/orchestrator/pkg/module" pkgerrors "github.com/pkg/errors" ) //Creating an embedded interface via anonymous variable //This allows us to make mockDB satisfy the DatabaseConnection //interface even if we are not implementing all the methods in it type mockProjectManager struct { // Items and err will be used to customize each test // via a localized instantiation of mockProjectManager Items []moduleLib.Project Err error } func (m *mockProjectManager) CreateProject(inp moduleLib.Project) (moduleLib.Project, error) { if m.Err != nil { return moduleLib.Project{}, m.Err } return m.Items[0], nil } func (m *mockProjectManager) GetProject(name string) (moduleLib.Project, error) { if m.Err != nil { return moduleLib.Project{}, m.Err } return m.Items[0], nil } func (m *mockProjectManager) DeleteProject(name string) error { return m.Err } func TestProjectCreateHandler(t *testing.T) { testCases := []struct { label string reader io.Reader expected moduleLib.Project expectedCode int projectClient *mockProjectManager }{ { label: "Missing Body Failure", expectedCode: http.StatusBadRequest, projectClient: &mockProjectManager{}, }, { label: "Create Project", expectedCode: http.StatusCreated, reader: bytes.NewBuffer([]byte(`{ "metadata" : { "name": "testProject", "description": "Test Project used for unit testing", "userData1": "data1", "userData2": "data2" } }`)), expected: moduleLib.Project{ MetaData: moduleLib.ProjectMetaData{ Name: "testProject", Description: "Test Project used for unit testing", UserData1: "data1", UserData2: "data2", }, }, projectClient: &mockProjectManager{ //Items that will be returned by the mocked Client Items: []moduleLib.Project{ moduleLib.Project{ MetaData: moduleLib.ProjectMetaData{ Name: "testProject", Description: "Test Project used for unit testing", UserData1: "data1", UserData2: "data2", }, }, }, }, }, { label: "Missing Project Name in Request Body", reader: bytes.NewBuffer([]byte(`{ "description":"test description" }`)), expectedCode: http.StatusBadRequest, projectClient: &mockProjectManager{}, }, } for _, testCase := range testCases { t.Run(testCase.label, func(t *testing.T) { request := httptest.NewRequest("POST", "/v2/projects", testCase.reader) resp := executeRequest(request, NewRouter(testCase.projectClient)) //Check returned code if resp.StatusCode != testCase.expectedCode { t.Fatalf("Expected %d; Got: %d", testCase.expectedCode, resp.StatusCode) } //Check returned body only if statusCreated if resp.StatusCode == http.StatusCreated { got := moduleLib.Project{} json.NewDecoder(resp.Body).Decode(&got) if reflect.DeepEqual(testCase.expected, got) == false { t.Errorf("createHandler returned unexpected body: got %v;"+ " expected %v", got, testCase.expected) } } }) } } func TestProjectGetHandler(t *testing.T) { testCases := []struct { label string expected moduleLib.Project name, version string expectedCode int projectClient *mockProjectManager }{ { label: "Get Project", expectedCode: http.StatusOK, expected: moduleLib.Project{ MetaData: moduleLib.ProjectMetaData{ Name: "testProject", Description: "Test Project used for unit testing", UserData1: "data1", UserData2: "data2", }, }, name: "testProject", projectClient: &mockProjectManager{ Items: []moduleLib.Project{ moduleLib.Project{ MetaData: moduleLib.ProjectMetaData{ Name: "testProject", Description: "Test Project used for unit testing", UserData1: "data1", UserData2: "data2", }, }, }, }, }, { label: "Get Non-Exiting Project", expectedCode: http.StatusInternalServerError, name: "nonexistingproject", projectClient: &mockProjectManager{ Items: []moduleLib.Project{}, Err: pkgerrors.New("Internal Error"), }, }, } for _, testCase := range testCases { t.Run(testCase.label, func(t *testing.T) { request := httptest.NewRequest("GET", "/v2/projects/"+testCase.name, nil) resp := executeRequest(request, NewRouter(testCase.projectClient)) //Check returned code if resp.StatusCode != testCase.expectedCode { t.Fatalf("Expected %d; Got: %d", testCase.expectedCode, resp.StatusCode) } //Check returned body only if statusOK if resp.StatusCode == http.StatusOK { got := moduleLib.Project{} json.NewDecoder(resp.Body).Decode(&got) if reflect.DeepEqual(testCase.expected, got) == false { t.Errorf("listHandler returned unexpected body: got %v;"+ " expected %v", got, testCase.expected) } } }) } } func TestProjectDeleteHandler(t *testing.T) { testCases := []struct { label string name string version string expectedCode int projectClient *mockProjectManager }{ { label: "Delete Project", expectedCode: http.StatusNoContent, name: "testProject", projectClient: &mockProjectManager{}, }, { label: "Delete Non-Exiting Project", expectedCode: http.StatusInternalServerError, name: "testProject", projectClient: &mockProjectManager{ Err: pkgerrors.New("Internal Error"), }, }, } for _, testCase := range testCases { t.Run(testCase.label, func(t *testing.T) { request := httptest.NewRequest("DELETE", "/v2/projects/"+testCase.name, nil) resp := executeRequest(request, NewRouter(testCase.projectClient)) //Check returned code if resp.StatusCode != testCase.expectedCode { t.Fatalf("Expected %d; Got: %d", testCase.expectedCode, resp.StatusCode) } }) } }