Fix GPU detection test

livepeer · Jun 9, 2022 · ddc9c1e · ddc9c1e
1 parent 71edaeb
commit ddc9c1e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/common/util_test.go b/common/util_test.go
@@ -353,6 +353,22 @@ func TestRatPriceInfo(t *testing.T) {
 	assert.Zero(priceInfo.Cmp(big.NewRat(7, 2)))
 }
 
+func TestParseAccelDevices_FailedDetection(t *testing.T) {
+	assert := assert.New(t)
+
+	getGPU = func() ([]*gpu.GraphicsCard, error) {
+		return []*gpu.GraphicsCard{}, nil
+	}
+	getPCI = func() ([]*pci.Device, error) {
+		return []*pci.Device{}, nil
+	}
+
+	ids, err := ParseAccelDevices("all", ffmpeg.Nvidia)
+
+	assert.NotNil(err)
+	assert.Equal(len(ids), 0)
+}
+
 func TestParseAccessDevices_Gpu(t *testing.T) {
 	assert := assert.New(t)