def test_api_key(db):
usera = User(id=uuid.uuid4().bytes, username='******')
db.add(usera)
userb = User(id=uuid.uuid4().bytes, username='******')
db.add(userb)
db.commit()
assert len(db.query(User).all()) >= 2
db.add(ApiKey(key=keya, user_id=usera.id, owner_id=usera.id))
db.add(ApiKey(key=keyb, user_id=userb.id, owner_id=userb.id))
db.commit()
channel1 = Channel(name="testchannel", private=False)
channel2 = Channel(name="privatechannel", private=True)
channel_member = ChannelMember(channel=channel2, user=usera, role='OWNER')
for el in [channel1, channel2, channel_member]:
db.add(el)
db.commit()
client = TestClient(app)
response = client.get('/')
assert (len(response.text))
response = client.get('/api/channels')
response = client.get('/api/channels', headers={"X-Api-Key": keya})
assert (len(response.json()) == 2)
response = client.get('/api/channels', headers={"X-Api-Key": keyb})
assert (len(response.json()) == 1)
def test_user(db):
user = User(id=uuid.uuid4().bytes, username='******')
db.add(user)
db.commit()
assert len(db.query(User).all()) == 1
found = User.find(db, 'paul')
assert found.username == user.username
found = User.find(db, 'dave')
assert found is None
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 user_without_profile(db, user_role):
new_user = User(id=uuid.uuid4().bytes, username="******", role=user_role)
db.add(new_user)
db.commit()
return new_user
def user(db):
user = User(id=uuid.uuid4().bytes, username="******")
profile = Profile(name="Bartosz", avatar_url="http:///avatar", user=user)
db.add(user)
db.add(profile)
db.commit()
yield user
def user(client, db, username, logged):
user = User(id=uuid.uuid4().bytes, username=username)
db.add(user)
db.commit()
if logged:
client.get(f"/api/dummylogin/{user.username}")
yield user
def other_user(db):
other_user = User(id=uuid.uuid4().bytes, username='******')
db.add(other_user)
yield other_user
db.delete(other_user)
db.commit()
def dummy_user(db):
new_user = User(id=uuid.uuid4().bytes, username="******", role="owner")
db.add(new_user)
db.commit()
yield new_user
db.delete(new_user)
db.commit()
def user(db, user_role):
new_user = User(id=uuid.uuid4().bytes, username="******", role=user_role)
profile = Profile(name="Bartosz",
avatar_url="http:///avatar",
user=new_user)
db.add(profile)
db.add(new_user)
db.commit()
yield new_user
db.delete(new_user)
db.commit()
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 user(db):
user = User(id=uuid.uuid4().bytes, username="******")
db.add(user)
db.commit()
yield user
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 member_user(db):
user = User(id=uuid.uuid4().bytes, username="******", role="member")
db.add(user)
db.commit()
yield user
def user(db):
user = User(id=uuid.uuid4().bytes, username="******")
db.add(user)
db.commit()
yield user
def other_user_without_profile(db):
user = User(id=uuid.uuid4().bytes, username="******")
db.add(user)
return user
import random
import uuid
from quetz.database import init_db, SessionLocal
from quetz.db_models import (User, Identity, Profile, Channel, ChannelMember,
Package, PackageMember, ApiKey)
init_db()
db = SessionLocal()
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='/static/avatar.jpg')
user.identities.append(identity)
user.profile = profile
db.add(user)
testUsers.append(user)
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 owner_user(db):
user = User(id=uuid.uuid4().bytes, username="******", role="owner")
db.add(user)
db.commit()
yield user
