-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_api.py
39 lines (29 loc) · 1.4 KB
/
test_api.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import unittest
from unittest.mock import patch, MagicMock
import api
class TestJiraAPIProgram(unittest.TestCase):
@patch('api.load_dotenv')
def test_load_env_variables(self, mock_load_dotenv):
api.load_dotenv()
mock_load_dotenv.assert_called_once()
def test_create_request_url(self):
test_endpoint = "/test"
expected_url = f"{api.JIRA_SITE_URL}{test_endpoint}"
self.assertEqual(api.create_request_url(test_endpoint), expected_url)
def test_create_request_headers_server(self):
expected_headers = {'Authorization': f'Bearer {api.JIRA_API_TOKEN}'}
self.assertEqual(api.create_request_headers_server(), expected_headers)
@patch('api.safe_requests.get')
def test_execute_request_server_success(self, mock_get):
mock_get.return_value = MagicMock(status_code=200, json=lambda: {"total": 42})
response = api.execute_request_server("http://test.com", {}, {})
self.assertIsNotNone(response)
self.assertEqual(response, {"total": 42})
@patch('api.safe_requests.get')
def test_execute_request_server_failure(self, mock_get):
mock_get.side_effect = api.requests.HTTPError()
response = api.execute_request_server("http://test.com", {}, {})
self.assertIsNone(response)
# Add more tests for handle_request_error, get_jql_query_results, etc.
if __name__ == '__main__':
unittest.main()