def set_up_license(target_file, index=0):
# Get BuildInfo for target file
file_path = os.path.join(TESTSERVER_ROOT, target_file)
build_info = BuildInfo(file_path)
# Insert license information into database
text = build_info.get("license-text", index)
digest = hashlib.md5(text).hexdigest()
theme = build_info.get("theme", index)
_insert_license_into_db(digest, text, theme)
return digest
def test_redirect_to_license_ste(self):
# Get BuildInfo for target file
target_file = "build-info/snowball-blob.txt"
file_path = os.path.join(TESTSERVER_ROOT, target_file)
build_info = BuildInfo(file_path)
# Try to fetch file from server - we should be redirected
url = urlparse.urljoin("http://testserver/", target_file)
response = self.client.get(url, follow=True)
digest = hashlib.md5(build_info.get("license-text")).hexdigest()
self.assertRedirects(response,
'/license?lic=%s&url=%s' % (digest, target_file))
# Make sure that we get the license text in the license page
self.assertContains(response, build_info.get("license-text"))
# Test that we use the "stericsson" theme. This contains igloo.png
self.assertContains(response, "igloo.png")
def test_no_exception_ip(self):
internal_host = '10.1.2.3'
target_file = 'build-info/origen-blob.txt'
file_path = os.path.join(TESTSERVER_ROOT, target_file)
build_info = BuildInfo(file_path)
# Try to fetch file from server - we should be redirected
url = urlparse.urljoin("http://testserver/", target_file)
response = self.client.get(url, follow=True, REMOTE_ADDR=internal_host)
digest = hashlib.md5(build_info.get("license-text")).hexdigest()
self.assertRedirects(response,
'/license?lic=%s&url=%s' % (digest, target_file))
# Make sure that we get the license text in the license page
self.assertContains(response, build_info.get("license-text"))
# Test that we use the "samsung" theme. This contains exynos.png
self.assertContains(response, "exynos.png")
def test_partial_build_info_file_protected(self):
target_file = ("partial-license-settings/"
"partially-complete-build-info/"
"should_be_protected.txt")
file_path = os.path.join(TESTSERVER_ROOT, target_file)
build_info = BuildInfo(file_path)
# Try to fetch file from server - we should be redirected
url = urlparse.urljoin("http://testserver/", target_file)
response = self.client.get(url, follow=True)
digest = hashlib.md5(build_info.get("license-text")).hexdigest()
self.assertRedirects(response,
'/license?lic=%s&url=%s' % (digest, target_file))