Function TestMigStrategyMixed

cmd/gpu-feature-discovery/mig_test.go:285–354

(t *testing.T)

}

func TestMigStrategyMixed(t *testing.T) {
	// create VGPU mock library with empty vgpu devices
	vgpuMock := NewTestVGPUMock()
	devices := []resource.Device{
		rt.NewMigEnabledDevice(
			rt.NewMigDevice(3, 0, 20),
			rt.NewMigDevice(1, 0, 5),
		),
	}

	nvmlMock := rt.NewManagerMockWithDevices(devices...)

	conf := &spec.Config{
		Flags: spec.Flags{
			CommandLineFlags: spec.CommandLineFlags{
				MigStrategy:     ptr("mixed"),
				FailOnInitError: ptr(true),
				GFD: &spec.GFDCommandLineFlags{
					Oneshot:         ptr(true),
					OutputFile:      ptr("./gfd-test-mig-mixed"),
					SleepInterval:   ptr(spec.Duration(time.Second)),
					NoTimestamp:     ptr(false),
					MachineTypeFile: ptr(testMachineTypeFile),
				},
			},
		},
	}

	setupMachineFile(t)
	defer removeMachineFile(t)

	labelOutputer, err := lm.NewOutputer(conf, flags.NodeConfig{}, flags.ClientSets{})
	require.NoError(t, err)

	d := gfd{
		manager:       nvmlMock,
		vgpu:          vgpuMock,
		config:        conf,
		labelOutputer: labelOutputer,
	}
	restart, err := d.run(nil)
	require.NoError(t, err, "Error from run function")
	require.False(t, restart)

	outFile, err := os.Open(*conf.Flags.GFD.OutputFile)
	require.NoError(t, err, "Opening output file")

	defer func() {
		err = outFile.Close()
		require.NoError(t, err, "Closing output file")
		err = os.Remove(*conf.Flags.GFD.OutputFile)
		require.NoError(t, err, "Removing output file")
	}()

	output, err := io.ReadAll(outFile)
	require.NoError(t, err, "Reading output file")

	err = checkResult(output, cfg.Path("tests/expected-output-mig-mixed.txt"), false)