chassis_test.go

/**
 * Copyright (c) 2020 eBay 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 goovn

import (
	"testing"
)

const (
	CHASSIS_HOSTNAME  = "fakehost"
	CHASSIS_NAME      = "fakechassis"
	IP                = "10.0.0.11"
	CHASSIS2_HOSTNAME = "fakehost2"
	CHASSIS2_NAME     = "fakechassis2"
	IP2               = "10.0.0.12"
)

// can be one or many encap_types similar to chassis-add of sbctl
var ENCAP_TYPES = []string{"stt", "geneve", "vxlan"}

func TestChassis(t *testing.T) {
	ovndbapi := getOVNClient(DBSB)
	t.Logf("Adding Chassis to OVN SB DB")
	ocmd, err := ovndbapi.ChassisAdd(CHASSIS_NAME, CHASSIS_HOSTNAME, ENCAP_TYPES, IP, nil, nil, nil)
	if err != nil {
		t.Fatal(err)
	}
	err = ovndbapi.Execute(ocmd)
	if err != nil {
		t.Fatalf("Adding Chassis to OVN failed with err %v", err)
	}
	t.Logf("Adding Chassis to OVN Done")

	t.Logf("Adding second Chassis to OVN SB DB")
	ocmd, err = ovndbapi.ChassisAdd(CHASSIS2_NAME, CHASSIS2_HOSTNAME, ENCAP_TYPES, IP2, nil, nil, nil)
	if err != nil {
		t.Fatal(err)
	}
	err = ovndbapi.Execute(ocmd)
	if err != nil {
		t.Fatalf("Adding second Chassis to OVN failed with err %v", err)
	}
	t.Logf("Adding second Chassis to OVN Done")

	t.Logf("Listing all the Chassis in OVN SB")
	chassis, err := ovndbapi.ChassisList()
	if err != nil {
		t.Fatal(err)
	}
	if len(chassis) != 2 {
		t.Fatalf("err listing chassis, total:%v", len(chassis))
	}
	t.Logf("Two Chassis found: %+v", chassis)

	t.Logf("Gettting Chassis by hostname")
	chassis, err = ovndbapi.ChassisGet(CHASSIS_HOSTNAME)
	if err != nil {
		t.Fatal(err)
	}
	if len(chassis) != 1 {
		t.Fatalf("err getting chassis, total:%v", len(chassis))
	}
	chName := chassis[0].Name
	t.Logf("Chassis found:%+v", chName)

	t.Logf("Deleting Chassis:%v", chName)
	ocmds := make([]*OvnCommand, 0, 2)
	ocmd, err = ovndbapi.ChassisDel(chName)
	if err != nil && err != ErrorNotFound {
		t.Fatal(err)
	}
	ocmds = append(ocmds, ocmd)
	ocmd, err = ovndbapi.ChassisDel(CHASSIS2_NAME)
	if err != nil && err != ErrorNotFound {
		t.Fatal(err)
	}
	ocmds = append(ocmds, ocmd)
	err = ovndbapi.Execute(ocmds...)
	if err != nil {
		t.Fatalf("err executing commands %v: %v", ocmds, err)
	}

	// Verify deletion
	chassis, err = ovndbapi.ChassisGet(CHASSIS_HOSTNAME)
	if err != nil {
		t.Fatal(err)
	}
	if len(chassis) != 0 {
		t.Fatalf("error: Chassis deletion not done, total:%v", len(chassis))
	}
	t.Logf("Chassis %s deleted", chName)

	t.Logf("Adding Chassis with empty hostname to OVN SB DB")
	ocmd, err = ovndbapi.ChassisAdd(CHASSIS_NAME, "", ENCAP_TYPES, IP, nil, nil, nil)
	if err != nil {
		t.Fatal(err)
	}
	err = ovndbapi.Execute(ocmd)
	if err != nil {
		t.Fatalf("Adding Chassis with empty hostname failed with err %v", err)
	}
	t.Logf("Adding Chassis with empty hostname to OVN Done")

	// verify addition
	t.Logf("Gettting Chassis by name")
	chassis, err = ovndbapi.ChassisGet(CHASSIS_NAME)
	if err != nil {
		t.Fatal(err)
	}
	if len(chassis) != 1 {
		t.Fatalf("err getting chassis, total:%v", len(chassis))
	}
	chName = chassis[0].Name
	t.Logf("Chassis found:%+v", chName)

	// delete chassis
	t.Logf("Deleting Chassis:%v", chName)
	ocmd, err = ovndbapi.ChassisDel(chName)
	if err != nil && err != ErrorNotFound {
		t.Fatal(err)
	}
	err = ovndbapi.Execute(ocmd)
	if err != nil {
		t.Fatalf("err executing command:%v", err)
	}

	//verify deletion
	chassis, err = ovndbapi.ChassisGet(CHASSIS_NAME)
	if err != nil {
		t.Fatal(err)
	}
	if len(chassis) != 0 {
		t.Fatalf("error: Chassis deletion not done, total:%v", len(chassis))
	}
	t.Logf("Chassis %s deleted", chName)
}