|
1 | | -import simplejson as json |
| 1 | +import json |
2 | 2 | from tests.integration.test_base import QuickbooksUnitTestCase |
3 | 3 |
|
4 | 4 | try: |
|
7 | 7 | from unittest.mock import patch, mock_open |
8 | 8 |
|
9 | 9 | from quickbooks.exceptions import QuickbooksException, SevereException, AuthorizationException |
10 | | -from quickbooks import client |
| 10 | +from quickbooks import client, mixins |
11 | 11 | from quickbooks.objects.salesreceipt import SalesReceipt |
12 | 12 |
|
13 | 13 |
|
@@ -142,7 +142,7 @@ def test_get_single_object(self, make_req): |
142 | 142 |
|
143 | 143 | @patch('quickbooks.client.QuickBooks.process_request') |
144 | 144 | def test_make_request(self, process_request): |
145 | | - process_request.return_value = MockResponseSimpleJson() |
| 145 | + process_request.return_value = MockResponseJson() |
146 | 146 |
|
147 | 147 | qb_client = client.QuickBooks() |
148 | 148 | qb_client.company_id = "1234" |
@@ -221,7 +221,7 @@ def test_download_pdf_not_authorized(self, process_request): |
221 | 221 | @patch('quickbooks.client.QuickBooks.process_request') |
222 | 222 | def test_make_request_file_closed(self, process_request): |
223 | 223 | file_path = '/path/to/file.txt' |
224 | | - process_request.return_value = MockResponseSimpleJson() |
| 224 | + process_request.return_value = MockResponseJson() |
225 | 225 | with patch('builtins.open', mock_open(read_data=b'file content')) as mock_file: |
226 | 226 | qb_client = client.QuickBooks(auth_client=self.auth_client) |
227 | 227 | qb_client.make_request('POST', |
@@ -254,14 +254,14 @@ def json(self): |
254 | 254 | def content(self): |
255 | 255 | return '' |
256 | 256 |
|
257 | | -class MockResponseSimpleJson: |
| 257 | +class MockResponseJson: |
258 | 258 | def __init__(self, json_data=None, status_code=200): |
259 | 259 | self.json_data = json_data or {} |
260 | 260 | self.status_code = status_code |
261 | 261 |
|
262 | 262 | @property |
263 | 263 | def text(self): |
264 | | - return json.dumps(self.json_data) # Ensure this uses simplejson if necessary |
| 264 | + return json.dumps(self.json_data, cls=mixins.DecimalEncoder) |
265 | 265 |
|
266 | 266 | def json(self): |
267 | 267 | return self.json_data |
|
0 commit comments