add tests on reqmodel

Author: giulio93

internal/api/docs/openapi.yaml

@@ -1195,6 +1195,8 @@ components:
           type: string
         name:
           type: string
+        require_model:
+          type: boolean
       required:
       - id
       - name
@@ -1318,6 +1320,8 @@ components:
           type: string
         readme:
           type: string
+        require_model:
+          type: boolean
         status:
           type: string
         used_by_apps:
@@ -1346,6 +1350,8 @@ components:
           type: string
         name:
           type: string
+        require_model:
+          type: boolean
         status:
           type: string
         variables:
@@ -1372,6 +1378,8 @@ components:
           type: array
         name:
           type: string
+        require_model:
+          type: boolean
         status:
           type: string
       type: object

internal/e2e/client/client.gen.go

@@ -783,9 +783,10 @@ func TestAppDetails(t *testing.T) {
 		require.Len(t, *detailsResp.JSON200.Bricks, 1)
 		require.Equal(t,
 			client.AppDetailedBrick{
-				Id:       ImageClassifactionBrickID,
-				Name:     "Image Classification",
-				Category: f.Ptr("video"),
+				Id:           ImageClassifactionBrickID,
+				Name:         "Image Classification",
+				Category:     f.Ptr("video"),
+				RequireModel: f.Ptr(true),
 			},
 			(*detailsResp.JSON200.Bricks)[0],
 		)

internal/e2e/daemon/app_test.go

@@ -83,6 +83,7 @@ func TestBricksList(t *testing.T) {
 		require.Equal(t, bIdx.Description, *brick.Description)
 		require.Equal(t, "Arduino", *brick.Author)
 		require.Equal(t, "installed", *brick.Status)
+		require.Equal(t, bIdx.ModelRequired, *brick.RequireModel)
 	}
 }
 

internal/e2e/daemon/brick_test.go

@@ -156,31 +156,44 @@ func TestBricksIndex(t *testing.T) {
   - name: EI_V_ANOMALY_DETECTION_MODEL
     default_value: /models/ootb/ei/concrete-crack-anomaly-detection.eim
     description: path to the model file
+- id: arduino:fake_no_model
+  name: Camera Scanner
+  description: Scans a camera for barcodes and QR codes
+  require_container: false
+  ports: []
 `
 
 	var index BricksIndex
 	err := yaml.Unmarshal([]byte(x), &index)
 	require.NoError(t, err)
-	require.Len(t, index.Bricks, 11)
+	require.Len(t, index.Bricks, 12)
 
 	// Check if ports are correctly set
-	b, found := index.FindBrickByID("arduino:web_ui")
+	b_ic, found := index.FindBrickByID("arduino:web_ui")
 	require.True(t, found)
-	require.Equal(t, []string{"7000"}, b.Ports)
+	require.Equal(t, []string{"7000"}, b_ic.Ports)
 
 	// Check if variables are correctly set
-	b, found = index.FindBrickByID("arduino:image_classification")
+	b_ic, found = index.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.ModelRequired)
+	require.Len(t, b_ic.Variables, 2)
+	require.Equal(t, "CUSTOM_MODEL_PATH", b_ic.Variables[0].Name)
+	require.Equal(t, "/opt/models/ei/", 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)
+	require.False(t, b_ic.Variables[0].IsRequired())
+	require.False(t, b_ic.Variables[1].IsRequired())
+
+	b_db, found := index.FindBrickByID("arduino:dbstorage_tsstore")
+	require.True(t, found)
+	require.False(t, b_db.ModelRequired)
+
+	b_fake, found := index.FindBrickByID("arduino:fake_no_model")
 	require.True(t, found)
-	require.Equal(t, "Image Classification", b.Name)
-	require.Equal(t, "mobilenet-image-classification", b.ModelName)
-	require.True(t, b.ModelRequired)
-	require.Len(t, b.Variables, 2)
-	require.Equal(t, "CUSTOM_MODEL_PATH", b.Variables[0].Name)
-	require.Equal(t, "/opt/models/ei/", b.Variables[0].DefaultValue)
-	require.Equal(t, "path to the custom model directory", b.Variables[0].Description)
-	require.Equal(t, "EI_CLASSIFICATION_MODEL", b.Variables[1].Name)
-	require.Equal(t, "/models/ootb/ei/mobilenet-v2-224px.eim", b.Variables[1].DefaultValue)
-	require.Equal(t, "path to the model file", b.Variables[1].Description)
-	require.False(t, b.Variables[0].IsRequired())
-	require.False(t, b.Variables[1].IsRequired())
+	require.False(t, b_fake.ModelRequired)
 }