mirror of
https://github.com/aljazceru/kata-containers.git
synced 2025-12-18 14:54:19 +01:00
03546f75a6
Change io/ioutil to io/os packages because io/ioutil package is deprecated from 1.16: Discard => io.Discard NopCloser => io.NopCloser ReadAll => io.ReadAll ReadDir => os.ReadDir ReadFile => os.ReadFile TempDir => os.MkdirTemp TempFile => os.CreateTemp WriteFile => os.WriteFile Details: https://go.dev/doc/go1.16#ioutil Fixes: #3265 Signed-off-by: bin <bin@hyper.sh>
123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
// Copyright (c) 2018 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package virtcontainers
|
|
|
|
import (
|
|
"net"
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func testEndpointTypeSet(t *testing.T, value string, expected EndpointType) {
|
|
var endpointType EndpointType
|
|
|
|
err := endpointType.Set(value)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, endpointType, expected)
|
|
}
|
|
|
|
func TestPhysicalEndpointTypeSet(t *testing.T) {
|
|
testEndpointTypeSet(t, "physical", PhysicalEndpointType)
|
|
}
|
|
|
|
func TestVethEndpointTypeSet(t *testing.T) {
|
|
testEndpointTypeSet(t, "virtual", VethEndpointType)
|
|
}
|
|
|
|
func TestVhostUserEndpointTypeSet(t *testing.T) {
|
|
testEndpointTypeSet(t, "vhost-user", VhostUserEndpointType)
|
|
}
|
|
|
|
func TestMacvlanEndpointTypeSet(t *testing.T) {
|
|
testEndpointTypeSet(t, "macvlan", MacvlanEndpointType)
|
|
}
|
|
|
|
func TestMacvtapEndpointTypeSet(t *testing.T) {
|
|
testEndpointTypeSet(t, "macvtap", MacvtapEndpointType)
|
|
}
|
|
|
|
func TestEndpointTypeSetFailure(t *testing.T) {
|
|
var endpointType EndpointType
|
|
|
|
assert.Error(t, endpointType.Set("wrong-value"))
|
|
}
|
|
|
|
func testEndpointTypeString(t *testing.T, endpointType *EndpointType, expected string) {
|
|
result := endpointType.String()
|
|
assert.Equal(t, result, expected)
|
|
}
|
|
|
|
func TestPhysicalEndpointTypeString(t *testing.T) {
|
|
endpointType := PhysicalEndpointType
|
|
testEndpointTypeString(t, &endpointType, string(PhysicalEndpointType))
|
|
}
|
|
|
|
func TestVethEndpointTypeString(t *testing.T) {
|
|
endpointType := VethEndpointType
|
|
testEndpointTypeString(t, &endpointType, string(VethEndpointType))
|
|
}
|
|
|
|
func TestVhostUserEndpointTypeString(t *testing.T) {
|
|
endpointType := VhostUserEndpointType
|
|
testEndpointTypeString(t, &endpointType, string(VhostUserEndpointType))
|
|
}
|
|
|
|
func TestMacvlanEndpointTypeString(t *testing.T) {
|
|
endpointType := MacvlanEndpointType
|
|
testEndpointTypeString(t, &endpointType, string(MacvlanEndpointType))
|
|
}
|
|
|
|
func TestMacvtapEndpointTypeString(t *testing.T) {
|
|
endpointType := MacvtapEndpointType
|
|
testEndpointTypeString(t, &endpointType, string(MacvtapEndpointType))
|
|
}
|
|
|
|
func TestIncorrectEndpointTypeString(t *testing.T) {
|
|
var endpointType EndpointType
|
|
testEndpointTypeString(t, &endpointType, "")
|
|
}
|
|
|
|
func TestSaveLoadIfPair(t *testing.T) {
|
|
macAddr := net.HardwareAddr{0x02, 0x00, 0xCA, 0xFE, 0x00, 0x04}
|
|
|
|
tmpfile, err := os.CreateTemp("", "vc-Save-Load-net-")
|
|
assert.Nil(t, err)
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
netPair := &NetworkInterfacePair{
|
|
TapInterface: TapInterface{
|
|
ID: "uniqueTestID-4",
|
|
Name: "br4_kata",
|
|
TAPIface: NetworkInterface{
|
|
Name: "tap4_kata",
|
|
HardAddr: macAddr.String(),
|
|
},
|
|
VMFds: []*os.File{tmpfile}, // won't be saved to disk
|
|
VhostFds: []*os.File{tmpfile}, // won't be saved to disk
|
|
},
|
|
VirtIface: NetworkInterface{
|
|
Name: "eth4",
|
|
HardAddr: macAddr.String(),
|
|
},
|
|
NetInterworkingModel: DefaultNetInterworkingModel,
|
|
}
|
|
|
|
// Save to disk then Load it back.
|
|
savedIfPair := saveNetIfPair(netPair)
|
|
loadedIfPair := loadNetIfPair(savedIfPair)
|
|
|
|
// Since VMFds and VhostFds are't saved, netPair and loadedIfPair are not equal.
|
|
assert.False(t, reflect.DeepEqual(netPair, loadedIfPair))
|
|
|
|
netPair.TapInterface.VMFds = nil
|
|
netPair.TapInterface.VhostFds = nil
|
|
// They are equal now.
|
|
assert.True(t, reflect.DeepEqual(netPair, loadedIfPair))
|
|
}
|