TestBricksList test

Author: giulio93

internal/orchestrator/bricks/bricks_test.go

@@ -119,19 +119,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{
@@ -190,3 +177,26 @@ func TestGetBrickInstanceVariableDetails(t *testing.T) {
 		})
 	}
 }
+
+func TestBricksList(t *testing.T) {
+	bricksIndex, err := bricksindex.GenerateBricksIndexFromFile(paths.New("testdata"))
+	require.Nil(t, err)
+
+	b_db, found := bricksIndex.FindBrickByID("arduino:dbstorage_sqlstore")
+	require.True(t, found)
+	require.Equal(t, "Database - SQL", b_db.Name)
+	require.False(t, b_db.RequireModel)
+
+	b_ic, found := bricksIndex.FindBrickByID("arduino:image_classification")
+	require.True(t, found)
+	require.Equal(t, "Image Classification", b_ic.Name)
+	require.Equal(t, "mobilenet-image-classification", b_ic.ModelName)
+	require.True(t, b_ic.RequireModel)
+	require.Len(t, b_ic.Variables, 2)
+	require.Equal(t, "CUSTOM_MODEL_PATH", b_ic.Variables[0].Name)
+	require.Equal(t, "/home/arduino/.arduino-bricks/ei-models", b_ic.Variables[0].DefaultValue)
+	require.Equal(t, "path to the custom model directory", b_ic.Variables[0].Description)
+	require.Equal(t, "EI_CLASSIFICATION_MODEL", b_ic.Variables[1].Name)
+	require.Equal(t, "/models/ootb/ei/mobilenet-v2-224px.eim", b_ic.Variables[1].DefaultValue)
+	require.Equal(t, "path to the model file", b_ic.Variables[1].Description)
+}

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