Skip to content

Mock Session to remove dependency on real sandbox #94

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
leits merged 1 commit into from
Jun 21, 2018
Merged

Mock Session to remove dependency on real sandbox #94

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 28 additions & 14 deletions openprocurement_client/tests/tests_sync.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,8 @@ def test_instance_initialization(self):
})
self.assertIsInstance(self.resource_feeder.queue, Queue)

def test_init_api_clients(self):
@mock.patch('openprocurement_client.templates.Session')
def test_init_api_clients(self, mock_session):
self.resource_feeder = ResourceFeeder()
self.resource_feeder.init_api_clients()
self.assertEqual(self.resource_feeder.backward_params, {
Expand All @@ -202,10 +203,11 @@ def test_handle_response_data(self):
self.assertIn('tender2', list(self.resource_feeder.queue.queue))
self.assertNotIn('tender3', list(self.resource_feeder.queue.queue))

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.clients.APIResourceClientSync.'
'sync_resource_items')
@mock.patch('openprocurement_client.resources.sync.spawn')
def test_start_sync(self, mock_spawn, mock_sync_resource_items):
def test_start_sync(self, mock_spawn, mock_sync_resource_items, mock_session):
mock_spawn.return_value = 'spawn result'
mock_sync_resource_items.return_value = self.response
self.resource_feeder = ResourceFeeder()
Expand All @@ -222,10 +224,11 @@ def test_start_sync(self, mock_spawn, mock_sync_resource_items):
self.assertEqual(self.resource_feeder.backward_worker, 'spawn result')
self.assertEqual(self.resource_feeder.forward_worker, 'spawn result')

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.clients.APIResourceClientSync.'
'sync_resource_items')
@mock.patch('openprocurement_client.resources.sync.spawn')
def test_restart_sync(self, mock_spawn, mock_sync_resource_items):
def test_restart_sync(self, mock_spawn, mock_sync_resource_items, mock_session):
mock_spawn.return_value = mock.MagicMock()
mock_spawn.return_value.kill = mock.MagicMock('kill result')
mock_sync_resource_items.return_value = self.response
Expand All @@ -235,11 +238,12 @@ def test_restart_sync(self, mock_spawn, mock_sync_resource_items):
self.resource_feeder.restart_sync()
self.assertEqual(mock_spawn.return_value.kill.call_count, 2)

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.clients.APIResourceClientSync.'
'sync_resource_items')
@mock.patch('openprocurement_client.resources.sync.spawn')
def test_get_resource_items_zero_value(self, mock_spawn,
mock_sync_resource_items):
mock_sync_resource_items, mock_session):
mock_sync_resource_items.side_effect = [self.response, munchify(
{'data': {},
'next_page': {'offset': 'next_page'},
Expand All @@ -253,11 +257,12 @@ def test_get_resource_items_zero_value(self, mock_spawn,
result = self.resource_feeder.get_resource_items()
self.assertEqual(tuple(result), tuple(self.response.data))

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.clients.APIResourceClientSync.'
'sync_resource_items')
@mock.patch('openprocurement_client.resources.sync.spawn')
def test_get_resource_items_non_zero_value(self, mock_spawn,
mock_sync_resource_items):
mock_sync_resource_items, mock_session):
mock_sync_resource_items.side_effect = [self.response, munchify(
{'data': {},
'next_page': {'offset': 'next_page'},
Expand All @@ -271,12 +276,13 @@ def test_get_resource_items_non_zero_value(self, mock_spawn,
result = self.resource_feeder.get_resource_items()
self.assertEqual(tuple(result), tuple(self.response.data))

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.clients.APIResourceClientSync.'
'sync_resource_items')
@mock.patch('openprocurement_client.resources.sync.spawn')
@mock.patch('openprocurement_client.resources.sync.sleep')
def test_feeder_zero_value(self, mock_sleep, mock_spawn,
mock_sync_resource_items):
mock_sync_resource_items, mock_session):
mock_sleep.return_value = 'sleeping'
mock_sync_resource_items.side_effect = [self.response, self.response,
ConnectionError('conn error')]
Expand All @@ -289,11 +295,12 @@ def test_feeder_zero_value(self, mock_sleep, mock_spawn,
self.assertEqual(e.exception.message, 'conn error')
self.assertEqual(mock_sleep.call_count, 1)

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.clients.APIResourceClientSync.'
'sync_resource_items')
@mock.patch('openprocurement_client.resources.sync.spawn')
@mock.patch('openprocurement_client.resources.sync.sleep')
def test_feeder(self, mock_sleep, mock_spawn, mock_sync_resource_items):
def test_feeder(self, mock_sleep, mock_spawn, mock_sync_resource_items, mock_session):
mock_sleep.return_value = 'sleeping'
mock_sync_resource_items.side_effect = [self.response, self.response,
ConnectionError('conn error')]
Expand All @@ -314,8 +321,9 @@ def test_run_feeder(self, mock_spawn):
mock_spawn.assert_called_with(self.resource_feeder.feeder)
self.assertEqual(result, self.resource_feeder.queue)

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_backward(self, mock_get_response):
def test_retriever_backward(self, mock_get_response, mock_session):
mock_get_response.side_effect = [self.response, munchify({'data': {}})]
self.resource_feeder = ResourceFeeder()
self.resource_feeder.init_api_clients()
Expand All @@ -327,8 +335,9 @@ def test_retriever_backward(self, mock_get_response):
self.assertEqual(self.resource_feeder.backward_params['offset'],
self.response.next_page.offset)

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_backward_wrong_cookies(self, mock_get_response):
def test_retriever_backward_wrong_cookies(self, mock_get_response, mock_session):
mock_get_response.return_value = self.response
self.resource_feeder = ResourceFeeder()
self.resource_feeder.init_api_clients()
Expand All @@ -340,8 +349,9 @@ def test_retriever_backward_wrong_cookies(self, mock_get_response):
self.resource_feeder.retriever_backward()
self.assertEqual(e.exception.message, 'LB Server mismatch')

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_forward_wrong_cookies(self, mock_get_response):
def test_retriever_forward_wrong_cookies(self, mock_get_response, mock_session):
mock_get_response.return_value = self.response
self.resource_feeder = ResourceFeeder()
self.resource_feeder.init_api_clients()
Expand All @@ -353,8 +363,9 @@ def test_retriever_forward_wrong_cookies(self, mock_get_response):
self.resource_feeder.retriever_forward()
self.assertEqual(e.exception.message, 'LB Server mismatch')

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_forward(self, mock_get_response):
def test_retriever_forward(self, mock_get_response, mock_session):
mock_get_response.side_effect = [
self.response,
self.response,
Expand All @@ -372,8 +383,9 @@ def test_retriever_forward(self, mock_get_response):
self.assertEqual(self.resource_feeder.forward_params['offset'],
self.response.next_page.offset)

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_forward_no_data(self, mock_get_response):
def test_retriever_forward_no_data(self, mock_get_response, mock_session):
mock_get_response.side_effect = [
munchify({'data': {}, 'next_page': {'offset': 'next_page'}}),
ConnectionError('connection error')
Expand All @@ -390,8 +402,9 @@ def test_retriever_forward_no_data(self, mock_get_response):
self.assertEqual(self.resource_feeder.forward_params['offset'],
'next_page')

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_forward_adaptive(self, mock_get_response):
def test_retriever_forward_adaptive(self, mock_get_response, mock_session):
mock_get_response.side_effect = [
self.response,
munchify({'data': {}, 'next_page': {'offset': 'next_page'}}),
Expand All @@ -410,8 +423,9 @@ def test_retriever_forward_adaptive(self, mock_get_response):
self.assertEqual(self.resource_feeder.forward_params['offset'],
self.response.next_page.offset)

@mock.patch('openprocurement_client.templates.Session')
@mock.patch('openprocurement_client.resources.sync.get_response')
def test_retriever_forward_no_data_adaptive(self, mock_get_response):
def test_retriever_forward_no_data_adaptive(self, mock_get_response, mock_session):
mock_get_response.side_effect = [
munchify({'data': {}, 'next_page': {'offset': 'next_page'}}),
munchify({'data': {}, 'next_page': {'offset': 'next_page'}}),
Expand Down