def test_domain_number_for_url(self):
storage = DomainShardingStorage()
domain_numbers = set([storage.get_domain_number_for_uri('avatars/{0}.jpg'.format(i)) for i in xrange(100)])
self.assertTrue(len(domain_numbers), settings.WEBPERF_DOMAINS_NUMBER)
for i in xrange(1, settings.WEBPERF_DOMAINS_NUMBER+1):
self.assertTrue(i in domain_numbers)
def test_sharded_url(self):
storage = DomainShardingStorage()
urls = [storage.url('images/example{0}.jpg'.format(i)) for i in xrange(100)]
for url in urls:
self.assertRegexpMatches(url, r'http://img\d+\.example\.com/media/images/example\d+\.jpg')
def test_not_configured(self):
storage = DomainShardingStorage()
url = storage.url('images/example.jpg')
self.assertEqual(url, 'images/example.jpg')
