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(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(db, user_without_profile):
profile = Profile(name="Bartosz",
avatar_url="http:///avatar",
user=user_without_profile)
db.add(profile)
db.commit()
yield user_without_profile
def other_user(other_user_without_profile, db):
profile = Profile(name="Other",
avatar_url="http:///avatar",
user=other_user_without_profile)
db.add(profile)
db.commit()
yield other_user_without_profile
def test_config_user_exists(routed_client, db, oauth_server, channel, user,
login, default_role, app):
profile = Profile(user=user)
identity = Identity(provider=oauth_server.provider,
user=user,
identity_id="existing_user_id")
db.add(identity)
db.add(profile)
db.commit()
response = routed_client.get(f'/auth/{oauth_server.provider}/authorize')
assert response.status_code == 200
db.refresh(user)
users = db.query(User).filter(User.username == login).all()
assert len(users) == 1
new_user = users[0]
assert new_user
assert new_user.username == login
assert login != "existing_user" or new_user.role == SERVER_OWNER
assert login == "existing_user" or new_user.role == default_role
assert new_user.profile
assert new_user.identities
assert new_user.identities[0].provider == oauth_server.provider
def profile(db, user):
user_profile = Profile(name="madhur",
avatar_url="madhur-tandon",
user_id=user.id,
user=user)
db.add(user_profile)
db.commit()
yield user
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 other_user(other_user_without_profile, db):
profile = Profile(name="Other",
avatar_url="http:///avatar",
user=other_user_without_profile)
identity = Identity(
provider="github",
identity_id="github",
username="******",
user=other_user_without_profile,
)
db.add(profile)
db.add(identity)
db.commit()
yield other_user_without_profile
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()
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)
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}')