@@ -75,7 +75,7 @@ def test_get(self):
75
75
_resp = self .client .get (self ._detail_path )
76
76
self .assertEqual (_resp .status_code , HTTPStatus .OK )
77
77
self .assertEqual (_resp .data ["target_id" ], self ._addon .target_id )
78
- self .assertEqual (_resp .data ["resource_type" ], self ._addon .resource_type )
78
+ self .assertEqual (_resp .data ["resource_type" ], self ._addon .resource_type . name )
79
79
80
80
def test_methods_not_allowed (self ):
81
81
_methods_not_allowed = {
@@ -115,13 +115,15 @@ def test_resource_type_property(self):
115
115
self ._addon .save ()
116
116
117
117
refreshed = db .ConfiguredLinkAddon .objects .get (id = self ._addon .id )
118
- self .assertEqual (refreshed .resource_type , "Book" )
118
+ self .assertEqual (refreshed .resource_type , SupportedResourceTypes .Book )
119
+ self .assertEqual (refreshed .resource_type .name , "Book" )
119
120
120
121
self ._addon .resource_type = SupportedResourceTypes .Other
121
122
self ._addon .save ()
122
123
123
124
refreshed = db .ConfiguredLinkAddon .objects .get (id = self ._addon .id )
124
- self .assertEqual (refreshed .resource_type , "Other" )
125
+ self .assertEqual (refreshed .resource_type , SupportedResourceTypes .Other )
126
+ self .assertEqual (refreshed .resource_type .name , "Other" )
125
127
126
128
def test_validator_valid_types (self ):
127
129
try :
@@ -202,7 +204,9 @@ def test_get(self):
202
204
203
205
with self .subTest ("Confirm expected attributes" ):
204
206
self .assertEqual (_resp .data ["target_id" ], self ._addon .target_id )
205
- self .assertEqual (_resp .data ["resource_type" ], self ._addon .resource_type )
207
+ self .assertEqual (
208
+ _resp .data ["resource_type" ], self ._addon .resource_type .name
209
+ )
206
210
self .assertIn ("connected_operation_names" , _resp .data )
207
211
208
212
with self .subTest ("Confirm expected relationships" ):
@@ -286,7 +290,7 @@ def test_create_addon(self):
286
290
"type" : "configured-link-addons" ,
287
291
"attributes" : {
288
292
"target_id" : "some-target-id" ,
289
- "resource_type" : "DATASET " ,
293
+ "resource_type" : "Other " ,
290
294
"connected_capabilities" : ["ACCESS" ],
291
295
"authorized_resource_uri" : self ._resource .resource_uri ,
292
296
},
@@ -315,8 +319,9 @@ def test_create_addon(self):
315
319
316
320
self .assertEqual (_resp .status_code , HTTPStatus .CREATED )
317
321
318
- self .assertEqual (_resp .data ["resource_type" ], " Other" )
322
+ self .assertEqual (_resp .data ["resource_type" ], SupportedResourceTypes . Other . name )
319
323
320
324
addon = db .ConfiguredLinkAddon .objects .get (id = _resp .data ["id" ])
321
325
self .assertEqual (addon .target_id , "some-target-id" )
322
- self .assertEqual (addon .resource_type , "Other" )
326
+ self .assertEqual (addon .resource_type , SupportedResourceTypes .Other )
327
+ self .assertEqual (addon .resource_type .name , "Other" )
0 commit comments