Skip to content

Commit f511704

Browse files
committed
add RequireModel to test
1 parent c44f649 commit f511704

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

internal/orchestrator/bricks/bricks_test.go

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -503,12 +503,14 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
503503
{Name: "EI_OBJ_DETECTION_MODEL", DefaultValue: "default_path", Description: "path to the model file"},
504504
{Name: "CUSTOM_MODEL_PATH", DefaultValue: "/home/arduino/.arduino-bricks/ei-models", Description: "path to the custom model directory"},
505505
},
506+
RequireModel: true,
506507
},
507508
{
508-
ID: "arduino:weather_forecast",
509-
Name: "Weather Forecast",
510-
Category: "miscellaneous",
511-
ModelName: "",
509+
ID: "arduino:weather_forecast",
510+
Name: "Weather Forecast",
511+
Category: "miscellaneous",
512+
ModelName: "",
513+
RequireModel: false,
512514
},
513515
},
514516
}
@@ -577,6 +579,7 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
577579
require.Equal(t, "installed", res.Status)
578580
require.Empty(t, res.ModelID)
579581
require.Empty(t, res.CompatibleModels)
582+
require.False(t, res.RequireModel)
580583
},
581584
},
582585
{
@@ -597,6 +600,7 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
597600
require.Len(t, res.CompatibleModels, 2)
598601
require.Equal(t, "yolox-object-detection", res.CompatibleModels[0].ID)
599602
require.Equal(t, "face-detection", res.CompatibleModels[1].ID)
603+
require.True(t, res.RequireModel)
600604
},
601605
},
602606
{
@@ -618,6 +622,7 @@ func TestAppBrickInstanceModelsDetails(t *testing.T) {
618622
require.Len(t, res.CompatibleModels, 2)
619623
require.Equal(t, "yolox-object-detection", res.CompatibleModels[0].ID)
620624
require.Equal(t, "face-detection", res.CompatibleModels[1].ID)
625+
require.True(t, res.RequireModel)
621626
},
622627
},
623628
}

0 commit comments

Comments
 (0)