ascend-docker-runtime/runtime/main_test.go

/* Copyright(C) 2022. 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 main
package main

import (
	"os"
	"testing"

	"github.com/opencontainers/runtime-spec/specs-go"
	"github.com/prashantv/gostub"
)

func TestArgsIsCreate(t *testing.T) {
	t.Log("进入测试用例")

	testArgs := []string{"create", "--bundle", "."}
	stub := gostub.Stub(&os.Args, testArgs)
	defer stub.Reset()

	stub.Stub(&execRunc, func() error {
		t.Log("execute stub")
		return nil
	})

	if err := doProcess(); err != nil {
		t.Fatalf("%v", err)
	}
}

func TestArgsIsCreateCase1(t *testing.T) {
	t.Log("进入测试用例")

	testArgs := []string{"create", "--bundle"}
	stub := gostub.Stub(&os.Args, testArgs)
	defer stub.Reset()

	stub.Stub(&execRunc, func() error {
		t.Log("execute stub")
		return nil
	})

	if err := doProcess(); err == nil {
		t.Fatalf("%v", err)
	}
}

func TestArgsIsCreateCase2(t *testing.T) {
	t.Log("进入测试用例")

	testArgs := []string{"create", "--bundle", ""}
	stub := gostub.Stub(&os.Args, testArgs)
	defer stub.Reset()

	stub.Stub(&execRunc, func() error {
		t.Log("execute stub")
		return nil
	})

	if err := doProcess(); err != nil {
		t.Fatalf("%v", err)
	}
}

func TestArgsIsCreateCase3(t *testing.T) {
	t.Log("进入测试用例")

	if err := os.Mkdir("./test", 0655); err != nil {
	}
	f, err := os.Create("./test/config.json")
	defer f.Close()
	if err != nil {
	}
	testArgs := []string{"create", "--bundle", "./test"}
	stub := gostub.Stub(&os.Args, testArgs)
	defer stub.Reset()

	stub.Stub(&execRunc, func() error {
		t.Log("execute stub")
		return nil
	})

	if err := doProcess(); err == nil {
		t.Fatalf("%v", err)
	}
}

func TestArgsIsCreateCase4(t *testing.T) {
	t.Log("进入测试用例")

	if err := os.Mkdir("./test", 0655); err != nil {
	}
	f, err := os.Create("./test/config.json")
	defer f.Close()
	if err != nil {
	}
	testArgs := []string{"spec", "--bundle", "./test"}
	stub := gostub.Stub(&os.Args, testArgs)
	defer stub.Reset()

	stub.Stub(&execRunc, func() error {
		t.Log("execute stub")
		return nil
	})

	if err := doProcess(); err == nil {
		t.Fatalf("%v", err)
	}
}

func TestModifySpecFile(t *testing.T) {
	if err := modifySpecFile("./test/config.json"); err != nil {
		t.Log("run modifySpecFile failed")
	}
}

func TestModifySpecFileCase1(t *testing.T) {
	file := "./test"
	if err := os.Mkdir("./test", 0400); err != nil {

	}

	if err := modifySpecFile(file); err != nil {
		t.Log("run modifySpecFile failed")
	}
	if err := os.Remove(file); err != nil {

	}
}

func TestModifySpecFileCase2(t *testing.T) {
	file := "./test.json"
	f, err := os.Create(file)
	defer f.Close()
	if err != nil {
		t.Log("create file error")
	}

	if err := modifySpecFile(file); err != nil {
		t.Log("run modifySpecFile failed")
	}
	if err := os.Remove(file); err != nil {

	}
}

func TestModifySpecFileCase3(t *testing.T) {
	file := "./test_spec.json"
	if err := modifySpecFile(file); err != nil {
		t.Log("run modifySpecFile failed")
	}
}

func TestAddHook(t *testing.T) {
	var specArgs = &specs.Spec{}
	if err := addHook(specArgs); err != nil {
	}
}

func TestAddHookCase1(t *testing.T) {
	var specArgs = &specs.Spec{}
	stub := gostub.Stub(&hookCliPath, ".")
	defer stub.Reset()

	if err := addHook(specArgs); err != nil {
	}
}

func TestAddHookCase2(t *testing.T) {
	var specArgs = &specs.Spec{}
	stub := gostub.Stub(&hookCliPath, ".")
	defer stub.Reset()
	stub.Stub(&hookDefaultFile, ".")
	if err := addHook(specArgs); err != nil {
	}
}

func TestAddHookCase3(t *testing.T) {
	file := "/usr/local/bin/ascend-docker-hook"
	filenew := "/usr/local/bin/ascend-docker-hook-1"

	if err := os.Rename(file, filenew); err != nil {
		t.Log("rename ", file)
	}
	var specArgs = &specs.Spec{}
	if err := addHook(specArgs); err != nil {
	}
	if err := os.Rename(filenew, file); err != nil {
		t.Log("rename ", file)
	}
}

func TestExecRunc(t *testing.T) {
	stub := gostub.Stub(&dockerRuncName, "abc-runc")
	stub.Stub(&runcName, "runc123")
	defer stub.Reset()

	if err := execRunc(); err != nil {

	}
}