Update InitializeServiceMixin

 * Add tests for InitializeServiceMixin

Author: Scandie

openprocurement/auction/worker/mixins.py

@@ -7,7 +7,7 @@
 from datetime import datetime, timedelta
 from copy import deepcopy
 from dateutil.tz import tzlocal
-from requests import request
+from requests import request, Session as RequestsSession
 from yaml import safe_dump as yaml_dump
 from couchdb.http import HTTPError, RETRYABLE_ERRORS
 from fractions import Fraction
@@ -687,7 +687,7 @@ def init_api(self):
             response = make_request(url=api_url.format(**self.worker_defaults),
                                     method="get", retry_count=5)
         if not response:
-            raise Exception("Auction DS can't be reached")
+            raise Exception("API can't be reached")
         else:
             self.tender_url = urljoin(
                 self.worker_defaults["resource_api_server"],
@@ -703,9 +703,7 @@ def init_ds(self):
         Check Document service availability and set session_ds attribute
         """
         ds_config = self.worker_defaults.get("DOCUMENT_SERVICE")
-        resp = request("GET", ds_config.get("url"), timeout=5)
-        if not resp or resp.status_code != 200:
-            raise Exception("Auction DS can't be reached")
+        request("GET", ds_config.get("url"), timeout=5)
         self.session_ds = RequestsSession()
 
     def init_database(self):

openprocurement/auction/worker/tests/unit/conftest.py

@@ -21,7 +21,6 @@
 from openprocurement.auction.worker.server import (
     app as worker_app, BidsForm
 )
-# from openprocurement.auction.tests.functional.main import update_auctionPeriod
 
 
 def update_auctionPeriod(data):
@@ -35,8 +34,14 @@ def update_auctionPeriod(data):
 PWD = os.path.dirname(os.path.realpath(__file__))
 
 worker_defaults_file_path = os.path.join(PWD, "../data/auction_worker_defaults.yaml")
-with open(worker_defaults_file_path) as stream:
-    worker_defaults = yaml.load(stream)
+
+
+@pytest.yield_fixture(scope='function')
+def worker_config():
+    update_auctionPeriod(tender_data)
+    with open(worker_defaults_file_path) as stream:
+        worker_defaults = yaml.load(stream)
+    return worker_defaults
 
 
 @pytest.yield_fixture(
@@ -57,13 +62,13 @@ def universal_auction(request):
         lot_id=request.param['lot_id']
     )
 
+
 @pytest.yield_fixture(scope="function")
-def auction():
-    update_auctionPeriod(tender_data)
+def auction(worker_config):
 
     yield Auction(
         tender_id=tender_data['data']['auctionID'],
-        worker_defaults=yaml.load(open(worker_defaults_file_path)),
+        worker_defaults=worker_config,
         auction_data=tender_data,
         lot_id=False
     )
@@ -90,9 +95,9 @@ def features_auction():
 
 
 @pytest.fixture(scope='function')
-def db(request):
-    server = couchdb.Server("http://" + worker_defaults['COUCH_DATABASE'].split('/')[2])
-    name = worker_defaults['COUCH_DATABASE'].split('/')[3]
+def db(request, worker_config):
+    server = couchdb.Server("http://" + worker_config['COUCH_DATABASE'].split('/')[2])
+    name = worker_config['COUCH_DATABASE'].split('/')[3]
 
     def delete():
         del server[name]

openprocurement/auction/worker/tests/unit/test_initialization.py

@@ -0,0 +1,55 @@
+from copy import deepcopy
+from uuid import uuid4
+
+import pytest
+from requests import exceptions
+
+from openprocurement.auction.worker.auction import Auction
+from openprocurement.auction.worker.mixins import Server
+
+
+def test_init_services(worker_config, logger, mocker):
+
+    test_config = deepcopy(worker_config)
+    test_config['with_document_service'] = True
+    test_config['DOCUMENT_SERVICE']['url'] = "http://1.2.3.4:6543/"
+    test_config['resource_api_server'] = "http://1.2.3.4:6543/"
+
+    tender_id = uuid4().hex  # random tender id
+
+    mock_make_request = mocker.MagicMock()
+    mock_make_request.side_effect = Exception("API can't be reached")
+    mocker.patch('openprocurement.auction.worker.mixins.make_request', mock_make_request)
+
+    with pytest.raises(Exception) as e:
+        Auction(tender_id, test_config)
+    log_strings = logger.log_capture_string.getvalue().split('\n')
+    assert e.value.message == "API can't be reached"
+
+    assert log_strings.count("API can't be reached") == 1
+    assert "ConnectTimeout" in log_strings[-2]
+
+    with pytest.raises(exceptions.RequestException) as e:
+        Auction(tender_id, test_config, {'test_auction_data': True})
+    log_strings = logger.log_capture_string.getvalue().split('\n')
+    assert "ConnectTimeout" in str(e.value.message)
+
+    assert log_strings.count("API can't be reached") == 1
+    assert "ConnectTimeout" in log_strings[-2]
+
+    test_config['with_document_service'] = False
+    auction = Auction(tender_id, test_config, {'test_auction_data': True})
+
+    assert auction.__getattribute__("tender_url") is not None
+    assert auction.__getattribute__("db") is not None
+
+    mock_server_connect = mocker.patch.object(Server, "__init__", autospec=True)
+    mock_server_connect.side_effect = Exception("Connection refused")
+
+    with pytest.raises(Exception) as e:
+        Auction(tender_id, test_config, {'test_auction_data': True})
+    log_strings = logger.log_capture_string.getvalue().split('\n')
+    assert e.value.message == "Connection refused"
+
+    assert log_strings.count("API can't be reached") == 1
+    assert log_strings.count("Connection refused") == 1