feat: add import statements on resources in top-level package

Author: Charles Larivier

metabase/__init__.py

@@ -1,6 +1,7 @@
 from metabase.metabase import Metabase
-from metabase.resources.database import Database
 from metabase.resources.user import User
-from metabase.resources.permission import PermissionGroup, PermissionMembership
-from metabase.resources.collection import Collection
-
+from metabase.resources.permission_group import PermissionGroup
+from metabase.resources.permission_membership import PermissionMembership
+from metabase.resources.table import Table
+from metabase.resources.segment import Segment
+from metabase.resources.metric import Metric

metabase/resources/table.py

@@ -7,7 +7,7 @@
 from metabase.resources.field import Field
 from metabase.missing import MISSING
 from metabase.resources.metric import Metric
-from resources.segment import Segment
+from metabase.resources.segment import Segment
 
 
 class Dimension(Resource):

tests/resources/test_metric.py

@@ -10,6 +10,11 @@ def tearDown(self) -> None:
         for metric in metrics:
             metric.archive()
 
+    def test_import(self):
+        """Ensure Metric can be imported from Metabase."""
+        from metabase import Metric
+        self.assertIsNotNone(Metric())
+
     def test_list(self):
         """Ensure Metric.list returns a list of Metric instances."""
         # fixture

tests/resources/test_permission_group.py

@@ -12,6 +12,11 @@ def tearDown(self) -> None:
                 # can't delete default groups
                 group.delete()
 
+    def test_import(self):
+        """Ensure PermissionGroup can be imported from Metabase."""
+        from metabase import PermissionGroup
+        self.assertIsNotNone(PermissionGroup())
+
     def test_list(self):
         """Ensure PermissionGroup.list returns a list of PermissionGroup instances."""
         groups = PermissionGroup.list()

tests/resources/test_permission_membership.py

@@ -8,8 +8,8 @@ class PermissionMembershipTests(IntegrationTestCase):
     def tearDown(self) -> None:
         memberships = PermissionMembership.list()
         for membership in memberships:
-            if membership.membership_id not in (1, 2):
-                # can't delete default memberships
+            if membership.group_id not in (1, 2):
+                # can't delete memberships in the default groups
                 membership.delete()
 
         groups = PermissionGroup.list()
@@ -18,12 +18,16 @@ def tearDown(self) -> None:
                 # can't delete default groups
                 group.delete()
 
+    def test_import(self):
+        """Ensure PermissionMembership can be imported from Metabase."""
+        from metabase import PermissionMembership
+        self.assertIsNotNone(PermissionMembership())
+
     def test_list(self):
         """Ensure PermissionMembership.list returns a list of PermissionMembership instances."""
         memberships = PermissionMembership.list()
-
         self.assertIsInstance(memberships, list)
-        self.assertEqual(2, len(memberships))    # there are 2 default groups in Metabase
+        self.assertTrue(len(memberships) > 0)
         self.assertTrue(all([isinstance(m, PermissionMembership) for m in memberships]))
 
     def test_create(self):

tests/resources/test_segment.py

@@ -10,6 +10,11 @@ def tearDown(self) -> None:
         for segment in segments:
             segment.archive()
 
+    def test_import(self):
+        """Ensure Segment can be imported from Metabase."""
+        from metabase import Segment
+        self.assertIsNotNone(Segment())
+
     def test_list(self):
         """Ensure Segment.list returns a list of Segment instances."""
         # fixture

tests/resources/test_table.py

@@ -9,6 +9,11 @@ class TableTests(IntegrationTestCase):
     def setUp(self) -> None:
         super(TableTests, self).setUp()
 
+    def test_import(self):
+        """Ensure Table can be imported from Metabase."""
+        from metabase import Table
+        self.assertIsNotNone(Table())
+
     def test_list(self):
         """Ensure Table.list() returns a list of Table instances."""
         tables = Table.list()

tests/resources/test_user.py

@@ -13,6 +13,11 @@ def tearDown(self) -> None:
             if user.id != 1:
                 user.delete()
 
+    def test_import(self):
+        """Ensure User can be imported from Metabase."""
+        from metabase import User
+        self.assertIsNotNone(User())
+
     def test_get(self):
         """
         Ensure User.get() returns a User instance for a given ID, or