def test_create_scheduled_query_pass(self, mock_login, mock_post):
"""
Test ._create_scheduled_query() passes
"""
# simulate login
mock_login.return_value = self.account_id
# Create the response
mock_response = Mock(spec=requests.Response)
mock_response.status_code = 201
mock_response.json.return_value = {
'scheduled_query': {
'id': '00000000-0000-03a2-0000-000000000000'
}
}
mock_response.ok = True
mock_post.return_value = mock_response
session = requests.session()
alert = AnomalyAlert(self.username, self.password, session)
query = 'where(status=404) calculate(COUNT)'
scope_count = 1
scope_unit = 'day'
change = '+15'
# Call .create
response = alert._create_scheduled_query(
query=query,
change=change,
scope_count=scope_count,
scope_unit=scope_unit,
)
self.assertDictEqual(
response,
{
'scheduled_query': {
'id': '00000000-0000-03a2-0000-000000000000'
}
}
)
