TestBricksList test

Author: giulio93

internal/orchestrator/bricks/bricks_test.go

@@ -252,19 +252,6 @@ func TestGetBrickInstanceVariableDetails(t *testing.T) {
 		expectedConfigVariables []BrickConfigVariable
 		expectedVariableMap     map[string]string
 	}{
-		{
-			name: "variable is present in the map",
-			brick: &bricksindex.Brick{
-				Variables: []bricksindex.BrickVariable{
-					{Name: "VAR1", Description: "desc"},
-				},
-			},
-			userVariables: map[string]string{"VAR1": "value1"},
-			expectedConfigVariables: []BrickConfigVariable{
-				{Name: "VAR1", Value: "value1", Description: "desc", Required: true},
-			},
-			expectedVariableMap: map[string]string{"VAR1": "value1"},
-		},
 		{
 			name: "variable not present in the map",
 			brick: &bricksindex.Brick{

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

@@ -23,3 +23,23 @@ bricks:
   mount_devices_into_container: false
   ports: []
   category: storage
+- id: arduino:image_classification
+  name: Image Classification
+  description: "Brick for image classification using a pre-trained model. It processes\
+    \ images and returns the predicted class label and confidence score.\nBrick is\
+    \ designed to work with pre-trained models provided by framework or with custom\
+    \ image classification models trained on Edge Impulse platform. \n"
+  require_container: true
+  require_model: true
+  require_devices: false
+  mount_devices_into_container: false
+  ports: []
+  category: video
+  model_name: mobilenet-image-classification
+  variables:
+  - name: CUSTOM_MODEL_PATH
+    default_value: /home/arduino/.arduino-bricks/ei-models
+    description: path to the custom model directory
+  - name: EI_CLASSIFICATION_MODEL
+    default_value: /models/ootb/ei/mobilenet-v2-224px.eim
+    description: path to the model file