# -*- coding: utf-8 -*-
"""
(c) 2018 - Copyright Red Hat Inc
Authors:
Clement Verna <cverna@tutanota.com>
"""
from __future__ import unicode_literals, absolute_import
import unittest
import sys
import os
import time
import mock
import pygit2
sys.path.insert(
0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")
)
import pagure.lib.git
import pagure.lib.query
import tests
from tests.test_pagure_lib_git_get_tags_objects import add_repo_tag
class PagureFlaskUiArchivesTest(tests.Modeltests):
""" Tests checking the archiving mechanism. """
def setUp(self):
""" Set up the environnment, ran before every tests. """
super(PagureFlaskUiArchivesTest, self).setUp()
tests.create_projects(self.session)
tests.create_projects_git(os.path.join(self.path, "repos"), bare=True)
project = pagure.lib.query._get_project(self.session, "test")
# test has both commits and tags
repopath = os.path.join(self.path, "repos", "test.git")
tests.add_readme_git_repo(repopath)
repo = pygit2.Repository(repopath)
add_repo_tag(self.path, repo, ["v1.0", "v1.1"], "test.git")
# test2 has only commits
tests.add_readme_git_repo(
os.path.join(self.path, "repos", "test2.git")
)
# somenamespace/test3 has neither commits nor tags
# Create the archive folder:
self.archive_path = os.path.join(self.path, "archives")
os.mkdir(self.archive_path)
def test_project_no_conf(self):
""" Test getting the archive when pagure isn't configured. """
output = self.app.get(
"/somenamespace/test3/archive/tag1/test3-tag1.zip",
follow_redirects=True,
)
self.assertEqual(output.status_code, 404)
self.assertIn(
"This pagure instance isn't configured to support "
"this feature",
output.get_data(as_text=True),
)
self.assertEqual(os.listdir(self.archive_path), [])
def test_project_invalid_conf(self):
""" Test getting the archive when pagure is wrongly configured. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "invalid")},
):
output = self.app.get(
"/somenamespace/test3/archive/tag1/test3-tag1.zip",
follow_redirects=True,
)
self.assertEqual(output.status_code, 500)
self.assertIn(
"Incorrect configuration, please contact your admin",
output.get_data(as_text=True),
)
self.assertEqual(os.listdir(self.archive_path), [])
def test_project_invalid_format(self):
""" Test getting the archive when the format provided is invalid. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/somenamespace/test3/archive/tag1/test3-tag1.unzip",
follow_redirects=True,
)
self.assertEqual(output.status_code, 404)
self.assertEqual(os.listdir(self.archive_path), [])
def test_project_no_commit(self):
""" Test getting the archive of an empty project. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/somenamespace/test3/archive/tag1/test3-tag1.zip",
follow_redirects=True,
)
self.assertEqual(output.status_code, 404)
self.assertIn(
"<p>Invalid commit provided</p>", output.get_data(as_text=True)
)
self.assertEqual(os.listdir(self.archive_path), [])
def test_project_no_tag(self):
""" Test getting the archive of a non-empty project but without
tags. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test2/archive/tag1/test2-tag1.zip", follow_redirects=True
)
self.assertEqual(output.status_code, 404)
self.assertIn(
"<p>Invalid commit provided</p>", output.get_data(as_text=True)
)
self.assertEqual(os.listdir(self.archive_path), [])
def test_project_no_tag(self):
""" Test getting the archive of an empty project. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test2/archive/tag1/test2-tag1.zip", follow_redirects=True
)
self.assertEqual(output.status_code, 404)
self.assertIn(
"<p>Invalid commit provided</p>", output.get_data(as_text=True)
)
self.assertEqual(os.listdir(self.archive_path), [])
def test_project_w_tag_zip(self):
""" Test getting the archive from a tag. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test/archive/v1.0/test-v1.0.zip", follow_redirects=True
)
self.assertEqual(output.status_code, 200)
self.assertEqual(os.listdir(self.archive_path), ["test"])
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test")), ["tags"]
)
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test", "tags")),
["v1.0"],
)
self.assertEqual(
len(
os.listdir(
os.path.join(self.archive_path, "test", "tags", "v1.0")
)
),
1,
)
files = os.listdir(
os.path.join(self.archive_path, "test", "tags", "v1.0")
)
self.assertEqual(
os.listdir(
os.path.join(
self.archive_path, "test", "tags", "v1.0", files[0]
)
),
["test-v1.0.zip"],
)
def test_project_w_tag_tar(self):
""" Test getting the archive from a tag. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test/archive/v1.0/test-v1.0.tar", follow_redirects=True
)
self.assertEqual(output.status_code, 200)
self.assertEqual(os.listdir(self.archive_path), ["test"])
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test")), ["tags"]
)
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test", "tags")),
["v1.0"],
)
self.assertEqual(
len(
os.listdir(
os.path.join(self.archive_path, "test", "tags", "v1.0")
)
),
1,
)
files = os.listdir(
os.path.join(self.archive_path, "test", "tags", "v1.0")
)
self.assertEqual(
os.listdir(
os.path.join(
self.archive_path, "test", "tags", "v1.0", files[0]
)
),
["test-v1.0.tar"],
)
def test_project_w_tag_tar_gz(self):
""" Test getting the archive from a tag. """
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test/archive/v1.0/test-v1.0.tar.gz", follow_redirects=True
)
self.assertEqual(output.status_code, 200)
self.assertEqual(os.listdir(self.archive_path), ["test"])
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test")), ["tags"]
)
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test", "tags")),
["v1.0"],
)
self.assertEqual(
len(
os.listdir(
os.path.join(self.archive_path, "test", "tags", "v1.0")
)
),
1,
)
files = os.listdir(
os.path.join(self.archive_path, "test", "tags", "v1.0")
)
self.assertEqual(
os.listdir(
os.path.join(
self.archive_path, "test", "tags", "v1.0", files[0]
)
),
["test-v1.0.tar.gz"],
)
def test_project_w_commit_tar_gz(self):
""" Test getting the archive from a commit. """
repopath = os.path.join(self.path, "repos", "test.git")
repo = pygit2.Repository(repopath)
commit = repo.head.target.hex
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test/archive/%s/test-v1.0.tar.gz" % commit,
follow_redirects=True,
)
self.assertEqual(output.status_code, 200)
self.assertEqual(os.listdir(self.archive_path), ["test"])
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test")), [commit]
)
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test", commit)),
["test-v1.0.tar.gz"],
)
def test_project_w_commit_tar_gz_twice(self):
""" Test getting the archive from a commit twice, so we hit the
disk cache. """
repopath = os.path.join(self.path, "repos", "test.git")
repo = pygit2.Repository(repopath)
commit = repo.head.target.hex
with mock.patch.dict(
"pagure.config.config",
{"ARCHIVE_FOLDER": os.path.join(self.path, "archives")},
):
output = self.app.get(
"/test/archive/%s/test-v1.0.tar.gz" % commit,
follow_redirects=True,
)
self.assertEqual(output.status_code, 200)
output = self.app.get(
"/test/archive/%s/test-v1.0.tar.gz" % commit,
follow_redirects=True,
)
self.assertEqual(output.status_code, 200)
self.assertEqual(os.listdir(self.archive_path), ["test"])
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test")), [commit]
)
self.assertEqual(
os.listdir(os.path.join(self.archive_path, "test", commit)),
["test-v1.0.tar.gz"],
)
if __name__ == "__main__":
unittest.main(verbosity=2)