mirror of
https://github.com/intel/intel-device-plugins-for-kubernetes.git
synced 2025-06-03 03:59:37 +00:00
332 lines
9.0 KiB
Go
332 lines
9.0 KiB
Go
// Copyright 2017 Intel Corporation. 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
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
pluginapi "k8s.io/kubernetes/pkg/kubelet/apis/deviceplugin/v1beta1"
|
|
|
|
"github.com/intel/intel-device-plugins-for-kubernetes/internal/deviceplugin"
|
|
)
|
|
|
|
func createTestDirs(devfs, sysfs string, devfsDirs, sysfsDirs []string, sysfsFiles map[string][]byte) error {
|
|
var err error
|
|
|
|
for _, devfsdir := range devfsDirs {
|
|
err = os.MkdirAll(path.Join(devfs, devfsdir), 0755)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to create fake device directory: %+v", err)
|
|
}
|
|
}
|
|
for _, sysfsdir := range sysfsDirs {
|
|
err = os.MkdirAll(path.Join(sysfs, sysfsdir), 0755)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to create fake device directory: %+v", err)
|
|
}
|
|
}
|
|
for filename, body := range sysfsFiles {
|
|
err = ioutil.WriteFile(path.Join(sysfs, filename), body, 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to create fake vendor file: %+v", err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func TestDiscoverFPGAs(t *testing.T) {
|
|
tmpdir := fmt.Sprintf("/tmp/fpgaplugin-TestDiscoverFPGAs-%d", time.Now().Unix())
|
|
sysfs := path.Join(tmpdir, "sys", "class", "fpga")
|
|
devfs := path.Join(tmpdir, "dev")
|
|
tcases := []struct {
|
|
devfsdirs []string
|
|
sysfsdirs []string
|
|
sysfsfiles map[string][]byte
|
|
expectedResult map[string]map[string]deviceplugin.DeviceInfo
|
|
expectedErr bool
|
|
}{
|
|
{
|
|
expectedResult: nil,
|
|
expectedErr: true,
|
|
},
|
|
{
|
|
sysfsdirs: []string{"intel-fpga-dev.0"},
|
|
expectedResult: nil,
|
|
expectedErr: true,
|
|
},
|
|
{
|
|
sysfsdirs: []string{"intel-fpga-dev.0/intel-fpga-port.0"},
|
|
expectedResult: nil,
|
|
expectedErr: true,
|
|
},
|
|
{
|
|
sysfsdirs: []string{
|
|
"intel-fpga-dev.0/intel-fpga-port.0",
|
|
},
|
|
sysfsfiles: map[string][]byte{
|
|
"intel-fpga-dev.0/intel-fpga-port.0/afu_id": []byte("d8424dc4a4a3c413f89e433683f9040b\n"),
|
|
},
|
|
expectedResult: nil,
|
|
expectedErr: true,
|
|
},
|
|
{
|
|
sysfsdirs: []string{
|
|
"intel-fpga-dev.0/intel-fpga-port.0",
|
|
"intel-fpga-dev.0/intel-fpga-fme.0",
|
|
"intel-fpga-dev.1/intel-fpga-port.1",
|
|
"intel-fpga-dev.1/intel-fpga-fme.1",
|
|
"intel-fpga-dev.2/intel-fpga-port.2",
|
|
"intel-fpga-dev.2/intel-fpga-fme.2",
|
|
},
|
|
sysfsfiles: map[string][]byte{
|
|
"intel-fpga-dev.0/intel-fpga-port.0/afu_id": []byte("d8424dc4a4a3c413f89e433683f9040b\n"),
|
|
"intel-fpga-dev.1/intel-fpga-port.1/afu_id": []byte("d8424dc4a4a3c413f89e433683f9040b\n"),
|
|
"intel-fpga-dev.2/intel-fpga-port.2/afu_id": []byte("47595d0fae972fbed0c51b4a41c7a349\n"),
|
|
},
|
|
devfsdirs: []string{
|
|
"intel-fpga-port.0", "intel-fpga-fme.0",
|
|
"intel-fpga-port.1", "intel-fpga-fme.1",
|
|
"intel-fpga-port.2", "intel-fpga-fme.2",
|
|
},
|
|
expectedResult: map[string]map[string]deviceplugin.DeviceInfo{
|
|
"d8424dc4a4a3c413f89e433683f9040b": map[string]deviceplugin.DeviceInfo{
|
|
"intel-fpga-dev.0": deviceplugin.DeviceInfo{
|
|
State: "Healthy",
|
|
Nodes: []string{
|
|
path.Join(tmpdir, "/dev/intel-fpga-port.0"),
|
|
},
|
|
},
|
|
"intel-fpga-dev.1": deviceplugin.DeviceInfo{
|
|
State: "Healthy",
|
|
Nodes: []string{
|
|
path.Join(tmpdir, "/dev/intel-fpga-port.1"),
|
|
},
|
|
},
|
|
},
|
|
"47595d0fae972fbed0c51b4a41c7a349": map[string]deviceplugin.DeviceInfo{
|
|
"intel-fpga-dev.2": deviceplugin.DeviceInfo{
|
|
State: "Healthy",
|
|
Nodes: []string{
|
|
path.Join(tmpdir, "/dev/intel-fpga-port.2"),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedErr: false,
|
|
},
|
|
}
|
|
|
|
for _, tcase := range tcases {
|
|
err := createTestDirs(devfs, sysfs, tcase.devfsdirs, tcase.sysfsdirs, tcase.sysfsfiles)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
result, err := discoverFPGAs(sysfs, devfs)
|
|
if tcase.expectedErr && err == nil {
|
|
t.Error("Expected error hasn't been triggered")
|
|
}
|
|
|
|
if tcase.expectedResult != nil && !reflect.DeepEqual(result, tcase.expectedResult) {
|
|
t.Logf("Expected result: %+v", tcase.expectedResult)
|
|
t.Logf("Actual result: %+v", result)
|
|
t.Error("Got unexpeced result")
|
|
}
|
|
|
|
err = os.RemoveAll(tmpdir)
|
|
if err != nil {
|
|
t.Fatalf("Failed to remove fake device directory: %+v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Minimal implementation of pluginapi.DevicePlugin_ListAndWatchServer
|
|
type listAndWatchServerStub struct {
|
|
testDM *deviceManager
|
|
t *testing.T
|
|
cdata chan []*pluginapi.Device
|
|
cerr chan error
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) Send(resp *pluginapi.ListAndWatchResponse) error {
|
|
fmt.Println("listAndWatchServerStub::Send", resp.Devices)
|
|
s.cdata <- resp.Devices
|
|
return nil
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) Context() context.Context {
|
|
return nil
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) RecvMsg(m interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) SendMsg(m interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) SendHeader(m metadata.MD) error {
|
|
return nil
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) SetHeader(m metadata.MD) error {
|
|
return nil
|
|
}
|
|
|
|
func (s *listAndWatchServerStub) SetTrailer(m metadata.MD) {
|
|
}
|
|
|
|
func TestListAndWatch(t *testing.T) {
|
|
afuID := "d8424dc4a4a3c413f89e433683f9040b"
|
|
tmpdir := fmt.Sprintf("/tmp/fpgaplugin-TestListAndWatch-%d", time.Now().Unix())
|
|
sysfs := path.Join(tmpdir, "sys", "class", "fpga")
|
|
devfs := path.Join(tmpdir, "dev")
|
|
|
|
tcases := []struct {
|
|
devfsdirs []string
|
|
sysfsdirs []string
|
|
sysfsfiles map[string][]byte
|
|
expectedResult []*pluginapi.Device
|
|
expectedErr bool
|
|
}{
|
|
{
|
|
expectedResult: nil,
|
|
expectedErr: true,
|
|
},
|
|
{
|
|
sysfsdirs: []string{
|
|
"intel-fpga-dev.0/intel-fpga-port.0",
|
|
},
|
|
sysfsfiles: map[string][]byte{
|
|
"intel-fpga-dev.0/intel-fpga-port.0/afu_id": []byte("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa\n"),
|
|
},
|
|
devfsdirs: []string{
|
|
"intel-fpga-port.0", "intel-fpga-fme.0",
|
|
},
|
|
expectedResult: nil,
|
|
expectedErr: true,
|
|
},
|
|
{
|
|
sysfsdirs: []string{
|
|
"intel-fpga-dev.0/intel-fpga-port.0",
|
|
"intel-fpga-dev.0/intel-fpga-fme.0",
|
|
"intel-fpga-dev.1/intel-fpga-port.1",
|
|
"intel-fpga-dev.1/intel-fpga-fme.1",
|
|
"intel-fpga-dev.2/intel-fpga-port.2",
|
|
"intel-fpga-dev.2/intel-fpga-fme.2",
|
|
},
|
|
sysfsfiles: map[string][]byte{
|
|
"intel-fpga-dev.0/intel-fpga-port.0/afu_id": []byte(afuID),
|
|
"intel-fpga-dev.1/intel-fpga-port.1/afu_id": []byte(afuID),
|
|
"intel-fpga-dev.2/intel-fpga-port.2/afu_id": []byte("47595d0fae972fbed0c51b4a41c7a349\n"),
|
|
},
|
|
devfsdirs: []string{
|
|
"intel-fpga-port.0", "intel-fpga-fme.0",
|
|
"intel-fpga-port.1", "intel-fpga-fme.1",
|
|
"intel-fpga-port.2", "intel-fpga-fme.2",
|
|
},
|
|
expectedResult: []*pluginapi.Device{
|
|
&pluginapi.Device{"intel-fpga-dev.0", "Healthy"},
|
|
&pluginapi.Device{"intel-fpga-dev.1", "Healthy"},
|
|
},
|
|
expectedErr: false,
|
|
},
|
|
}
|
|
|
|
resourceName := resourceNamePrefix + "-" + afuID
|
|
testDM := newDeviceManager(resourceName, afuID, tmpdir)
|
|
if testDM == nil {
|
|
t.Fatal("Failed to create a deviceManager")
|
|
}
|
|
|
|
server := &listAndWatchServerStub{
|
|
testDM: testDM,
|
|
t: t,
|
|
cdata: make(chan []*pluginapi.Device),
|
|
cerr: make(chan error),
|
|
}
|
|
|
|
for _, tcase := range tcases {
|
|
err := createTestDirs(devfs, sysfs, tcase.devfsdirs, tcase.sysfsdirs, tcase.sysfsfiles)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
go func() {
|
|
err = testDM.ListAndWatch(&pluginapi.Empty{}, server)
|
|
if err != nil {
|
|
server.cerr <- err
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case result := <-server.cdata:
|
|
if tcase.expectedErr {
|
|
t.Error("Expected error hasn't been triggered")
|
|
} else if tcase.expectedResult != nil && !reflect.DeepEqual(result, tcase.expectedResult) {
|
|
t.Logf("Expected result: %+v", tcase.expectedResult)
|
|
t.Logf("Actual result: %+v", result)
|
|
t.Error("Got unexpeced result")
|
|
}
|
|
testDM.srv.Stop()
|
|
case err = <-server.cerr:
|
|
if !tcase.expectedErr {
|
|
t.Errorf("Unexpected error has been triggered: %+v", err)
|
|
}
|
|
}
|
|
|
|
err = os.RemoveAll(tmpdir)
|
|
if err != nil {
|
|
t.Fatalf("Failed to remove fake device directory: %+v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAllocate(t *testing.T) {
|
|
testDM := newDeviceManager("", "", "")
|
|
if testDM == nil {
|
|
t.Fatal("Failed to create a deviceManager")
|
|
}
|
|
|
|
rqt := &pluginapi.AllocateRequest{
|
|
ContainerRequests: []*pluginapi.ContainerAllocateRequest{
|
|
&pluginapi.ContainerAllocateRequest{
|
|
DevicesIDs: []string{"dev1"},
|
|
},
|
|
},
|
|
}
|
|
|
|
testDM.devices["dev1"] = deviceplugin.DeviceInfo{pluginapi.Healthy, []string{"/dev/dev1"}}
|
|
resp, err := testDM.Allocate(nil, rqt)
|
|
if err != nil {
|
|
t.Fatalf("Failed to allocate healthy device: %+v", err)
|
|
}
|
|
|
|
if len(resp.ContainerResponses[0].Devices) != 1 {
|
|
t.Fatal("Allocated wrong number of devices")
|
|
}
|
|
}
|