@@ -37,21 +37,6 @@ const (
3737)
3838
3939var (
40- expectedConfigVariables = []client.BrickConfigVariable {
41- {
42- Description : f .Ptr ("path to the custom model directory" ),
43- Name : f .Ptr ("CUSTOM_MODEL_PATH" ),
44- Required : f .Ptr (false ),
45- Value : f .Ptr ("/home/arduino/.arduino-bricks/ei-models" ),
46- },
47- {
48- Description : f .Ptr ("path to the model file" ),
49- Name : f .Ptr ("EI_CLASSIFICATION_MODEL" ),
50- Required : f .Ptr (false ),
51- Value : f .Ptr ("/models/ootb/ei/mobilenet-v2-224px.eim" ),
52- },
53- }
54-
5540 expectedModelInfo = []client.AIModel {
5641 {
5742 Id : f .Ptr ("mobilenet-image-classification" ),
@@ -97,21 +82,15 @@ func TestGetAppBrickInstances(t *testing.T) {
9782 var actualBody models.ErrorResponse
9883 createResp , httpClient := setupTestApp (t )
9984 t .Run ("GetAppBrickInstances_Success" , func (t * testing.T ) {
100- expectedVariables := map [string ]string {
101- "CUSTOM_MODEL_PATH" : "/home/arduino/.arduino-bricks/ei-models" ,
102- "EI_CLASSIFICATION_MODEL" : "/models/ootb/ei/mobilenet-v2-224px.eim" ,
103- }
104-
10585 brickInstances , err := httpClient .GetAppBrickInstancesWithResponse (t .Context (), * createResp .JSON201 .Id , func (ctx context.Context , req * http.Request ) error { return nil })
10686 require .NoError (t , err )
10787 require .Len (t , * brickInstances .JSON200 .Bricks , 1 )
10888 require .Equal (t , ImageClassifactionBrickID , * (* brickInstances .JSON200 .Bricks )[0 ].Id )
109- require .Equal (t , expectedConfigVariables , * (* brickInstances .JSON200 .Bricks )[0 ].ConfigVariables )
89+ require .Nil (t , (* brickInstances .JSON200 .Bricks )[0 ].ConfigVariables )
11090 require .Equal (t , "Arduino" , * (* brickInstances .JSON200 .Bricks )[0 ].Author )
11191 require .Equal (t , "video" , * (* brickInstances .JSON200 .Bricks )[0 ].Category )
11292 require .True (t , * (* brickInstances .JSON200 .Bricks )[0 ].RequireModel )
113- require .Equal (t , expectedVariables , * (* brickInstances .JSON200 .Bricks )[0 ].Variables )
114-
93+ require .Nil (t , (* brickInstances .JSON200 .Bricks )[0 ].Variables )
11594 })
11695
11796 t .Run ("GetAppBrickInstances_InvalidAppID_Fail" , func (t * testing.T ) {
@@ -154,7 +133,7 @@ func TestGetAppBrickInstanceById(t *testing.T) {
154133 require .NoError (t , err )
155134 require .NotEmpty (t , brickInstance .JSON200 )
156135 require .Equal (t , ImageClassifactionBrickID , * brickInstance .JSON200 .Id )
157- require .Equal (t , expectedConfigVariables , ( * brickInstance .JSON200 .ConfigVariables ) )
136+ require .Nil (t , brickInstance .JSON200 .ConfigVariables )
158137 require .NotNil (t , brickInstance .JSON200 .CompatibleModels )
159138 require .Equal (t , expectedModelInfo , * (brickInstance .JSON200 .CompatibleModels ))
160139 })
@@ -220,7 +199,7 @@ func TestUpsertAppBrickInstance(t *testing.T) {
220199 require .NoError (t , err )
221200 require .NotEmpty (t , brickInstance .JSON200 )
222201 require .Equal (t , ImageClassifactionBrickID , * brickInstance .JSON200 .Id )
223- require .Equal (t , "/home/arduino/.arduino-bricks/ei-models" , ( * brickInstance .JSON200 .Variables )[ "CUSTOM_MODEL_PATH" ] )
202+ require .Nil (t , brickInstance .JSON200 .Variables )
224203 require .Equal (t , "mobilenet-image-classification" , * brickInstance .JSON200 .Model )
225204
226205 t .Run ("OverrideBrickInstance" , func (t * testing.T ) {
@@ -243,7 +222,7 @@ func TestUpsertAppBrickInstance(t *testing.T) {
243222 require .NoError (t , err )
244223 require .NotEmpty (t , brickInstance .JSON200 )
245224 require .Equal (t , ImageClassifactionBrickID , * brickInstance .JSON200 .Id )
246- require .NotEqual (t , "overidden" , ( * brickInstance .JSON200 .Variables )[ "CUSTOM_MODEL_PATH" ] )
225+ require .Nil (t , brickInstance .JSON200 .Variables )
247226 require .Equal (t , "mobilenet-image-classification" , * brickInstance .JSON200 .Model )
248227 })
249228
@@ -369,7 +348,7 @@ func TestUpdateAppBrickInstance(t *testing.T) {
369348 require .NoError (t , err )
370349 require .NotEmpty (t , brickInstance .JSON200 )
371350 require .Equal (t , ImageClassifactionBrickID , * brickInstance .JSON200 .Id )
372- require .Equal (t , "overidden" , ( * brickInstance .JSON200 .Variables )[ "CUSTOM_MODEL_PATH" ] )
351+ require .Nil (t , brickInstance .JSON200 .Variables )
373352 require .Equal (t , "person-classification" , * brickInstance .JSON200 .Model )
374353 })
375354 t .Run ("UpdateOnlyModel" , func (t * testing.T ) {
0 commit comments