Skip to content

Commit af45937

Browse files
committed
add tests on reqmodel
1 parent 834851e commit af45937

File tree

5 files changed

+28
-13
lines changed

5 files changed

+28
-13
lines changed

internal/api/docs/openapi.yaml

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1204,6 +1204,8 @@ components:
12041204
type: string
12051205
name:
12061206
type: string
1207+
require_model:
1208+
type: boolean
12071209
required:
12081210
- id
12091211
- name
@@ -1332,6 +1334,8 @@ components:
13321334
type: string
13331335
readme:
13341336
type: string
1337+
require_model:
1338+
type: boolean
13351339
status:
13361340
type: string
13371341
used_by_apps:
@@ -1365,6 +1369,8 @@ components:
13651369
type: string
13661370
name:
13671371
type: string
1372+
require_model:
1373+
type: boolean
13681374
status:
13691375
type: string
13701376
variables:
@@ -1386,6 +1392,8 @@ components:
13861392
type: string
13871393
name:
13881394
type: string
1395+
require_model:
1396+
type: boolean
13891397
status:
13901398
type: string
13911399
type: object

internal/e2e/client/client.gen.go

Lines changed: 14 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

internal/e2e/daemon/app_test.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -783,9 +783,10 @@ func TestAppDetails(t *testing.T) {
783783
require.Len(t, *detailsResp.JSON200.Bricks, 1)
784784
require.Equal(t,
785785
client.AppDetailedBrick{
786-
Id: ImageClassifactionBrickID,
787-
Name: "Image Classification",
788-
Category: f.Ptr("video"),
786+
Id: ImageClassifactionBrickID,
787+
Name: "Image Classification",
788+
Category: f.Ptr("video"),
789+
RequireModel: f.Ptr(true),
789790
},
790791
(*detailsResp.JSON200.Bricks)[0],
791792
)

internal/e2e/daemon/brick_test.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ func TestBricksList(t *testing.T) {
8383
require.Equal(t, bIdx.Description, *brick.Description)
8484
require.Equal(t, "Arduino", *brick.Author)
8585
require.Equal(t, "installed", *brick.Status)
86+
require.Equal(t, bIdx.ModelRequired, *brick.RequireModel)
8687
}
8788
}
8889

internal/orchestrator/bricksindex/bricks_index_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ func TestGenerateBricksIndexFromFile(t *testing.T) {
3737
require.True(t, found)
3838
require.Equal(t, "Image Classification", b.Name)
3939
require.Equal(t, "mobilenet-image-classification", b.ModelName)
40-
require.True(t, b.ModelRequired)
40+
require.True(t, b.RequireModel)
4141
require.Len(t, b.Variables, 2)
4242
require.Equal(t, "CUSTOM_MODEL_PATH", b.Variables[0].Name)
4343
require.Equal(t, "/opt/models/ei/", b.Variables[0].DefaultValue)

0 commit comments

Comments
 (0)