add TUVote SQLAlchemy ORM model

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

test/test_tu_vote.py

@@ -0,0 +1,56 @@
+from datetime import datetime
+
+import pytest
+
+from sqlalchemy.exc import IntegrityError
+
+from aurweb.db import create, query, rollback
+from aurweb.models.account_type import AccountType
+from aurweb.models.tu_vote import TUVote
+from aurweb.models.tu_voteinfo import TUVoteInfo
+from aurweb.models.user import User
+from aurweb.testing import setup_test_db
+
+user = tu_voteinfo = None
+
+
+@pytest.fixture(autouse=True)
+def setup():
+    global user, tu_voteinfo
+
+    setup_test_db("Users", "TU_VoteInfo", "TU_Votes")
+
+    tu_type = query(AccountType,
+                    AccountType.AccountType == "Trusted User").first()
+    user = create(User, Username="test", Email="test@example.org",
+                  RealName="Test User", Passwd="testPassword",
+                  AccountType=tu_type)
+
+    ts = int(datetime.utcnow().timestamp())
+    tu_voteinfo = create(TUVoteInfo,
+                         Agenda="Blah blah.",
+                         User=user.Username,
+                         Submitted=ts, End=ts + 5,
+                         Quorum=0.5,
+                         Submitter=user)
+
+
+def test_tu_vote_creation():
+    tu_vote = create(TUVote, User=user, VoteInfo=tu_voteinfo)
+    assert tu_vote.VoteInfo == tu_voteinfo
+    assert tu_vote.User == user
+
+    assert tu_vote in user.tu_votes
+    assert tu_vote in tu_voteinfo.tu_votes
+
+
+def test_tu_vote_null_user_raises_exception():
+    with pytest.raises(IntegrityError):
+        create(TUVote, VoteInfo=tu_voteinfo)
+    rollback()
+
+
+def test_tu_vote_null_voteinfo_raises_exception():
+    with pytest.raises(IntegrityError):
+        create(TUVote, User=user)
+    rollback()