From fb484f1779bcdee07f9c648617ff914c08a0d0ee Mon Sep 17 00:00:00 2001
From: gshaibi <39763067+gshaibi@users.noreply.github.com>
Date: Mon, 29 Jan 2024 14:42:55 +0200
Subject: [PATCH] RUN-15689 Tests fixes (#62)

---
 internal/status-exporter/app_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/internal/status-exporter/app_test.go b/internal/status-exporter/app_test.go
index 327afd9..b6de76a 100644
--- a/internal/status-exporter/app_test.go
+++ b/internal/status-exporter/app_test.go
@@ -186,6 +186,7 @@ func getTestCases() map[string]testCase {
 				"nvidia.com/gpu.count":    "1",
 				"nvidia.com/mig.strategy": "mixed",
 				"nvidia.com/gpu.product":  "Tesla P100",
+				"run.ai/fake.gpu":         "true",
 			},
 			expectedMetrics: []*dto.MetricFamily{
 				{
@@ -310,6 +311,7 @@ func getTestCases() map[string]testCase {
 				"nvidia.com/gpu.count":    "2",
 				"nvidia.com/mig.strategy": "mixed",
 				"nvidia.com/gpu.product":  "Tesla P100",
+				"run.ai/fake.gpu":         "true",
 			},
 			expectedMetrics: []*dto.MetricFamily{
 				{