def create_app_root(dbconn): zodb_root = get_database_root(dbconn) if not '{}_root'.format(APP_NAME) in zodb_root: zodb_root['{}_root'.format(APP_NAME)] = Root() if not 'user' in zodb_root['{}_root'.format(APP_NAME)]: zodb_root['{}_root'.format(APP_NAME)]['user'] = Root() if not 'repository' in zodb_root['{}_root'.format(APP_NAME)]: zodb_root['{}_root'.format(APP_NAME)]['repository'] = Root()
def get_root(self): from papaye.models import Root root = Root() def root_func(): return {'papaye_root': {'repository': {}, 'user': {}}} class P_jar: pass root._p_jar = P_jar() root._p_jar.root = root_func return root
def get_root(self): from papaye.models import Root root = Root() def root_func(): return {'papaye_root': {'repository': {}, 'user': {}}} class P_jar: pass root._p_jar = P_jar() root._p_jar.root = root_func return root
def test_list_releases_files(self): from papaye.views.simple import ListReleaseFileView from papaye.models import Package, Release, Root, ReleaseFile # Test packages root = Root() root['package1'] = Package(name='package1') root['package1']['release1'] = Release(name='release1', version='1.0', metadata={}) root['package1']['release1'].__parent__ = root['package1'] root['package1']['release1']['releasefile1.tar.gz'] = ReleaseFile( filename='releasefile1.tar.gz', content=b'', md5_digest='12345') root['package1']['release1']['releasefile1.tar.gz'].__parent__ = root[ 'package1']['release1'] view = ListReleaseFileView(root['package1'], self.request) response = view() self.assertIsInstance(response, dict) self.assertIn('objects', response) self.assertIsInstance(response['objects'], types.GeneratorType) self.assertEqual([ (url, release) for url, release in response['objects'] ], [( 'http://example.com/simple/package1/release1/releasefile1.tar.gz#md5=12345', root['package1']['release1']['releasefile1.tar.gz'])])
def test_upload_release_already_exists(self): from papaye.models import Root, Package, Release, ReleaseFile from papaye.views.simple import UploadView # Create a fake test file uploaded_file = io.BytesIO(b"content") storage = FieldStorage() storage.filename = 'foo.tar.gz' storage.file = uploaded_file self.request.POST = { "content": storage, "some_metadata": "Fake Metadata", "version": "1.0", "name": "my_package", ":action": "file_upload", } root = Root() # Create initial release package = Package('my_package') package['1.0'] = Release('1.0', '1.0', metadata={}) package['1.0']['foo.tar.gz'] = ReleaseFile('foo.tar.gz', b'') root['my_package'] = package view = UploadView(root, self.request) result = view() self.assertIsInstance(result, Response) self.assertEqual(result.status_int, 409)
def test_build_repository_with_specified_root(self, mock): from papaye.proxy import PyPiProxy from papaye.models import Package, ReleaseFile, Root mock.return_value = self.pypi_response info_dict = json.loads(self.pypi_response.content.decode('utf-8')) root = Root() proxy = PyPiProxy(self.request, 'pyramid') result = proxy.build_repository(with_metadata=True, root=root) self.assertIsInstance(result, Package) for release in result.releases.values(): self.assertIn(release.__name__, info_dict['releases'].keys()) self.assertEqual(len(result.releases.keys()), len(info_dict['releases'].keys())) self.assertEqual(len(result['1.5'].release_files.keys()), 1) self.assertIsInstance(result['1.5'].release_files.values()[0], ReleaseFile) self.assertTrue( getattr(result['1.5'].release_files.values()[0], 'pypi_url', None)) self.assertEqual( result['1.5'].release_files.values()[0].pypi_url, "https://pypi.python.org/packages/source/p/pyramid/pyramid-1.5.tar.gz" ) self.assertIsNotNone(result['1.5'].metadata) assert result.__parent__ is root
def test_get_index(self): from papaye.models import Root, Package root = Root() package = Package(name='package1') root['package'] = package result = root[0] assert result == package
def test_get_package_unknown(setup): from papaye.views.api import get_package from papaye.models import Root request = testing.DummyRequest() root = Root() request.context = root request.matchdict = {'package_name': 'package1'} result = get_package(request) assert isinstance(result, HTTPNotFound) is True
def test_iter(self): from papaye.models import Root, Package root = Root() package = Package(name='package1') root['package'] = package result = iter(root) assert isinstance(result, types.GeneratorType) assert list(result) == [ package, ]
def test_list_packages_without_package(self): from papaye.views.simple import ListPackagesView from papaye.models import Root # Test packages root = Root() view = ListPackagesView(root, self.request) response = view() self.assertIsInstance(response, dict) self.assertIn('objects', response) self.assertIsInstance(response['objects'], types.GeneratorType) self.assertEqual( [package.name for url, package in response['objects']], [])
def test_get_packages_with_no_package_in_database(setup): from papaye.views.api import list_packages from papaye.models import Root request = testing.DummyRequest() root = Root() request.context = root result = list_packages(request) assert isinstance(result, dict) is True assert 'result' in result assert isinstance(result['result'], list) is True assert 'count' in result assert result['count'] == 0 assert len(result['result']) == 0 assert result['result'] == []
def test_get_package_by_version_with_unknown_release(setup): from papaye.views.api import get_package_by_version from papaye.models import Package, Root, Release request = testing.DummyRequest() root = Root() root['package1'] = Package(name='package1') root['package2'] = Package(name='package2') root['package1']['1.0'] = Release('1.0', '1.0', {'summary': 'The package 1'}) root['package2']['1.0'] = Release('1.0', '1.0', {'summary': 'The package 2'}) request.context = root request.matchdict = {'package_name': 'package2', 'version': '2.0'} result = get_package_by_version(request) assert isinstance(result, HTTPNotFound) is True
def test_upload_release_with_spaces(self): from papaye.models import Root, Package, Release, ReleaseFile from papaye.views.simple import UploadView # Create a fake test file uploaded_file = io.BytesIO(b"content") storage = FieldStorage() storage.filename = 'foo.tar.gz' storage.file = uploaded_file self.request.POST = { "content": storage, "some_metadata": "Fake Metadata", "version": "1.0", "name": "my package", ":action": "file_upload", "md5_digest": "Fake MD5" } root = Root() self.request.root = root view = UploadView(root, self.request) result = view() self.assertIsInstance(result, Response) self.assertEqual(result.status_int, 200) self.assertTrue('my-package' in root) self.assertIsInstance(root['my-package'], Package) self.assertTrue(root['my-package'].releases.get('1.0', False)) self.assertIsInstance(root['my-package']['1.0'], Release) self.assertTrue(root['my-package']['1.0'].release_files.get( 'foo.tar.gz', b'')) self.assertIsInstance(root['my-package']['1.0']['foo.tar.gz'], ReleaseFile) self.assertEqual(root['my-package']['1.0']['foo.tar.gz'].md5_digest, "Fake MD5") self.assertIsNotNone(root['my-package']['1.0'].metadata) self.assertIsInstance(root['my-package']['1.0'].metadata, dict) self.assertEqual( root['my-package']['1.0'].release_files.get('foo.tar.gz', b'').size, 7)
def test_list_releases_files_with_new_remotes_release( self, mock, requests_mock): from papaye.views.simple import ListReleaseFileView from papaye.models import Package, Release, Root, ReleaseFile mock.return_value = "3.0" # Test packages root = Root() root['package1'] = Package(name='package1') root['package1']['release1'] = Release(name='release1', version='1.0', metadata={}) root['package1']['release1'].__parent__ = root['package1'] root['package1']['release1']['releasefile1.tar.gz'] = ReleaseFile( filename='releasefile1.tar.gz', content=b'', md5_digest='12345') root['package1']['release1']['releasefile1.tar.gz'].__parent__ = root[ 'package1']['release1'] self.request.matchdict['traverse'] = ( root['package1'].__name__, root['package1']['release1'].__name__) self.request.registry.settings['papaye.proxy'] = 'true' view = ListReleaseFileView(root['package1'], self.request) pypi_result = { 'info': { 'name': 'package1', }, 'releases': { 'release1': [{ 'filename': 'releasefile1.tar.gz', 'url': 'http://example.com/', 'md5_digest': 'fake md5', }] } } requests_mock.return_value = FakeGRequestResponse( 200, bytes(json.dumps(pypi_result), 'utf-8')) response = view() self.assertIsInstance(response, dict)
def test_get_package(setup): from papaye.views.api import get_package from papaye.models import Package, Root, Release request = testing.DummyRequest() root = Root() root['package1'] = Package(name='package1') root['package2'] = Package(name='package2') root['package1']['1.0'] = Release('1.0', '1.0', { 'summary': 'The package 1', 'description': 'A description', }) root['package2']['1.0'] = Release('1.0', '1.0', { 'summary': 'The package 2', 'description': 'A description', }) request.context = root request.matchdict = {'package_name': 'package1'} result = get_package(request) assert isinstance(result, dict) is True assert result['name'] == 'package1' assert 'metadata' in result
def test_get_packages(setup): from papaye.views.api import list_packages from papaye.models import Package, Root, Release request = testing.DummyRequest() root = Root() root['package1'] = Package(name='package1') root['package2'] = Package(name='package2') root['package1']['1.0'] = Release('1.0', '1.0', { 'summary': 'The package 1', 'description': 'A description', }) root['package2']['1.0'] = Release('1.0', '1.0', { 'summary': 'The package 2', 'description': 'A description', }) expected = [ { 'name': 'package1', 'summary': 'The package 1', }, { 'name': 'package2', 'summary': 'The package 2', }, ] request.context = root result = list_packages(request) assert isinstance(result, dict) is True assert 'result' in result assert isinstance(result['result'], list) is True assert 'count' in result assert result['count'] == 2 assert len(result['result']) == 2 assert isinstance(result['result'][0], dict) is True assert [dict(element) for element in result['result']] == expected
def build_repository(self, release_name=None, with_metadata=False, root=None): package_name = self.get_remote_package_name(self.package_name) if not package_name: return None info = self.get_remote_informations(self.pypi_url.format(package_name)) if info: package_root = Root() if root is None else root package = Package(info['info']['name']) package.__parent__ = package_root package_root[package.name] = package remote_releases = [ release_name, ] if release_name else info['releases'].keys() for remote_release in remote_releases: release = Release(remote_release, remote_release, metadata=info['info'], deserialize_metadata=with_metadata) package[remote_release] = release for remote_release_file in info['releases'][remote_release]: filename = remote_release_file['filename'] md5_digest = remote_release_file['md5_digest'] release_file = ReleaseFile(filename, b'', md5_digest) setattr(release_file, 'pypi_url', remote_release_file['url']) release[filename] = release_file if root: return self.smart_merge(root, package) return package return None