add DependencyType SQLAlchemy ORM model

Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris 2021-06-01 06:44:24 -07:00
parent 4201348dea
commit 068c8ba638
2 changed files with 42 additions and 0 deletions

View file

@ -0,0 +1,11 @@
from sqlalchemy.orm import mapper
from aurweb.schema import DependencyTypes
class DependencyType:
def __init__(self, Name: str = None):
self.Name = Name
mapper(DependencyType, DependencyTypes)

View file

@ -0,0 +1,31 @@
import pytest
from aurweb.db import create, delete, query
from aurweb.models.dependency_type import DependencyType
from aurweb.testing import setup_test_db
@pytest.fixture(autouse=True)
def setup():
setup_test_db()
def test_dependency_types():
dep_types = ["depends", "makedepends", "checkdepends", "optdepends"]
for dep_type in dep_types:
dependency_type = query(DependencyType,
DependencyType.Name == dep_type).first()
assert dependency_type is not None
def test_dependency_type_creation():
dependency_type = create(DependencyType, Name="Test Type")
assert bool(dependency_type.ID)
assert dependency_type.Name == "Test Type"
delete(DependencyType, DependencyType.ID == dependency_type.ID)
def test_dependency_type_null_name_uses_default():
dependency_type = create(DependencyType)
assert dependency_type.Name == str()
delete(DependencyType, DependencyType.ID == dependency_type.ID)