Merge "Allow empty reseller prefixes in domain_remap"

This commit is contained in:
Jenkins 2014-08-31 04:29:41 +00:00 committed by Gerrit Code Review
commit c5013783ac
2 changed files with 21 additions and 12 deletions

View File

@ -98,6 +98,7 @@ class DomainRemapMiddleware(object):
body='Bad domain in host header', body='Bad domain in host header',
content_type='text/plain') content_type='text/plain')
return resp(env, start_response) return resp(env, start_response)
if len(self.reseller_prefixes) > 0:
if '_' not in account and '-' in account: if '_' not in account and '-' in account:
account = account.replace('-', '_', 1) account = account.replace('-', '_', 1)
account_reseller_prefix = account.split('_', 1)[0].lower() account_reseller_prefix = account.split('_', 1)[0].lower()

View File

@ -130,6 +130,14 @@ class TestDomainRemap(unittest.TestCase):
resp = self.app(req.environ, start_response) resp = self.app(req.environ, start_response)
self.assertEquals(resp, '/test') self.assertEquals(resp, '/test')
def test_domain_remap_configured_with_no_prefixes(self):
conf = {'reseller_prefixes': ''}
self.app = domain_remap.DomainRemapMiddleware(FakeApp(), conf)
req = Request.blank('/test', environ={'REQUEST_METHOD': 'GET'},
headers={'Host': 'c.uuid.example.com'})
resp = self.app(req.environ, start_response)
self.assertEquals(resp, '/v1/uuid/c/test')
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()