def test_check_that_alert_matches_zone(self):
"""When we match a zone and all other criteria, we should create an alert."""
history = [{
u'Timestamp':
datetime.datetime(2015, 12, 31, 22, 13, 43, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.105200',
u'AvailabilityZone':
'us-east-1d'
}, {
u'Timestamp':
datetime.datetime(2015, 12, 31, 21, 56, 18, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.104400',
u'AvailabilityZone':
'us-east-1d'
}]
match_1d = {
'name': 'Sub for just 1d',
'threshold': '0.05',
'region': 'us-east-1',
'zone': 'us-east-1d',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
assert not alert.check_for_alert(
history,
match_1d) is None, "There should be an alert from match_1d"
match_1q = {
'name': 'Sub for 1q',
'threshold': '0.05',
'region': 'us-east-1',
'zone': 'us-east-1q',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
assert alert.check_for_alert(
history,
match_1q) is None, "There should not be any alerts for us_east-1q"
def test_check_for_alert_with_no_matched_zones(self):
"""Alerts are only valid if the availability zone in the history matches an availability zone in the subscription"""
history = [{u'Timestamp': datetime.datetime(2015, 12, 31, 22, 13, 43,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.105200',
u'AvailabilityZone': 'us-east-1d'},
{u'Timestamp': datetime.datetime(2015, 12, 31, 21, 56, 18,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.104400',
u'AvailabilityZone': 'us-east-1d'}]
just_1a = {'name': 'Just 1a',
'threshold':'0.05',
'region':'us-east-1',
'zone': 'us-east-1a',
'instance_type':'g2.2xlarge',
'product':'Windows',
'user':'******',
'last_alert':'Under'}
result = alert.check_for_alert(history, just_1a)
assert result is None, 'There should not be an alert for Just 1a'
def test_check_for_alert_sets_spotprice(self):
"""check_for_alert should set the last_alert attribute of the alert to indication the type of the alert."""
history = [{
u'Timestamp':
datetime.datetime(2015, 12, 31, 22, 13, 43, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.105200',
u'AvailabilityZone':
'us-east-1b'
}]
subscription = {
'name': 'Sub for 1b',
'threshold': '0.05',
'region': 'us-east-1',
'zone': 'us-east-1b',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
result = alert.check_for_alert(history, subscription)
assert not result is None, "There should be an alert for us_east-1b"
assert result['spot_price'] == 0.105200
def test_check_that_alert_matches_zone(self):
"""When we match a zone and all other criteria, we should create an alert."""
history = [{u'Timestamp': datetime.datetime(2015, 12, 31, 22, 13, 43,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.105200',
u'AvailabilityZone': 'us-east-1d'},
{u'Timestamp': datetime.datetime(2015, 12, 31, 21, 56, 18,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.104400',
u'AvailabilityZone': 'us-east-1d'}]
match_1d = {'name': 'Sub for just 1d',
'threshold':'0.05',
'region':'us-east-1',
'zone': 'us-east-1d',
'instance_type':'g2.2xlarge',
'product':'Windows',
'user':'******',
'last_alert':'Under'}
assert not alert.check_for_alert(history, match_1d) is None, "There should be an alert from match_1d"
match_1q = {'name': 'Sub for 1q',
'threshold':'0.05',
'region':'us-east-1',
'zone': 'us-east-1q',
'instance_type':'g2.2xlarge',
'product':'Windows',
'user':'******',
'last_alert':'Under'}
assert alert.check_for_alert(history, match_1q) is None, "There should not be any alerts for us_east-1q"
def test_check_for_alert_sets_last_alert(self):
"""check_for_alert should set the last_alert attribute of the alert to indication the type of the alert."""
history = [ {u'Timestamp': datetime.datetime(2015, 12, 31, 22, 13, 43,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.105200',
u'AvailabilityZone': 'us-east-1b'}]
subscription = {'name': 'Sub for 1b',
'threshold':'0.05',
'region':'us-east-1',
'zone': 'us-east-1b',
'instance_type':'g2.2xlarge',
'product':'Windows',
'user':'******',
'last_alert':'Under'}
result = alert.check_for_alert(history, subscription)
assert not result is None, "There should be an alert for us_east-1b"
assert result['last_alert'] == 'Over'
def test_check_for_alert_with_no_matched_zones(self):
"""Alerts are only valid if the availability zone in the history matches an availability zone in the subscription"""
history = [{
u'Timestamp':
datetime.datetime(2015, 12, 31, 22, 13, 43, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.105200',
u'AvailabilityZone':
'us-east-1d'
}, {
u'Timestamp':
datetime.datetime(2015, 12, 31, 21, 56, 18, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.104400',
u'AvailabilityZone':
'us-east-1d'
}]
just_1a = {
'name': 'Just 1a',
'threshold': '0.05',
'region': 'us-east-1',
'zone': 'us-east-1a',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
result = alert.check_for_alert(history, just_1a)
assert result is None, 'There should not be an alert for Just 1a'
def test_check_for_alert_over_under(self):
""" Test that we can match an alert description against relevant history. """
history = [ {u'Timestamp': datetime.datetime(2015, 12, 31, 22, 13, 43,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.105200',
u'AvailabilityZone': 'us-east-1b'},
{u'Timestamp': datetime.datetime(2015, 12, 31, 21, 56, 18,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.104400',
u'AvailabilityZone': 'us-east-1d'},
{u'Timestamp': datetime.datetime(2015, 12, 31, 21, 56, 18,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.106300',
u'AvailabilityZone': 'us-east-1c'},
{u'Timestamp': datetime.datetime(2015, 12, 31, 21, 31, 6,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.767100',
u'AvailabilityZone': 'us-east-1e'},
{u'Timestamp': datetime.datetime(2015, 12, 31, 21, 29, 47,
tzinfo=tzutc()),
u'ProductDescription': 'Windows',
u'InstanceType': 'g2.2xlarge',
u'SpotPrice': '0.105300',
u'AvailabilityZone': 'us-east-1b'},
]
virginia_under_a_nickle = {'name': 'Virginia Under A Nickle',
'threshold':'0.05', 'region':'us-east-1', 'zone':
'us-east-1b', 'instance_type':'g2.2xlarge',
'product':'Windows', 'user':'******', 'last_alert':'Under'}
virginia_over_twenty = {'name': 'Virginia Over Twenty',
'threshold':'0.2', 'region':'us-east-1', 'zone':
'us-east-1b', 'instance_type':'g2.2xlarge',
'product':'Windows', 'user':'******', 'last_alert':'Over'}
assert not alert.check_for_alert(history, virginia_under_a_nickle) is None, "Should see an alert for Virginia Under A Nickle"
assert not alert.check_for_alert(history, virginia_over_twenty) is None, "Should see an alert for Virginia Over Twenty"
dublin_under_twenty = {'name': 'Dublin Under Twenty',
'threshold':'0.2', 'region':'eu-west-1', 'zone': 'eu-west-1b',
'instance_type':'g2.2xlarge', 'product':'Windows', 'user':'******',
'last_alert':'Under'}
virginia_under_twenty = {'name': 'Virginia Under Twenty',
'threshold':'0.2', 'region':'us-east-1', 'zone': 'us-east-1b',
'instance_type':'g2.2xlarge', 'product':'Windows', 'user':'******',
'last_alert':'Under'}
assert alert.check_for_alert(history, dublin_under_twenty) is None, "Should not see an alert for Dublin Under Twenty"
assert alert.check_for_alert(history, virginia_under_twenty) is None, "Should not see an alert for Virginia Under Twenty"
def test_check_for_alert_empty_history_and_subscriptions(self):
""" Test that we handle no history and no subscriptions in a sane way."""
assert alert.check_for_alert([],None) is None, "Alerts should have been an empty list."
def test_check_for_alert_over_under(self):
""" Test that we can match an alert description against relevant history. """
history = [
{
u'Timestamp':
datetime.datetime(2015, 12, 31, 22, 13, 43, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.105200',
u'AvailabilityZone':
'us-east-1b'
},
{
u'Timestamp':
datetime.datetime(2015, 12, 31, 21, 56, 18, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.104400',
u'AvailabilityZone':
'us-east-1d'
},
{
u'Timestamp':
datetime.datetime(2015, 12, 31, 21, 56, 18, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.106300',
u'AvailabilityZone':
'us-east-1c'
},
{
u'Timestamp':
datetime.datetime(2015, 12, 31, 21, 31, 6, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.767100',
u'AvailabilityZone':
'us-east-1e'
},
{
u'Timestamp':
datetime.datetime(2015, 12, 31, 21, 29, 47, tzinfo=tzutc()),
u'ProductDescription':
'Windows',
u'InstanceType':
'g2.2xlarge',
u'SpotPrice':
'0.105300',
u'AvailabilityZone':
'us-east-1b'
},
]
virginia_under_a_nickle = {
'name': 'Virginia Under A Nickle',
'threshold': '0.05',
'region': 'us-east-1',
'zone': 'us-east-1b',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
virginia_over_twenty = {
'name': 'Virginia Over Twenty',
'threshold': '0.2',
'region': 'us-east-1',
'zone': 'us-east-1b',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Over'
}
assert not alert.check_for_alert(
history, virginia_under_a_nickle
) is None, "Should see an alert for Virginia Under A Nickle"
assert not alert.check_for_alert(
history, virginia_over_twenty
) is None, "Should see an alert for Virginia Over Twenty"
dublin_under_twenty = {
'name': 'Dublin Under Twenty',
'threshold': '0.2',
'region': 'eu-west-1',
'zone': 'eu-west-1b',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
virginia_under_twenty = {
'name': 'Virginia Under Twenty',
'threshold': '0.2',
'region': 'us-east-1',
'zone': 'us-east-1b',
'instance_type': 'g2.2xlarge',
'product': 'Windows',
'user': '******',
'last_alert': 'Under'
}
assert alert.check_for_alert(
history, dublin_under_twenty
) is None, "Should not see an alert for Dublin Under Twenty"
assert alert.check_for_alert(
history, virginia_under_twenty
) is None, "Should not see an alert for Virginia Under Twenty"
def test_check_for_alert_empty_history_and_subscriptions(self):
""" Test that we handle no history and no subscriptions in a sane way."""
assert alert.check_for_alert(
[], None) is None, "Alerts should have been an empty list."
