TestAppBrickInstanceDetails

Author: giulio93

internal/orchestrator/bricks/bricks_test.go

@@ -200,3 +200,43 @@ func TestBricksList(t *testing.T) {
 	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)
 }
+
+func TestAppBrickInstanceDetails(t *testing.T) {
+	bricksIndex, err := bricksindex.GenerateBricksIndexFromFile(paths.New("testdata"))
+	require.Nil(t, err)
+	brickService := NewService(nil, bricksIndex, nil)
+	tempDirectory := t.TempDir()
+
+	app := app.ArduinoApp{
+		Name: "TestApp",
+		Descriptor: app.AppDescriptor{
+			Bricks: []app.Brick{
+				{
+					ID: "arduino:image_classification",
+				},
+			},
+		},
+		FullPath: paths.New(tempDirectory),
+	}
+	configVariables, _ := brickService.AppBrickInstanceDetails(&app, bricksIndex.Bricks[2].ID)
+
+	expectedConfigVariables := BrickInstance{
+		Name:     "Image Classification",
+		ID:       "arduino:image_classification",
+		Author:   "Arduino",
+		Category: "video",
+		Status:   "installed",
+		Variables: map[string]string{
+			"CUSTOM_MODEL_PATH":       "/home/arduino/.arduino-bricks/ei-models",
+			"EI_CLASSIFICATION_MODEL": "/models/ootb/ei/mobilenet-v2-224px.eim",
+		},
+		ConfigVariables: []BrickConfigVariable{
+			{Name: "CUSTOM_MODEL_PATH", Value: "/home/arduino/.arduino-bricks/ei-models", Description: "path to the custom model directory", Required: false},
+			{Name: "EI_CLASSIFICATION_MODEL", Value: "/models/ootb/ei/mobilenet-v2-224px.eim", Description: "path to the model file", Required: false},
+		},
+		RequireModel: true,
+		ModelID:      "mobilenet-image-classification",
+	}
+
+	require.Equal(t, expectedConfigVariables, configVariables)
+}