Skip to content

Commit 5c2e119

Browse files
committed
update unit tests
1 parent 85c6245 commit 5c2e119

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

internal/orchestrator/bricks/bricks_test.go

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -402,6 +402,21 @@ func TestBricksDetails(t *testing.T) {
402402
})
403403

404404
t.Run("Success - Full Details - multiple models", func(t *testing.T) {
405+
expectConfigVariables := []BrickConfigVariable{
406+
{
407+
Name: "EI_OBJ_DETECTION_MODEL",
408+
Value: "default_path",
409+
Description: "path to the model file",
410+
Required: false,
411+
},
412+
{
413+
Name: "CUSTOM_MODEL_PATH",
414+
Value: "/home/arduino/.arduino-bricks/ei-models",
415+
Description: "path to the custom model directory",
416+
Required: false,
417+
},
418+
}
419+
405420
res, err := svc.BricksDetails("arduino:object_detection", idProvider, cfg)
406421
require.NoError(t, err)
407422

@@ -425,6 +440,8 @@ func TestBricksDetails(t *testing.T) {
425440
require.Equal(t, "face-detection", res.CompatibleModels[1].ID)
426441
require.Equal(t, "Lightweight-Face-Detection", res.CompatibleModels[1].Name)
427442
require.Equal(t, "", res.CompatibleModels[1].Description)
443+
require.Len(t, res.ConfigVariables, 2)
444+
require.Equal(t, expectConfigVariables, res.ConfigVariables)
428445
})
429446

430447
t.Run("Success - Full Details - no models", func(t *testing.T) {
@@ -444,6 +461,7 @@ func TestBricksDetails(t *testing.T) {
444461
require.Equal(t, "My App", res.UsedByApps[0].Name)
445462
require.NotEmpty(t, res.UsedByApps[0].ID)
446463
require.Len(t, res.CompatibleModels, 0)
464+
require.Empty(t, res.ConfigVariables)
447465
})
448466

449467
t.Run("Success - Full Details - one model", func(t *testing.T) {
@@ -456,6 +474,8 @@ func TestBricksDetails(t *testing.T) {
456474
require.Equal(t, "face-detection", res.CompatibleModels[0].ID)
457475
require.Equal(t, "Lightweight-Face-Detection", res.CompatibleModels[0].Name)
458476
require.Equal(t, "", res.CompatibleModels[0].Description)
477+
require.Empty(t, res.ConfigVariables)
478+
require.Empty(t, res.Variables)
459479
})
460480
}
461481

0 commit comments

Comments
 (0)