client/profile_test.go

// Copyright (c) 2017 Huawei Technologies Co., Ltd. All Rights Reserved.
//
// 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 client

import (
	"encoding/json"
	"errors"
	"reflect"
	"strings"
	"testing"

	"github.com/opensds/opensds/pkg/model"
	. "github.com/opensds/opensds/testutils/collection"
)

var fpr = &ProfileMgr{
	Receiver: NewFakeProfileReceiver(),
}

func NewFakeProfileReceiver() Receiver {
	return &fakeProfileReceiver{}
}

type fakeProfileReceiver struct{}

func (*fakeProfileReceiver) Recv(
	string,
	method string,
	in interface{},
	out interface{},
) error {
	switch strings.ToUpper(method) {
	case "POST":
		switch out.(type) {
		case *model.ProfileSpec:
			if err := json.Unmarshal([]byte(ByteProfile), out); err != nil {
				return err
			}
			break
		case *model.ExtraSpec:
			if err := json.Unmarshal([]byte(ByteExtras), out); err != nil {
				return err
			}
			break
		default:
			return errors.New("output format not supported!")
		}
		break
	case "GET":
		switch out.(type) {
		case *model.ProfileSpec:
			if err := json.Unmarshal([]byte(ByteProfile), out); err != nil {
				return err
			}
			break
		case *[]*model.ProfileSpec:
			if err := json.Unmarshal([]byte(ByteProfiles), out); err != nil {
				return err
			}
			break
		case *model.ExtraSpec:
			if err := json.Unmarshal([]byte(ByteExtras), out); err != nil {
				return err
			}
			break
		default:
			return errors.New("output format not supported!")
		}
		break
	case "DELETE":
		break
	default:
		return errors.New("inputed method format not supported!")
	}

	return nil
}

func TestCreateProfile(t *testing.T) {
	expected := &model.ProfileSpec{
		BaseModel: &model.BaseModel{
			Id: "1106b972-66ef-11e7-b172-db03f3689c9c",
		},
		Name:        "default",
		Description: "default policy",
	}

	prf, err := fpr.CreateProfile(&model.ProfileSpec{})
	if err != nil {
		t.Error(err)
		return
	}

	if !reflect.DeepEqual(prf, expected) {
		t.Errorf("Expected %v, got %v", expected, prf)
		return
	}
}

func TestGetProfile(t *testing.T) {
	var prfID = "1106b972-66ef-11e7-b172-db03f3689c9c"
	expected := &model.ProfileSpec{
		BaseModel: &model.BaseModel{
			Id: "1106b972-66ef-11e7-b172-db03f3689c9c",
		},
		Name:        "default",
		Description: "default policy",
	}

	prf, err := fpr.GetProfile(prfID)
	if err != nil {
		t.Error(err)
		return
	}

	if !reflect.DeepEqual(prf, expected) {
		t.Errorf("Expected %v, got %v", expected, prf)
		return
	}
}

func TestListProfiles(t *testing.T) {
	expected := []*model.ProfileSpec{
		{
			BaseModel: &model.BaseModel{
				Id: "1106b972-66ef-11e7-b172-db03f3689c9c",
			},
			Name:        "default",
			Description: "default policy",
		},
		{
			BaseModel: &model.BaseModel{
				Id: "2f9c0a04-66ef-11e7-ade2-43158893e017",
			},
			Name:        "silver",
			Description: "silver policy",
			Extras: model.ExtraSpec{
				"dataStorage": map[string]interface{}{
					"provisioningPolicy": "Thin",
					"isSpaceEfficient":   true,
				},
				"ioConnectivity": map[string]interface{}{
					"accessProtocol": "rbd",
					"maxIOPS":        float64(5000000),
					"maxBWS":         float64(500),
				},
			},
		},
	}

	prfs, err := fpr.ListProfiles()
	if err != nil {
		t.Error(err)
		return
	}

	if !reflect.DeepEqual(prfs, expected) {
		t.Errorf("Expected %v, got %v", expected[1], prfs[1])
		return
	}
}

func TestDeleteProfile(t *testing.T) {
	var prfID = "1106b972-66ef-11e7-b172-db03f3689c9c"

	if err := fpr.DeleteProfile(prfID); err != nil {
		t.Error(err)
		return
	}
}

func TestAddExtraProperty(t *testing.T) {
	var prfID = "2f9c0a04-66ef-11e7-ade2-43158893e017"
	expected := &SampleExtras

	exts, err := fpr.AddExtraProperty(prfID, &model.ExtraSpec{})
	if err != nil {
		t.Error(err)
		return
	}

	if !reflect.DeepEqual(exts, expected) {
		t.Errorf("Expected %v, got %v", expected, exts)
		return
	}
}

func TestListExtraProperties(t *testing.T) {
	var prfID = "2f9c0a04-66ef-11e7-ade2-43158893e017"
	expected := &SampleExtras

	exts, err := fpr.ListExtraProperties(prfID)
	if err != nil {
		t.Error(err)
		return
	}

	if !reflect.DeepEqual(exts, expected) {
		t.Errorf("Expected %v, got %v", expected, exts)
		return
	}
}

func TestRemoveExtraProperty(t *testing.T) {
	var prfID, extraKey = "2f9c0a04-66ef-11e7-ade2-43158893e017", "diskType"

	if err := fpr.RemoveExtraProperty(prfID, extraKey); err != nil {
		t.Error(err)
		return
	}
}