Fixed extensions tests (test_base.py) to reflect new (& much older!) …

…changes to extension registration logic.

Also added proper setUp() and tearDown().
This should resolve the travis failures on this branch...

Signed-off-by: Matthias Stevens <matthias.stevens@gmail.com>

geokey/extensions/tests/test_base.py

@@ -1,33 +1,25 @@
 from django.test import TestCase
-from geokey.extensions.base import ExtensionExists, register, extensions as ext
+from ..base import ExtensionExists, register, deregister, extensions as ext
 
 
 class RegisterExtensionTest(TestCase):
+    def setUp(self):
+        register('test_ext', 'Test', True, True, '1.0.0')
+
+    def tearDown(self):
+        deregister('test_ext')
+
     def test(self):
-        register('text_ext', 'Test', True)
-        extension = ext.get('text_ext')
-        self.assertEqual(extension.get('ext_id'), 'text_ext')
+        extension = ext.get('test_ext')
+        self.assertEqual(extension.get('ext_id'), 'test_ext')
         self.assertEqual(extension.get('name'), 'Test')
-        self.assertEqual(extension.get('index_url'), 'text_ext:index')
+        self.assertEqual(extension.get('index_url'), 'test_ext:index')
         self.assertTrue(extension.get('display_admin'))
+        self.assertTrue(extension.get('superuser'))
+        self.assertEqual(extension.get('version'), '1.0.0')
 
     def test_exiting(self):
-        ext['text_ext'] = {
-            'ext_id': 'text_ext',
-            'name': 'Test',
-            'display_admin': True,
-            'index_url': 'text_ext:index'
-        }
-        try:
-            register('text_ext', 'Test Name', False)
-        except ExtensionExists:
-            extension = ext.get('text_ext')
-            self.assertEqual(extension.get('ext_id'), 'text_ext')
-            self.assertEqual(extension.get('name'), 'Test')
-            self.assertEqual(
-                extension.get('index_url'),
-                'text_ext:index'
-            )
-            self.assertTrue(extension.get('display_admin'))
-        else:
-            self.fail('ExtensionExists not raised')
+        self.assertRaises(
+            ExtensionExists,
+            register,
+            'test_ext', 'abc', False, False, '0.0.1')