add PackageRelation SQLAlchemy ORM model

Signed-off-by: Kevin Morris <kevr@0cost.org>

aurweb/models/package_relation.py

@@ -0,0 +1,33 @@
+from sqlalchemy.orm import mapper
+
+from aurweb.db import make_relationship
+from aurweb.models.package import Package
+from aurweb.models.relation_type import RelationType
+from aurweb.schema import PackageRelations
+
+
+class PackageRelation:
+    def __init__(self, Package: Package = None,
+                 RelationType: RelationType = None,
+                 RelName: str = None, RelCondition: str = None,
+                 RelArch: str = None):
+        self.Package = Package
+        self.RelationType = RelationType
+        self.RelName = RelName  # nullable=False
+        self.RelCondition = RelCondition
+        self.RelArch = RelArch
+
+
+properties = {
+    "Package": make_relationship(Package, PackageRelations.c.PackageID,
+                                 "package_relations"),
+    "RelationType": make_relationship(RelationType,
+                                      PackageRelations.c.RelTypeID,
+                                      "package_relations")
+}
+
+mapper(PackageRelation, PackageRelations, properties=properties,
+       primary_key=[
+           PackageRelations.c.PackageID,
+           PackageRelations.c.RelTypeID
+       ])

test/test_package_relation.py

@@ -0,0 +1,100 @@
+import pytest
+
+from sqlalchemy.exc import IntegrityError
+
+from aurweb.db import create, query
+from aurweb.models.account_type import AccountType
+from aurweb.models.package import Package
+from aurweb.models.package_base import PackageBase
+from aurweb.models.package_relation import PackageRelation
+from aurweb.models.relation_type import RelationType
+from aurweb.models.user import User
+from aurweb.testing import setup_test_db
+
+user = pkgbase = package = None
+
+
+@pytest.fixture(autouse=True)
+def setup():
+    global user, pkgbase, package
+
+    setup_test_db("Users", "PackageBases", "Packages", "PackageRelations")
+
+    account_type = query(AccountType,
+                         AccountType.AccountType == "User").first()
+    user = create(User, Username="test", Email="test@example.org",
+                  RealName="Test User", Passwd="testPassword",
+                  AccountType=account_type)
+
+    pkgbase = create(PackageBase,
+                     Name="test-package",
+                     Maintainer=user)
+    package = create(Package,
+                     PackageBase=pkgbase,
+                     Name=pkgbase.Name,
+                     Description="Test description.",
+                     URL="https://test.package")
+
+
+def test_package_dependencies():
+    conflicts = query(RelationType, RelationType.Name == "conflicts").first()
+    pkgrel = create(PackageRelation, Package=package,
+                    RelationType=conflicts,
+                    RelName="test-relation")
+    assert pkgrel.RelName == "test-relation"
+    assert pkgrel.Package == package
+    assert pkgrel.RelationType == conflicts
+    assert pkgrel in conflicts.package_relations
+    assert pkgrel in package.package_relations
+
+    provides = query(RelationType, RelationType.Name == "provides").first()
+    pkgrel = create(PackageRelation, Package=package,
+                    RelationType=provides,
+                    RelName="test-relation")
+    assert pkgrel.RelName == "test-relation"
+    assert pkgrel.Package == package
+    assert pkgrel.RelationType == provides
+    assert pkgrel in provides.package_relations
+    assert pkgrel in package.package_relations
+
+    replaces = query(RelationType, RelationType.Name == "replaces").first()
+    pkgrel = create(PackageRelation, Package=package,
+                    RelationType=replaces,
+                    RelName="test-relation")
+    assert pkgrel.RelName == "test-relation"
+    assert pkgrel.Package == package
+    assert pkgrel.RelationType == replaces
+    assert pkgrel in replaces.package_relations
+    assert pkgrel in package.package_relations
+
+
+def test_package_dependencies_null_package_raises_exception():
+    from aurweb.db import session
+
+    conflicts = query(RelationType, RelationType.Name == "conflicts").first()
+    with pytest.raises(IntegrityError):
+        create(PackageRelation,
+               RelationType=conflicts,
+               RelName="test-relation")
+    session.rollback()
+
+
+def test_package_dependencies_null_dependency_type_raises_exception():
+    from aurweb.db import session
+
+    with pytest.raises(IntegrityError):
+        create(PackageRelation,
+               Package=package,
+               RelName="test-relation")
+    session.rollback()
+
+
+def test_package_dependencies_null_depname_raises_exception():
+    from aurweb.db import session
+
+    depends = query(RelationType, RelationType.Name == "depends").first()
+    with pytest.raises(IntegrityError):
+        create(PackageRelation,
+               Package=package,
+               RelationType=depends)
+    session.rollback()