def test_bootstrap_ignore_remote_caching(self, requests): requests.get.return_value = mock.Mock() requests.get.return_value.json.return_value = { 'hey': { 'methods': ['IGNORE'] } } app = mock.Mock() app.extensions = {'symbolic_ratelimits': {}} app.app_context = mock.mock_open() app.config = {'WEBSERVICES_DISCOVERY_CACHE_DIR': ''} with mock.patch("__builtin__.open", mock.mock_open()) as mock_file: discovery_utils.bootstrap_remote_service( 'http://foo.bar-us-east-1.com:8980/tee', '/foo', app) requests.get.assert_called_with( 'http://foo.bar-us-east-1.com:8980/', timeout=5) try: mock_file.assert_called_with( '/tmp/http:foobar-us-east-1com:8980tee', 'w') raise Exception('The file was opened!') except AssertionError: pass
def test_bootstrap_from_cache(self, m_json, requests, m_os): requests.get.side_effect = ConnectionError() requests.exceptions.ConnectionError = ConnectionError app = mock.Mock() app.app_context = mock.mock_open() app.config = {'WEBSERVICES_DISCOVERY_CACHE_DIR' : '/tmp'} m_os.path.join = os.path.join m_os.path.exists.return_value = True with mock.patch("__builtin__.open", mock.mock_open(read_data='{"hey": {"methods": ["IGNORE"]}}')) as mock_file: discovery_utils.bootstrap_remote_service('http://foo.bar-us-east-1.com:8980/tee', '/foo', app) requests.get.assert_called_with('http://foo.bar-us-east-1.com:8980/', timeout=5) m_os.path.exists.called_with('/tmp/http:foobar-us-east-1com:8980tee') m_json.loads.assert_called_with('{"hey": {"methods": ["IGNORE"]}}')
def test_bootstrap_from_cache(self, m_json, requests, m_os): requests.get.side_effect = ConnectionError() requests.exceptions.ConnectionError = ConnectionError app = mock.Mock() app.app_context = mock.mock_open() app.config = {'WEBSERVICES_DISCOVERY_CACHE_DIR' : '/tmp'} m_os.path.join = os.path.join m_os.path.exists.return_value = True with mock.patch("__builtin__.open", mock.mock_open(read_data='{"hey": {"methods": ["IGNORE"]}}')) as mock_file: discovery_utils.bootstrap_remote_service('http://foo.bar-us-east-1.com:8980/tee', '/foo', app) requests.get.assert_called_with('http://foo.bar-us-east-1.com:8980/', timeout=5) m_os.path.exists.called_with('/tmp/http:foobar-us-east-1com:8980tee') m_json.loads.assert_called_with('{"hey": {"methods": ["IGNORE"]}}')
def test_bootstrap_remote_caching(self, requests): requests.get.return_value = mock.Mock() requests.get.return_value.json.return_value = {'hey': {'methods': ['IGNORE']}} app = mock.Mock() app.app_context = mock.mock_open() app.config = {'WEBSERVICES_DISCOVERY_CACHE_DIR' : '/tmp'} with mock.patch("__builtin__.open", mock.mock_open()) as mock_file: discovery_utils.bootstrap_remote_service('http://foo.bar-us-east-1.com:8980/tee', '/foo', app) requests.get.assert_called_with('http://foo.bar-us-east-1.com:8980/', timeout=5) mock_file.assert_called_with('/tmp/http:foobar-us-east-1com:8980tee', 'w') file_handle = mock_file.return_value.__enter__.return_value file_handle.write.assert_called_with('{"hey": {"methods": ["IGNORE"]}}')
def test_bootstrap_remote_caching(self, requests): requests.get.return_value = mock.Mock() requests.get.return_value.json.return_value = {'hey': {'methods': ['IGNORE']}} app = mock.Mock() app.app_context = mock.mock_open() app.config = {'WEBSERVICES_DISCOVERY_CACHE_DIR' : '/tmp'} with mock.patch("__builtin__.open", mock.mock_open()) as mock_file: discovery_utils.bootstrap_remote_service('http://foo.bar-us-east-1.com:8980/tee', '/foo', app) requests.get.assert_called_with('http://foo.bar-us-east-1.com:8980/', timeout=5) mock_file.assert_called_with('/tmp/http:foobar-us-east-1com:8980tee', 'w') file_handle = mock_file.return_value.__enter__.return_value file_handle.write.assert_called_with('{"hey": {"methods": ["IGNORE"]}}')
def test_bootstrap_ignore_remote_caching(self, requests): requests.get.return_value = mock.Mock() requests.get.return_value.json.return_value = {'hey': {'methods': ['IGNORE']}} app = mock.Mock() app.app_context = mock.mock_open() app.config = {'WEBSERVICES_DISCOVERY_CACHE_DIR' : ''} with mock.patch("__builtin__.open", mock.mock_open()) as mock_file: discovery_utils.bootstrap_remote_service('http://foo.bar-us-east-1.com:8980/tee', '/foo', app) requests.get.assert_called_with('http://foo.bar-us-east-1.com:8980/', timeout=5) try: mock_file.assert_called_with('/tmp/http:foobar-us-east-1com:8980tee', 'w') raise Exception('The file was opened!') except AssertionError: pass