From a9cfbce11e3c16c22d168f8fc55238f17ea78273 Mon Sep 17 00:00:00 2001
From: Kevin Morris <kevr@0cost.org>
Date: Tue, 1 Jun 2021 07:37:05 -0700
Subject: [PATCH] add RelationType SQLAlchemy ORM model

Signed-off-by: Kevin Morris <kevr@0cost.org>
---
 aurweb/models/relation_type.py | 11 +++++++++++
 test/test_relation_type.py     | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 43 insertions(+)
 create mode 100644 aurweb/models/relation_type.py
 create mode 100644 test/test_relation_type.py

diff --git a/aurweb/models/relation_type.py b/aurweb/models/relation_type.py
new file mode 100644
index 00000000..b4d1efbc
--- /dev/null
+++ b/aurweb/models/relation_type.py
@@ -0,0 +1,11 @@
+from sqlalchemy.orm import mapper
+
+from aurweb.schema import RelationTypes
+
+
+class RelationType:
+    def __init__(self, Name: str = None):
+        self.Name = Name
+
+
+mapper(RelationType, RelationTypes)
diff --git a/test/test_relation_type.py b/test/test_relation_type.py
new file mode 100644
index 00000000..bf23505c
--- /dev/null
+++ b/test/test_relation_type.py
@@ -0,0 +1,32 @@
+import pytest
+
+from aurweb.db import create, delete, query
+from aurweb.models.relation_type import RelationType
+from aurweb.testing import setup_test_db
+
+
+@pytest.fixture(autouse=True)
+def setup():
+    setup_test_db()
+
+
+def test_relation_type_creation():
+    relation_type = create(RelationType, Name="test-relation")
+    assert bool(relation_type.ID)
+    assert relation_type.Name == "test-relation"
+
+    delete(RelationType, RelationType.ID == relation_type.ID)
+
+
+def test_relation_types():
+    conflicts = query(RelationType, RelationType.Name == "conflicts").first()
+    assert conflicts is not None
+    assert conflicts.Name == "conflicts"
+
+    provides = query(RelationType, RelationType.Name == "provides").first()
+    assert provides is not None
+    assert provides.Name == "provides"
+
+    replaces = query(RelationType, RelationType.Name == "replaces").first()
+    assert replaces is not None
+    assert replaces.Name == "replaces"