From 943d97efac1f6fca6c823e0edb416b3c300f4b3d Mon Sep 17 00:00:00 2001
From: Kevin Morris <kevr@0cost.org>
Date: Tue, 1 Jun 2021 04:48:49 -0700
Subject: [PATCH] add License SQLAlchemy ORM model

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

diff --git a/aurweb/models/license.py b/aurweb/models/license.py
new file mode 100644
index 00000000..1c174925
--- /dev/null
+++ b/aurweb/models/license.py
@@ -0,0 +1,11 @@
+from sqlalchemy.orm import mapper
+
+from aurweb.schema import Licenses
+
+
+class License:
+    def __init__(self, Name: str = None):
+        self.Name = Name
+
+
+mapper(License, Licenses)
diff --git a/test/test_license.py b/test/test_license.py
new file mode 100644
index 00000000..feb7a396
--- /dev/null
+++ b/test/test_license.py
@@ -0,0 +1,25 @@
+import pytest
+
+from sqlalchemy.exc import IntegrityError
+
+from aurweb.db import create
+from aurweb.models.license import License
+from aurweb.testing import setup_test_db
+
+
+@pytest.fixture(autouse=True)
+def setup():
+    setup_test_db("Licenses")
+
+
+def test_license_creation():
+    license = create(License, Name="Test License")
+    assert bool(license.ID)
+    assert license.Name == "Test License"
+
+
+def test_license_null_name_raises_exception():
+    from aurweb.db import session
+    with pytest.raises(IntegrityError):
+        create(License)
+    session.rollback()