From 451d4a5ac093e74689daf668a9984623a8362eb9 Mon Sep 17 00:00:00 2001
From: leigh-mil <leigh-ctr@friends.dds.mil>
Date: Wed, 28 Nov 2018 14:39:03 -0500
Subject: [PATCH] Add tests

---
 tests/models/test_invitation.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 tests/models/test_invitation.py

diff --git a/tests/models/test_invitation.py b/tests/models/test_invitation.py
new file mode 100644
index 00000000..44fb30fb
--- /dev/null
+++ b/tests/models/test_invitation.py
@@ -0,0 +1,23 @@
+import pytest
+import datetime
+
+from atst.models.invitation import Invitation, Status
+
+from tests.factories import InvitationFactory
+
+
+def test_expired_invite_is_not_revokable():
+    invite = InvitationFactory.create(
+        expiration_time=datetime.datetime.now() - datetime.timedelta(minutes=60)
+    )
+    assert not invite.is_revokable
+
+
+def test_unexpired_invite_is_revokable():
+    invite = InvitationFactory.create()
+    assert invite.is_revokable
+
+
+def test_invite_is_not_revokable_if_invite_is_not_pending():
+    invite = InvitationFactory.create(status=Status.ACCEPTED)
+    assert not invite.is_revokable