77
88from django .contrib .auth .models import User
99from django .core .cache import cache
10+ from django .conf import settings
1011from rest_framework .authtoken .models import Token
1112
1213from api .models import Domain
@@ -66,7 +67,9 @@ def test_strip_dot(self):
6667 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
6768 response = self .client .get (url )
6869 expected = [data ['domain' ] for data in response .data ['results' ]]
69- self .assertEqual (sorted ([self .app_id , domain ]), expected , msg )
70+ self .assertEqual (
71+ sorted (["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN ), domain ]),
72+ expected , msg )
7073
7174 def test_manage_idn_domain (self ):
7275 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
@@ -102,7 +105,9 @@ def test_manage_idn_domain(self):
102105 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
103106 response = self .client .get (url )
104107 expected = [data ['domain' ] for data in response .data ['results' ]]
105- self .assertEqual (sorted ([self .app_id , ace_domain ]), sorted (expected ), msg )
108+ self .assertEqual (
109+ sorted (["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN ), ace_domain ]),
110+ sorted (expected ), msg )
106111
107112 # Verify creation failure for same domain with different encoding
108113 if ace_domain != domain :
@@ -127,7 +132,9 @@ def test_manage_idn_domain(self):
127132
128133 # verify only app domain is left
129134 expected = [data ['domain' ] for data in response .data ['results' ]]
130- self .assertEqual ([self .app_id ], expected , msg )
135+ self .assertEqual (
136+ ["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN )],
137+ expected , msg )
131138
132139 # Use different encoding for creating and deleting (ACE)
133140 if ace_domain != domain :
@@ -139,7 +146,9 @@ def test_manage_idn_domain(self):
139146 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
140147 response = self .client .get (url )
141148 expected = [data ['domain' ] for data in response .data ['results' ]]
142- self .assertEqual (sorted ([self .app_id , ace_domain ]), sorted (expected ), msg )
149+ self .assertEqual (
150+ sorted (["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN ), ace_domain ]),
151+ sorted (expected ), msg )
143152
144153 # Delete
145154 url = '/v2/apps/{app_id}/domains/{hostname}' .format (hostname = ace_domain ,
@@ -154,7 +163,9 @@ def test_manage_idn_domain(self):
154163
155164 # verify only app domain is left
156165 expected = [data ['domain' ] for data in response .data ['results' ]]
157- self .assertEqual ([self .app_id ], expected , msg )
166+ self .assertEqual (
167+ ["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN )],
168+ expected , msg )
158169
159170 # Use different encoding for creating and deleting (Unicode)
160171 if unicode_domain != domain :
@@ -166,7 +177,9 @@ def test_manage_idn_domain(self):
166177 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
167178 response = self .client .get (url )
168179 expected = [data ['domain' ] for data in response .data ['results' ]]
169- self .assertEqual (sorted ([self .app_id , ace_domain ]), sorted (expected ), msg )
180+ self .assertEqual (
181+ sorted (["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN ), ace_domain ]),
182+ sorted (expected ), msg )
170183
171184 # Delete
172185 url = '/v2/apps/{app_id}/domains/{hostname}' .format (hostname = unicode_domain ,
@@ -181,7 +194,9 @@ def test_manage_idn_domain(self):
181194
182195 # verify only app domain is left
183196 expected = [data ['domain' ] for data in response .data ['results' ]]
184- self .assertEqual ([self .app_id ], expected , msg )
197+ self .assertEqual (
198+ ["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN )],
199+ expected , msg )
185200
186201 def test_manage_domain (self ):
187202 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
@@ -211,7 +226,9 @@ def test_manage_domain(self):
211226 url = '/v2/apps/{app_id}/domains' .format (app_id = self .app_id )
212227 response = self .client .get (url )
213228 expected = [data ['domain' ] for data in response .data ['results' ]]
214- self .assertEqual (sorted ([self .app_id , domain ]), sorted (expected ), msg )
229+ self .assertEqual (
230+ sorted (["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN ), domain ]),
231+ sorted (expected ), msg )
215232
216233 # Delete
217234 url = '/v2/apps/{app_id}/domains/{hostname}' .format (hostname = domain ,
@@ -226,7 +243,9 @@ def test_manage_domain(self):
226243
227244 # verify only app domain is left
228245 expected = [data ['domain' ] for data in response .data ['results' ]]
229- self .assertEqual ([self .app_id ], expected , msg )
246+ self .assertEqual (
247+ ["%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN )],
248+ expected , msg )
230249
231250 def test_delete_domain_does_not_exist (self ):
232251 """Remove a domain that does not exist"""
@@ -253,6 +272,13 @@ def test_delete_domain_does_not_remove_latest(self):
253272 with self .assertRaises (Domain .DoesNotExist ):
254273 Domain .objects .get (domain = test_domains [0 ])
255274
275+ def test_delete_domain_does_not_remove_default (self ):
276+ domain = "%s.%s" % (self .app_id , settings .PLATFORM_DOMAIN )
277+ url = '/v2/apps/{app_id}/domains/{domain}' .format (domain = domain ,
278+ app_id = self .app_id )
279+ response = self .client .delete (url )
280+ self .assertEqual (response .status_code , 403 , response .data )
281+
256282 def test_delete_domain_does_not_remove_others (self ):
257283 """https://github.com/drycc/drycc/issues/3475"""
258284 self .test_delete_domain_does_not_remove_latest ()
0 commit comments