Function TestGPUs

xunix/gpu_test.go:46–133

(t *testing.T)

}

func TestGPUs(t *testing.T) {
	t.Parallel()

	t.Run("OK", func(t *testing.T) {
		t.Parallel()

		var (
			fs               = &xunixfake.MemFS{MemMapFs: &afero.MemMapFs{}}
			mounter          = &mount.FakeMounter{}
			log              = slogtest.Make(t, nil)
			usrLibMountpoint = "/var/coder/usr/lib"
			// expectedUsrLibFiles are files that we expect to be returned bind mounts
			// for.
			expectedUsrLibFiles = []string{
				filepath.Join(usrLibMountpoint, "nvidia", "libglxserver_nvidia.so"),
				filepath.Join(usrLibMountpoint, "libnvidia-ml.so"),
				filepath.Join(usrLibMountpoint, "nvidia", "libglxserver_nvidia.so.1"),
			}

			// fakeUsrLibFiles are files that we do not expect to be returned
			// bind mounts for.
			fakeUsrLibFiles = []string{
				filepath.Join(usrLibMountpoint, "libcurl-gnutls.so"),
				filepath.Join(usrLibMountpoint, "libglib.so"),
			}

			// allUsrLibFiles are all the files that should be written to the
			// "mounted" /usr/lib directory. It includes files that shouldn't
			// be returned.
			allUsrLibFiles = append(expectedUsrLibFiles, fakeUsrLibFiles...)
		)

		ctx := xunix.WithFS(context.Background(), fs)
		ctx = xunix.WithMounter(ctx, mounter)

		mounter.MountPoints = []mount.MountPoint{
			{
				Device: "/dev/sda1",
				Path:   "/usr/local/nvidia",
				Opts:   []string{"ro"},
			},
			{
				Device: "/dev/sda2",
				Path:   "/etc/hosts",
			},
			{
				Path: "/dev/nvidia0",
			},
			{
				Path: "/dev/nvidia1",
			},
		}

		err := fs.MkdirAll(filepath.Join(usrLibMountpoint, "nvidia"), 0o755)
		require.NoError(t, err)

		for _, file := range allUsrLibFiles {
			_, err = fs.Create(file)