Fixed test_get_token() for updated code

Author: Juliana Mashon

main.py

@@ -13,10 +13,11 @@ def get_token(self, email, password, server="https://my.farm.bot"):
 
         self.token = token_data
 
+        # Set API tokens
         self.api.token = token_data
         self.api.api_connect.token = token_data
 
-        # self.broker.token = token_data
+        # Set broker tokens
         self.broker.broker_connect.token = token_data
         self.broker.api.api_connect.token = token_data
 
@@ -30,28 +31,28 @@ def set_info(self, endpoint, field, value, id=None):
 
     def env(self, id=None, field=None, new_val=None):
         return self.api.env(id, field, new_val)
-    
+
     def log(self, message, type=None, channel=None):
         return self.api.log(message, type, channel)
-    
+
     def safe_z(self):
         return self.api.safe_z()
-    
+
     def garden_size(self):
         return self.api.garden_size()
-    
+
     def group(self, id):
         return self.api.group(id)
-    
+
     def curve(self, id):
         return self.api.curve(id)
-    
+
     def read_status(self):
         return self.broker.read_status()
-    
+
     def read_sensor(self, id, mode, label='---'):
         return self.broker.read_sensor(id, mode, label)
-    
+
     def message(self, message, type=None, channel=None):
         return self.broker.message(message, type, channel)
 

testing.py

@@ -1,97 +1,30 @@
 import unittest
-import json
-from main import Farmbot
-from unittest.mock import patch, Mock
-
-class TestFarmbot(unittest.TestCase):
-
-    ## POSITIVE TEST: function called with email, password, and default server
-    @patch('requests.post')
-    def test_get_token_default_server(self, mock_post):
-        mock_response = Mock()
-        expected_token = {'token': 'abc123'}
-        mock_response.json.return_value = expected_token
-        mock_response.status_code = 200
-        mock_post.return_value = mock_response
-
-        fb = Farmbot()
-
-        # Call with default server
-        fb.get_token('[email protected]', 'test_pass_123')
-
-        mock_post.assert_called_once_with(
-            'https://my.farm.bot/api/tokens',
-            headers={'content-type': 'application/json'},
-            json={'user': {'email': '[email protected]', 'password': 'test_pass_123'}}
-        )
-
-        self.assertEqual(fb.token, expected_token)
-        self.assertEqual(mock_post.return_value.status_code, 200)
-
-    # POSITIVE TEST: function called with email, password, and custom server
-    @patch('requests.post')
-    def test_get_token_custom_server(self, mock_post):
-        mock_response = Mock()
-        expected_token = {'token': 'abc123'}
-        mock_response.json.return_value = expected_token
-        mock_response.status_code = 200
-        mock_post.return_value = mock_response
-
-        fb = Farmbot()
+from unittest.mock import MagicMock, patch
 
-        # Call with custom server
-        fb.get_token('[email protected]', 'test_pass_123', 'https://staging.farm.bot')
-
-        mock_post.assert_called_once_with(
-            'https://staging.farm.bot/api/tokens',
-            headers={'content-type': 'application/json'},
-            json={'user': {'email': '[email protected]', 'password': 'test_pass_123'}}
-        )
-
-        self.assertEqual(fb.token, expected_token)
-        self.assertEqual(mock_post.return_value.status_code, 200)
-
-    # NEGATIVE TEST: function called with bad email or password (HTTP error)
-    @patch('requests.post')
-    def test_get_token_bad_email(self, mock_post):
-        mock_response = Mock()
-        mock_response.status_code = 422
-        mock_post.return_value = mock_response
-
-        fb = Farmbot()
-
-        # Call with bad email
-        fb.get_token('[email protected]', 'test_pass_123', 'https://staging.farm.bot')
+from main import Farmbot
+from api_functions import ApiFunctions
+from broker_functions import BrokerFunctions
 
-        mock_post.assert_called_once_with(
-            'https://staging.farm.bot/api/tokens',
-            headers={'content-type': 'application/json'},
-            json={'user': {'email': '[email protected]', 'password': 'test_pass_123'}}
-        )
+class TestApiFunctions(unittest.TestCase):
 
-        self.assertIsNone(fb.token)
-        self.assertEqual(mock_post.return_value.status_code, 422)
+    @patch('main.ApiFunctions')
+    def setUp(self, MockApiFunctions):
+        self.mock_api = MockApiFunctions.return_value
+        self.farmbot = Farmbot()
 
-    # NEGATIVE TEST: function called with bad server address (HTTP error)
-    @patch('requests.post')
-    def test_get_token_bad_email(self, mock_post):
-        mock_response = Mock()
-        mock_response.status_code = 404
-        mock_post.return_value = mock_response
+    def test_get_token(self):
+        self.mock_api.get_token.return_value = 'mock_token'
+        email = '[email protected]'
+        password = 'password'
+        server = 'https://my.farm.bot'
 
-        fb = Farmbot()
-
-        # Call with bad email
-        fb.get_token('[email protected]', 'test_pass_123', 'https://bad.farm.bot')
-
-        mock_post.assert_called_once_with(
-            'https://bad.farm.bot/api/tokens',
-            headers={'content-type': 'application/json'},
-            json={'user': {'email': '[email protected]', 'password': 'test_pass_123'}}
-        )
+        token = self.farmbot.get_token(email, password, server)
 
-        self.assertIsNone(fb.token)
-        self.assertEqual(mock_post.return_value.status_code, 404)
+        self.assertEqual(token, 'mock_token')
+        self.assertEqual(self.farmbot.token, 'mock_token')
+        self.assertEqual(self.farmbot.api.token, 'mock_token')
+        self.assertEqual(self.farmbot.api.api_connect.token, 'mock_token')
+        self.mock_api.get_token.assert_called_with(email, password, server)
 
     # POSITIVE TEST: function called with endpoint only
     @patch('requests.get')
@@ -216,4 +149,4 @@ def test_get_info_with_id(self, mock_get):
 #         mock_curve.assert_called_once_with('seq', 0, 0, 10, 10, 5, 5)
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()