feat: Add unit tests

Author: ZPascal

grafana_api/datasource.py

@@ -1074,7 +1074,7 @@ class DatasourceLabelBasedAccessControl:
     def __init__(self, grafana_api_model: APIModel):
         self.grafana_api_model = grafana_api_model
 
-    def get_lbac_rules_for_datasoure(self, uid: str) -> list:
+    def get_lbac_rules_for_datasource(self, uid: str) -> list:
         """The method includes a functionality to get all datasource label based access control rules for team specified by the datasource uid
 
         Args:
@@ -1106,7 +1106,7 @@ def get_lbac_rules_for_datasoure(self, uid: str) -> list:
             logging.error("There is no uid defined.")
             raise ValueError
 
-    def update_lbac_rules_for_datasoure(self, uid: str) -> dict:
+    def update_lbac_rules_for_datasource(self, uid: str) -> dict:
         """The method includes a functionality to enable the datasource cache specified by the datasource uid
 
         Args:

tests/unittests/test_datasource.py

@@ -12,6 +12,7 @@
     DatasourcePermissions,
     DatasourceLegacyPermissions,
     DatasourceQueryResourceCaching,
+    DatasourceLabelBasedAccessControl,
 )
 
 
@@ -1022,3 +1023,63 @@ def test_update_datasource_cache_no_valid_result(self, call_the_api_mock):
 
         with self.assertRaises(Exception):
             datasource.update_datasource_cache("test", datasource_cache)
+
+class DatasourceLabelBasedAccessControlTestCase(TestCase):
+    @patch("grafana_api.api.Api.call_the_api")
+    def test_get_lbac_rules_for_datasource(self, call_the_api_mock):
+        model: APIModel = APIModel(host=MagicMock(), token=MagicMock())
+        datasource: DatasourceLabelBasedAccessControl = DatasourceLabelBasedAccessControl(grafana_api_model=model)
+
+        call_the_api_mock.return_value = list([{"id": 1}])
+
+        self.assertEqual([{"id": 1}], datasource.get_lbac_rules_for_datasource("test"))
+
+    @patch("grafana_api.api.Api.call_the_api")
+    def test_get_lbac_rules_for_datasource_no_uid(self, call_the_api_mock):
+        model: APIModel = APIModel(host=MagicMock(), token=MagicMock())
+        datasource: DatasourceLabelBasedAccessControl = DatasourceLabelBasedAccessControl(grafana_api_model=model)
+
+        call_the_api_mock.return_value = None
+
+        with self.assertRaises(ValueError):
+            datasource.get_lbac_rules_for_datasource("")
+
+    @patch("grafana_api.api.Api.call_the_api")
+    def test_get_lbac_rules_for_datasource_no_valid_rules(self, call_the_api_mock):
+        model: APIModel = APIModel(host=MagicMock(), token=MagicMock())
+        datasource: DatasourceLabelBasedAccessControl = DatasourceLabelBasedAccessControl(grafana_api_model=model)
+
+        call_the_api_mock.return_value = None
+
+        with self.assertRaises(Exception):
+            datasource.get_lbac_rules_for_datasource("test")
+
+    @patch("grafana_api.api.Api.call_the_api")
+    def test_update_lbac_rules_for_datasource(self, call_the_api_mock):
+        model: APIModel = APIModel(host=MagicMock(), token=MagicMock())
+        datasource: DatasourceLabelBasedAccessControl = DatasourceLabelBasedAccessControl(grafana_api_model=model)
+
+        call_the_api_mock.return_value = dict({"dataSourceID": 1})
+
+        self.assertEqual({"dataSourceID": 1}, datasource.update_lbac_rules_for_datasource("test"))
+
+    @patch("grafana_api.api.Api.call_the_api")
+    def test_update_lbac_rules_for_datasource_no_uid(self, call_the_api_mock):
+        model: APIModel = APIModel(host=MagicMock(), token=MagicMock())
+        datasource: DatasourceLabelBasedAccessControl = DatasourceLabelBasedAccessControl(grafana_api_model=model)
+
+        call_the_api_mock.return_value = None
+
+        with self.assertRaises(ValueError):
+            datasource.update_lbac_rules_for_datasource("")
+
+    @patch("grafana_api.api.Api.call_the_api")
+    def test_update_lbac_rules_for_datasource_no_update_possible(self, call_the_api_mock):
+        model: APIModel = APIModel(host=MagicMock(), token=MagicMock())
+        datasource: DatasourceLabelBasedAccessControl = DatasourceLabelBasedAccessControl(grafana_api_model=model)
+
+        call_the_api_mock.return_value = dict()
+
+        with self.assertRaises(Exception):
+            datasource.update_lbac_rules_for_datasource("test")
+