Merge pull request #273 from NebraLtd/shawaj/testsbc

sbc: add test cases

hm_pyhelper/sbc.py

@@ -207,7 +207,7 @@ def is_nebra_fleet() -> bool:
     api_url = os.environ.get('BALENA_API_URL')
     fleet_id = int(os.environ.get('BALENA_APP_ID'))
 
-    if api_url is not NEBRA_API_URL or (fleet_id not in COMMERCIAL_FLEETS and fleet_id not in NON_COMMERCIAL_FLEETS):
+    if (api_url != NEBRA_API_URL) or (fleet_id not in COMMERCIAL_FLEETS and fleet_id not in NON_COMMERCIAL_FLEETS):
         return False
 
     return True 

hm_pyhelper/tests/test_sbc.py

@@ -0,0 +1,27 @@
+import unittest
+
+from unittest.mock import patch
+from hm_pyhelper.sbc import is_commercial_fleet, is_nebra_fleet
+
+
+class TestSBC(unittest.TestCase):
+
+    @patch.dict('os.environ', {"BALENA_API_URL": "https://api.cloud.nebra.com"})
+    @patch.dict('os.environ', {"BALENA_APP_ID": "55"})
+    def test_is_nebra_fleet_true(self):
+        self.assertTrue(is_nebra_fleet())
+
+    @patch.dict('os.environ', {"BALENA_API_URL": "https://test.com"})
+    @patch.dict('os.environ', {"BALENA_APP_ID": "5500"})
+    def test_is_nebra_fleet_false(self):
+        self.assertFalse(is_nebra_fleet())
+
+    @patch.dict('os.environ', {"BALENA_APP_NAME": "test-c"})
+    @patch.dict('os.environ', {"BALENA_APP_ID": "87"})
+    def test_is_commercial_fleet_true(self):
+        self.assertTrue(is_commercial_fleet())
+
+    @patch.dict('os.environ', {"BALENA_APP_NAME": "test"})
+    @patch.dict('os.environ', {"BALENA_APP_ID": "8700"})
+    def test_is_commercial_fleet_false(self):
+        self.assertFalse(is_commercial_fleet())

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "hm_pyhelper"
-version = "0.14.18"
+version = "0.14.19"
 description = "Helium Python Helper"
 authors = ["Nebra Ltd <[email protected]>"]
 readme = "README.md"