def test_scalyr_timeseries_end(monkeypatch, begin_and_end): start, end = begin_and_end read_key = '123' post = MagicMock() post.return_value.json.return_value = dict({ 'status': 'success', 'results': [{ 'values': [1] }] }) monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) scalyr.timeseries('', minutes=start, end=end, align=0) query = get_query('facet', 'count', read_key, **{ 'minutes': start, 'end': end, 'align': 0 }) query.pop('queryType') final_q = {'token': query.pop('token'), 'queries': [query]} post.assert_called_with(scalyr._ScalyrWrapper__timeseries_url, json=final_q, headers={'Content-Type': 'application/json'})
def test_scalyr_facets(monkeypatch, fx_facets): kwargs, res = fx_facets read_key = '123' post = MagicMock() post.return_value.json.return_value = res monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) result = scalyr.facets(**kwargs) assert result == res if 'minutes' not in kwargs: kwargs['minutes'] = 30 query = get_query('facet', None, read_key, **kwargs) query.pop('function', None) query.pop('buckets', None) query['maxCount'] = kwargs.get('max_count', 5) post.assert_called_with(scalyr._ScalyrWrapper__facet_url, json=query, headers={'Content-Type': 'application/json'})
def test_scalyr_timeseries_aligned(monkeypatch, fx_timeseries_aligned): kwargs, res, exp = fx_timeseries_aligned read_key = '123' post = MagicMock() post.return_value.json.return_value = res monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) result = scalyr.timeseries(**kwargs) assert result == exp query = get_query('facet', kwargs.get('function', 'count'), read_key, **kwargs) query.pop('queryType') final_q = {'token': query.pop('token'), 'queries': [query]} post.assert_called_with(scalyr._ScalyrWrapper__timeseries_url, json=final_q, headers={'Content-Type': 'application/json'})
def test_scalyr_timeseries_aligned(monkeypatch, fx_timeseries_aligned): kwargs, res, exp = fx_timeseries_aligned read_key = '123' post = MagicMock() post.return_value.json.return_value = res monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) result = scalyr.timeseries(**kwargs) assert result == exp query = get_query('facet', kwargs.get('function', 'count'), read_key, **kwargs) query.pop('queryType') final_q = { 'token': query.pop('token'), 'queries': [query] } post.assert_called_with( scalyr._ScalyrWrapper__timeseries_url, json=final_q, headers={'Content-Type': 'application/json'})
def test_scalyr_power_query(monkeypatch, fx_power_query): kwargs, request, response, expected_exception = fx_power_query post = MagicMock() post.return_value.json.return_value = response monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(SCALYR_READ_KEY) try: result = scalyr.power_query(**kwargs) assert result == response assert expected_exception is None post.assert_called_with(scalyr._ScalyrWrapper__power_query_url, json=request, headers={ 'Content-Type': 'application/json', 'errorStatus': 'always200' }) except AssertionError: raise except Exception as error: assert isinstance(error, type(expected_exception)) assert error.args == expected_exception.args
def test_scalyr_facets(monkeypatch, fx_facets): kwargs, res = fx_facets read_key = '123' post = MagicMock() post.return_value.json.return_value = res monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) result = scalyr.facets(**kwargs) assert result == res if 'minutes' not in kwargs: kwargs['minutes'] = 30 query = get_query( 'facet', None, read_key, **kwargs) query.pop('function', None) query.pop('buckets', None) query['maxCount'] = kwargs.get('max_count', 5) post.assert_called_with( scalyr._ScalyrWrapper__facet_url, json=query, headers={'Content-Type': 'application/json'})
def test_scalyr_logs(monkeypatch, fx_logs): kwargs, res, exp = fx_logs read_key = '123' post = MagicMock() post.return_value.json.return_value = res monkeypatch.setattr('requests.post', post) def expected_query(): query = get_query('log', None, read_key, **kwargs) query.pop('function', None) query.pop('buckets', None) query['maxCount'] = kwargs.get('max_count', 100) if 'continuation_token' in kwargs: query['continuationToken'] = kwargs['continuation_token'] if 'columns' in kwargs: query['columns'] = ','.join(kwargs['columns']) if type(kwargs['columns']) is list else kwargs['columns'] return query scalyr = ScalyrWrapper(read_key) try: result = scalyr.logs(**kwargs) assert result == exp query = expected_query() post.assert_called_with( scalyr._ScalyrWrapper__query_url, json=query, headers={'Content-Type': 'application/json', 'errorStatus': 'always200'}) except CheckError as e: if not res: assert '{}'.format(e) == 'query "{}" is not allowed to be blank'.format(kwargs['query']) post.assert_not_called elif 'message' in res: assert '{}'.format(e) == res['message'] query = expected_query() post.assert_called_with( scalyr._ScalyrWrapper__query_url, json=query, headers={'Content-Type': 'application/json', 'errorStatus': 'always200'}) elif not exp: assert '{}'.format(e) == 'No logs or error message was returned from scalyr' query = expected_query() post.assert_called_with( scalyr._ScalyrWrapper__query_url, json=query, headers={'Content-Type': 'application/json', 'errorStatus': 'always200'}) else: raise
def test_scalyr_function(monkeypatch, fx_function): kwargs, res, exp = fx_function read_key = '123' post = MagicMock() post.return_value.json.return_value = res monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) count = scalyr.function(**kwargs) assert count == exp query = get_query('numeric', kwargs['function'], read_key, **kwargs) post.assert_called_with( scalyr._ScalyrWrapper__numeric_url, json=query, headers={'Content-Type': 'application/json'})
def test_scalyr_http_error(monkeypatch, method): post = MagicMock() post.return_value.raise_for_status.side_effect = requests.exceptions.HTTPError monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper('123') m, args = method f = getattr(scalyr, m) with pytest.raises(requests.exceptions.HTTPError): f(*args)
def test_scalyr_eu_region(): read_key = '123' region = 'eu' numeric_url = 'https://eu.scalyr.com/api/numericQuery' timeseries_url = 'https://eu.scalyr.com/api/timeseriesQuery' facet_url = 'https://eu.scalyr.com/api/facetQuery' scalyr = ScalyrWrapper(read_key, region) assert numeric_url == scalyr._ScalyrWrapper__numeric_url assert timeseries_url == scalyr._ScalyrWrapper__timeseries_url assert facet_url == scalyr._ScalyrWrapper__facet_url
def test_scalyr_timeseries_end(monkeypatch, begin_and_end): start, end = begin_and_end read_key = '123' post = MagicMock() post.return_value.json.return_value = dict({'status': 'success', 'results': [{'values': [1]}]}) monkeypatch.setattr('requests.post', post) scalyr = ScalyrWrapper(read_key) scalyr.timeseries('', minutes=start, end=end, align=0) query = get_query('facet', 'count', read_key, **{'minutes': start, 'end': end, 'align': 0}) query.pop('queryType') final_q = { 'token': query.pop('token'), 'queries': [query] } post.assert_called_with( scalyr._ScalyrWrapper__timeseries_url, json=final_q, headers={'Content-Type': 'application/json'})
def test_scalyr_config_error(monkeypatch): with pytest.raises(ConfigurationError): ScalyrWrapper('')