def __init__(self, db):
self.keya = "akey"
self.keyb = "bkey"
self.usera = User(id=uuid.uuid4().bytes, username='******')
Profile(name='usera', user=self.usera, avatar_url='')
db.add(self.usera)
self.userb = User(id=uuid.uuid4().bytes, username='******')
Profile(name='userb', user=self.userb, avatar_url='')
db.add(self.userb)
self.userc = User(id=uuid.uuid4().bytes, username='******', role="owner")
Profile(name='userc', user=self.userc, avatar_url='')
db.add(self.userc)
assert len(db.query(User).all()) == 3
db.add(ApiKey(key=self.keya, user_id=self.usera.id, owner_id=self.usera.id))
db.add(ApiKey(key=self.keyb, user_id=self.userb.id, owner_id=self.userb.id))
self.channel1 = Channel(name="testchannel", private=False)
self.channel2 = Channel(name="privatechannel", private=True)
self.package1 = Package(name="Package1", channel=self.channel1)
self.package2 = Package(name="Package2", channel=self.channel2)
self.channel_member = ChannelMember(
channel=self.channel2, user=self.usera, role='maintainer'
)
self.channel_member_userc = ChannelMember(
channel=self.channel2, user=self.userc, role='owner'
)
self.package_member = PackageMember(
channel=self.channel2, user=self.userc, package=self.package2, role="owner"
)
for el in [
self.channel1,
self.channel2,
self.channel_member,
self.channel_member_userc,
self.package1,
self.package2,
self.package_member,
]:
db.add(el)
db.commit()
self.db = db
def package_version(
dao: "quetz.dao.Dao",
user,
channel,
package_name,
db,
package_file_name,
package_format,
package_subdir,
):
channel_data = json.dumps({"subdirs": [package_subdir]})
package_data = Package(name=package_name)
package = dao.create_package(channel.name, package_data, user.id, "owner")
package.channeldata = channel_data
db.commit()
package_info = (
'{"run_exports": {"weak": ["otherpackage > 0.1"]}, "size": 100, "depends": []}'
)
version = dao.create_version(
channel.name,
package_name,
package_format,
package_subdir,
"0.1",
"0",
"0",
package_file_name,
package_info,
user.id,
size=0,
)
yield version
def package_repodata_patches(
dao: "quetz.dao.Dao",
user,
channel,
db,
pkgstore,
repodata_name,
repodata_file_name,
repodata_archive,
archive_format,
):
package_name = repodata_name
package_data = Package(name=package_name)
dao.create_package(channel.name, package_data, user.id, "owner")
package_info = '{"size": 100, "depends":[]}'
package_format = archive_format
version = dao.create_version(
channel.name,
package_name,
package_format,
"noarch",
"0.1",
"0",
"0",
repodata_file_name,
package_info,
user.id,
)
pkgstore.add_package(repodata_archive, channel.name, f"noarch/{repodata_file_name}")
return version
def test_create_versions_from_repodata(dao, user, local_channel, db):
pkg = Package(name="other-package", channel=local_channel)
db.add(pkg)
repodata = json.loads(repodata_json)
create_versions_from_repodata(local_channel.name, user.id, repodata, dao)
version = (db.query(PackageVersion).filter(
PackageVersion.package_name == "other-package").one())
assert version
def mirror_package(mirror_channel, db):
pkg = Package(name="mirror_package",
channel_name=mirror_channel.name,
channel=mirror_channel)
db.add(pkg)
db.commit()
yield pkg
db.delete(pkg)
db.commit()
def test_handle_repodata_package_with_plugin(dao, local_channel,
dummy_package_file, rules, config,
db, plugin, user):
pkg = Package(name="other-package", channel=local_channel)
db.add(pkg)
repodata = json.loads(repodata_json)
package_name, package_data = list(repodata["packages"].items())[0]
pkgstore = config.get_package_store()
files_metadata = [(dummy_package_file, package_name, package_data)]
handle_repodata_package(local_channel.name, files_metadata, dao, rules,
False, pkgstore, config)
assert plugin.about['conda_version'] == '4.8.4'
def test_handle_repodata_package(dao, user, local_channel, dummy_package_file,
rules, config, db):
pkg = Package(name="other-package", channel=local_channel)
db.add(pkg)
repodata = json.loads(repodata_json)
package_name, package_data = list(repodata["packages"].items())[0]
pkgstore = config.get_package_store()
files_metadata = [(dummy_package_file, package_name, package_data)]
handle_repodata_package(local_channel.name, files_metadata, dao, rules,
False, pkgstore, config)
version = (db.query(PackageVersion).filter(
PackageVersion.package_name == "other-package").one())
assert version
pkgstore.serve_path(local_channel.name, f"linux-64/{package_name}")
def _fill_test_database(db: Session) -> NoReturn:
"""Create dummy users and channels to allow further testing in dev mode."""
from quetz.dao import Dao
test_users = []
dao = Dao(db)
try:
for index, username in enumerate(['alice', 'bob', 'carol', 'dave']):
user = dao.create_user_with_role(username)
identity = Identity(
provider='dummy',
identity_id=str(index),
username=username,
)
profile = Profile(name=username.capitalize(),
avatar_url='/avatar.jpg')
user.identities.append(identity) # type: ignore
user.profile = profile
db.add(user)
test_users.append(user)
for channel_index in range(3):
channel = Channel(
name=f'channel{channel_index}',
description=f'Description of channel{channel_index}',
private=False,
)
for package_index in range(random.randint(5, 10)):
package = Package(
name=f'package{package_index}',
summary=f'package {package_index} summary text',
description=f'Description of package{package_index}',
)
channel.packages.append(package) # type: ignore
test_user = test_users[random.randint(0, len(test_users) - 1)]
package_member = PackageMember(package=package,
channel=channel,
user=test_user,
role='owner')
db.add(package_member)
test_user = test_users[random.randint(0, len(test_users) - 1)]
if channel_index == 0:
package = Package(name='xtensor',
description='Description of xtensor')
channel.packages.append(package) # type: ignore
package_member = PackageMember(package=package,
channel=channel,
user=test_user,
role='owner')
db.add(package_member)
# create API key
key = uuid.uuid4().hex
key_user = User(id=uuid.uuid4().bytes)
api_key = ApiKey(key=key,
description='test API key',
user=test_user,
owner=test_user)
db.add(api_key)
print(
f'Test API key created for user "{test_user.username}": {key}'
)
key_package_member = PackageMember(
user=key_user,
channel_name=channel.name,
package_name=package.name,
role='maintainer',
)
db.add(key_package_member)
db.add(channel)
channel_member = ChannelMember(
channel=channel,
user=test_user,
role='owner',
)
db.add(channel_member)
db.commit()
finally:
db.close()
def __init__(self, db):
self.keya = "akey"
self.keyb = "bkey"
self.usera = User(id=uuid.uuid4().bytes, username='******')
Profile(name='usera', user=self.usera, avatar_url='')
db.add(self.usera)
self.userb = User(id=uuid.uuid4().bytes, username='******')
Profile(name='userb', user=self.userb, avatar_url='')
db.add(self.userb)
self.userc = User(id=uuid.uuid4().bytes, username='******', role="owner")
Profile(name='userc', user=self.userc, avatar_url='')
db.add(self.userc)
assert len(db.query(User).all()) == 3
self.keya_obj = ApiKey(
key=self.keya,
time_created=date.today(),
expire_at=date(2030, 1, 1),
user_id=self.usera.id,
owner_id=self.usera.id,
)
db.add(self.keya_obj)
db.add(
ApiKey(
key=self.keyb,
time_created=date.today(),
expire_at=date(2030, 1, 1),
user_id=self.userb.id,
owner_id=self.userb.id,
)
)
self.channel1 = Channel(name="testchannel", private=False)
self.channel2 = Channel(name="privatechannel", private=True)
self.package1 = Package(name="package1", channel=self.channel1)
self.package2 = Package(name="package2", channel=self.channel2)
self.package_version_1 = PackageVersion(
id=uuid.uuid4().bytes,
channel_name=self.channel1.name,
package_name=self.package1.name,
package_format='tarbz2',
platform='noarch',
version="0.0.1",
build_number="0",
build_string="",
filename="filename.tar.bz2",
info="{}",
uploader_id=self.usera.id,
size=101,
)
self.package_version_2 = PackageVersion(
id=uuid.uuid4().bytes,
channel_name=self.channel2.name,
package_name=self.package2.name,
package_format='tarbz2',
platform='noarch',
version="0.0.1",
build_number="0",
build_string="",
filename="filename2.tar.bz2",
info="{}",
uploader_id=self.usera.id,
size=101,
)
self.channel_member = ChannelMember(
channel=self.channel2, user=self.usera, role='maintainer'
)
self.channel_member_userc = ChannelMember(
channel=self.channel2, user=self.userc, role='owner'
)
self.package_member = PackageMember(
channel=self.channel2, user=self.userc, package=self.package2, role="owner"
)
for el in [
self.channel1,
self.channel2,
self.channel_member,
self.channel_member_userc,
self.package1,
self.package2,
self.package_version_1,
self.package_version_2,
self.package_member,
]:
db.add(el)
db.commit()
self.db = db
def init_test_db():
config = Config()
init_db(config.sqlalchemy_database_url)
db = get_session(config.sqlalchemy_database_url)
testUsers = []
try:
for index, username in enumerate(['alice', 'bob', 'carol', 'dave']):
user = User(id=uuid.uuid4().bytes, username=username)
identity = Identity(
provider='dummy',
identity_id=str(index),
username=username,
)
profile = Profile(name=username.capitalize(),
avatar_url='/avatar.jpg')
user.identities.append(identity)
user.profile = profile
db.add(user)
testUsers.append(user)
for channel_index in range(30):
channel = Channel(
name=f'channel{channel_index}',
description=f'Description of channel{channel_index}',
private=False,
)
for package_index in range(random.randint(5, 100)):
package = Package(
name=f'package{package_index}',
description=f'Description of package{package_index}',
)
channel.packages.append(package)
test_user = testUsers[random.randint(0, len(testUsers) - 1)]
package_member = PackageMember(package=package,
channel=channel,
user=test_user,
role='owner')
db.add(package_member)
if channel_index == 0:
package = Package(name='xtensor',
description='Description of xtensor')
channel.packages.append(package)
test_user = testUsers[random.randint(0, len(testUsers) - 1)]
package_member = PackageMember(package=package,
channel=channel,
user=test_user,
role='owner')
db.add(package_member)
# create API key
key = 'E_KaBFstCKI9hTdPM7DQq56GglRHf2HW7tQtq6si370'
key_user = User(id=uuid.uuid4().bytes)
api_key = ApiKey(key=key,
description='test API key',
user=key_user,
owner=test_user)
db.add(api_key)
key_package_member = PackageMember(
user=key_user,
channel_name=channel.name,
package_name=package.name,
role='maintainer',
)
db.add(key_package_member)
db.add(channel)
channel_member = ChannelMember(
channel=channel,
user=testUsers[random.randint(0,
len(testUsers) - 1)],
role='owner',
)
db.add(channel_member)
db.commit()
finally:
db.close()
def local_package(db, local_channel):
pkg = Package(name="other-package", channel=local_channel)
db.add(pkg)
db.commit()
# do not keep the instance because sqlalchemy won't allow creating duplicate
return
profile = Profile(name=username.capitalize(),
avatar_url='/static/avatar.jpg')
user.identities.append(identity)
user.profile = profile
db.add(user)
testUsers.append(user)
for channel_index in range(30):
channel = Channel(name=f'channel{channel_index}',
description=f'Description of channel{channel_index}',
private=False)
for package_index in range(random.randint(5, 100)):
package = Package(
name=f'package{package_index}',
description=f'Description of package{package_index}')
channel.packages.append(package)
test_user = testUsers[random.randint(0, len(testUsers) - 1)]
package_member = PackageMember(package=package,
channel=channel,
user=test_user,
role='owner')
db.add(package_member)
if channel_index == 0:
package = Package(name=f'xtensor',
description=f'Description of xtensor')
channel.packages.append(package)
