From ed6bafbed82b35913965a6d95ba11b4f3eaf4b02 Mon Sep 17 00:00:00 2001
From: David Read <david.read@hackneyworkshop.com>
Date: Tue, 10 Nov 2015 12:58:00 +0000
Subject: [PATCH] Test for previous ResourceGroup change.

---
 tests/test_model.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 tests/test_model.py

diff --git a/tests/test_model.py b/tests/test_model.py
new file mode 100644
index 00000000..c25d8645
--- /dev/null
+++ b/tests/test_model.py
@@ -0,0 +1,24 @@
+import ckanext.archiver.model as archiver_model
+try:
+    from ckan.tests.helpers import reset_db
+    from ckan.tests import factories as ckan_factories
+except ImportError:
+    from ckan.new_tests.helpers import reset_db
+    from ckan.new_tests import factories as ckan_factories
+from ckan import model
+
+Archival = archiver_model.Archival
+
+
+class TestArchival(object):
+    @classmethod
+    def setup_class(cls):
+        reset_db()
+        archiver_model.init_tables(model.meta.engine)
+
+    def test_create(self):
+        dataset = ckan_factories.Dataset()
+        res = ckan_factories.Resource(package_id=dataset['id'])
+        archival = Archival.create(res['id'])
+        assert isinstance(archival, Archival)
+        assert archival.package_id == dataset['id']