@@ -23,7 +23,7 @@ def test_retrieve_view(self):
23
23
pk = self .attachment .pk
24
24
data = {
25
25
'url' : 'http://testserver/api/v1/media/%s' % pk ,
26
- 'download_url' : 'http://testserver/api/v1/media/%s.jpg' % pk ,
26
+ 'download_url' : self . attachment . media_file . url ,
27
27
'id' : pk ,
28
28
'xform' : self .xform .pk ,
29
29
'instance' : self .attachment .instance .pk ,
@@ -104,9 +104,7 @@ def test_direct_image_link(self):
104
104
response = self .retrieve_view (request , pk = self .attachment .pk )
105
105
self .assertEqual (response .status_code , 200 )
106
106
self .assertTrue (isinstance (response .data , basestring ))
107
- self .assertEqual (response .data ,
108
- 'http://testserver/api/v1/media/%s.jpg'
109
- % self .attachment .pk )
107
+ self .assertEqual (response .data , self .attachment .media_file .url )
110
108
111
109
data ['filename' ] = 10000000
112
110
request = self .factory .get ('/' , data , ** self .extra )
@@ -131,6 +129,4 @@ def test_direct_image_link_uppercase(self):
131
129
response = self .retrieve_view (request , pk = self .attachment .pk )
132
130
self .assertEqual (response .status_code , 200 )
133
131
self .assertTrue (isinstance (response .data , basestring ))
134
- self .assertEqual (response .data ,
135
- 'http://testserver/api/v1/media/%s.jpg'
136
- % self .attachment .pk )
132
+ self .assertEqual (response .data , self .attachment .media_file .url )
0 commit comments