add test IC

Author: giulio93

internal/orchestrator/bricksindex/bricks_index_test.go

@@ -27,31 +27,28 @@ func TestGenerateBricksIndexFromFile(t *testing.T) {
 	index, err := Load(paths.New("testdata"))
 	require.NoError(t, err)
 
-	// Check if ports are correctly set
-	b, found := index.FindBrickByID("arduino:web_ui")
-	require.True(t, found)
-	require.Equal(t, []string{"7000"}, b.Ports)
-
 	// Check if variables are correctly set
 	bWebUI, found := index.FindBrickByID("arduino:web_ui")
 	require.True(t, found)
 	require.Equal(t, []string{"7000"}, bWebUI.Ports)
 
-	// Check if variables are correctly set
-	bWebUI, found = index.FindBrickByID("arduino:image_classification")
+	bIC, found := index.FindBrickByID("arduino:image_classification")
+	require.True(t, found)
+	require.Equal(t, "Image Classification", bIC.Name)
+	require.Equal(t, "mobilenet-image-classification", bIC.ModelName)
+	require.Len(t, bIC.Variables, 2)
+	require.Equal(t, "CUSTOM_MODEL_PATH", bIC.Variables[0].Name)
+	require.Equal(t, "/opt/models/ei/", bIC.Variables[0].DefaultValue)
+	require.Equal(t, "path to the custom model directory", bIC.Variables[0].Description)
+	require.Equal(t, "EI_CLASSIFICATION_MODEL", bIC.Variables[1].Name)
+	require.Equal(t, "/models/ootb/ei/mobilenet-v2-224px.eim", bIC.Variables[1].DefaultValue)
+	require.Equal(t, "path to the model file", bIC.Variables[1].Description)
+	require.False(t, bIC.Variables[0].IsRequired())
+	require.False(t, bIC.Variables[1].IsRequired())
+
+	bRequireModel, found := index.FindBrickByID("arduino:model_required")
 	require.True(t, found)
-	require.Equal(t, "Image Classification", bWebUI.Name)
-	require.Equal(t, "mobilenet-image-classification", bWebUI.ModelName)
-	require.True(t, bWebUI.RequireModel)
-	require.Len(t, bWebUI.Variables, 2)
-	require.Equal(t, "CUSTOM_MODEL_PATH", bWebUI.Variables[0].Name)
-	require.Equal(t, "/opt/models/ei/", bWebUI.Variables[0].DefaultValue)
-	require.Equal(t, "path to the custom model directory", bWebUI.Variables[0].Description)
-	require.Equal(t, "EI_CLASSIFICATION_MODEL", bWebUI.Variables[1].Name)
-	require.Equal(t, "/models/ootb/ei/mobilenet-v2-224px.eim", bWebUI.Variables[1].DefaultValue)
-	require.Equal(t, "path to the model file", bWebUI.Variables[1].Description)
-	require.False(t, bWebUI.Variables[0].IsRequired())
-	require.False(t, bWebUI.Variables[1].IsRequired())
+	require.True(t, bRequireModel.RequireModel)
 
 	bDb, found := index.FindBrickByID("arduino:dbstorage_tsstore")
 	require.True(t, found)

internal/orchestrator/bricksindex/testdata/bricks-list.yaml

@@ -16,6 +16,10 @@ bricks:
   - name: EI_CLASSIFICATION_MODEL
     default_value: /models/ootb/ei/mobilenet-v2-224px.eim
     description: path to the model file
+- id: arduino:model_required
+  name: Model Required Brick
+  description: A brick that requires a model
+  require_model: true
 - id: arduino:camera_scanner
   name: Camera Scanner
   description: Scans a camera for barcodes and QR codes