More testing

Author: areed1192

ibw/client.py

@@ -114,6 +114,7 @@ def create_session(self) -> bool:
         # more than likely it's running let's try and see if we can authenticate.
         auth_response = self.is_authenticated()
 
+
         if 'authenticated' in auth_response.keys() and auth_response['authenticated'] == True:
 
             if self._set_server():
@@ -932,10 +933,6 @@ def update_server_account(self, account_id: str, check: bool = False) -> Dict:
 
         content = self._make_request(endpoint = endpoint, req_type = req_type, params = params)
 
-        # if 'status_code' in content.keys():
-        #     time.sleep(1)
-        #     content = self._make_request(endpoint = endpoint, req_type = req_type, params = params)
-
         return content
 
 

tests/unit/test_client.py

@@ -66,12 +66,18 @@ def test_create_session(self):
         session_response = self.ibw_client.create_session()
         self.assertTrue(session_response)
         self.assertTrue(self.ibw_client.authenticated)
+
+    def test_close_session(self):
+        """Test Closing the session."""
+        
+        with self.assertRaises(SystemExit) as cm:
+            self.ibw_client.close_session()
+            self.assertEqual(cm.exception.code, 1)
 
     def tearDown(self) -> None:
         """Teardown the Session."""
 
         self.ibw_client = None
 
-
 if __name__ == '__main__':
     unittest.main()