|
| 1 | +import json |
1 | 2 | from typing import Optional
|
2 | 3 | from unittest import IsolatedAsyncioTestCase
|
3 | 4 |
|
|
8 | 9 | from ...ledger.multiple_ledger.ledger_requests_executor import (
|
9 | 10 | IndyLedgerRequestsExecutor,
|
10 | 11 | )
|
| 12 | +from ...ledger.multiple_ledger.ledger_config_schema import ConfigurableWriteLedgersSchema |
| 13 | + |
11 | 14 | from ...multitenant.base import BaseMultitenantManager
|
12 | 15 | from ...multitenant.manager import MultitenantManager
|
13 | 16 | from ...tests import mock
|
@@ -865,3 +868,58 @@ async def test_get_ledger_config(self):
|
865 | 868 | async def test_get_ledger_config_x(self):
|
866 | 869 | with self.assertRaises(test_module.web.HTTPForbidden):
|
867 | 870 | await test_module.get_ledger_config(self.request)
|
| 871 | + |
| 872 | + # Multiple Ledgers Configured |
| 873 | + async def test_get_write_ledgers_multiple(self): |
| 874 | + # Mock the multiple ledger manager |
| 875 | + mock_manager = mock.MagicMock(BaseMultipleLedgerManager) |
| 876 | + mock_manager.get_write_ledgers = mock.CoroutineMock( |
| 877 | + return_value=["ledger1", "ledger2", "ledger3"] |
| 878 | + ) |
| 879 | + self.profile.context.injector.bind_instance( |
| 880 | + BaseMultipleLedgerManager, mock_manager |
| 881 | + ) |
| 882 | + |
| 883 | + with mock.patch.object( |
| 884 | + test_module.web, "json_response", mock.Mock() |
| 885 | + ) as json_response: |
| 886 | + result = await test_module.get_write_ledgers(self.request) |
| 887 | + |
| 888 | + # Assert the response matches the expected structure |
| 889 | + json_response.assert_called_once_with( |
| 890 | + {"write_ledgers": ["ledger1", "ledger2", "ledger3"]} |
| 891 | + ) |
| 892 | + assert result is json_response.return_value |
| 893 | + |
| 894 | + # Single Ledger (No Multi-Ledger Manager) |
| 895 | + async def test_get_write_ledgers_single(self): |
| 896 | + # Ensure no multi-ledger manager is bound |
| 897 | + self.profile.context.injector.clear_binding(BaseMultipleLedgerManager) |
| 898 | + |
| 899 | + result = await test_module.get_write_ledgers(self.request) |
| 900 | + |
| 901 | + # Extract the JSON body from the response |
| 902 | + response_body = result.text |
| 903 | + response_body = json.loads(response_body) |
| 904 | + |
| 905 | + # Assert the response is correct |
| 906 | + self.assertEqual(response_body, {"write_ledgers": ["default"]}) |
| 907 | + |
| 908 | + # Schema Validation |
| 909 | + async def test_get_write_ledgers_schema(self): |
| 910 | + # Mock the multiple ledger manager |
| 911 | + mock_manager = mock.MagicMock(BaseMultipleLedgerManager) |
| 912 | + mock_manager.get_write_ledgers = mock.CoroutineMock( |
| 913 | + return_value=["ledger1", "ledger2"] |
| 914 | + ) |
| 915 | + self.profile.context.injector.bind_instance( |
| 916 | + BaseMultipleLedgerManager, mock_manager |
| 917 | + ) |
| 918 | + |
| 919 | + response = await test_module.get_write_ledgers(self.request) |
| 920 | + |
| 921 | + # Validate against the schema |
| 922 | + schema = ConfigurableWriteLedgersSchema() |
| 923 | + data = json.loads(response.body) |
| 924 | + validated = schema.validate(data) |
| 925 | + assert validated == {} |
0 commit comments