def test_it_handles_successful_queries(table):
resp = update_stats(
"job123",
[{
"Id": "job123",
"Sk": "123456",
"Type": "JobEvent",
"CreatedAt": 123.0,
"EventName": "QuerySucceeded",
"EventData": {
"Statistics": {
"DataScannedInBytes": 10,
"EngineExecutionTimeInMillis": 100,
}
},
}],
)
table.update_item.assert_called_with(
Key={
"Id": "job123",
"Sk": "job123",
},
ConditionExpression="#Id = :Id AND #Sk = :Sk",
UpdateExpression="set #qt = if_not_exists(#qt, :z) + :qt, "
"#qs = if_not_exists(#qs, :z) + :qs, "
"#qf = if_not_exists(#qf, :z) + :qf, "
"#qb = if_not_exists(#qb, :z) + :qb, "
"#qm = if_not_exists(#qm, :z) + :qm, "
"#ou = if_not_exists(#ou, :z) + :ou, "
"#of = if_not_exists(#of, :z) + :of, "
"#or = if_not_exists(#or, :z) + :or",
ExpressionAttributeNames={
"#Id": "Id",
"#Sk": "Sk",
"#qt": "TotalQueryCount",
"#qs": "TotalQuerySucceededCount",
"#qf": "TotalQueryFailedCount",
"#qb": "TotalQueryScannedInBytes",
"#qm": "TotalQueryTimeInMillis",
"#ou": "TotalObjectUpdatedCount",
"#of": "TotalObjectUpdateFailedCount",
"#or": "TotalObjectRollbackFailedCount",
},
ExpressionAttributeValues={
":Id": "job123",
":Sk": "job123",
":qt": 1,
":qs": 1,
":qf": 0,
":qb": 10,
":qm": 100,
":ou": 0,
":of": 0,
":or": 0,
":z": 0,
},
ReturnValues="ALL_NEW",
)
def test_it_handles_already_failed_jobs(table, ddb):
e = boto3.client("dynamodb").exceptions.ConditionalCheckFailedException
ddb.meta.client.exceptions.ConditionalCheckFailedException = e
table.update_item.side_effect = e({}, "ConditionalCheckFailedException")
update_stats("job123", [{
"Id": "job123",
"Sk": "123456",
"Type": "JobEvent",
"CreatedAt": 123.0,
"EventName": "QuerySucceeded",
"EventData": {
"Statistics": {
"DataScannedInBytes": 10,
"EngineExecutionTimeInMillis": 100
}
}
}])
table.update_item.assert_called()
def test_it_handles_failed_queries(table):
resp = update_stats("job123", [{
"Id": "job123",
"Sk": "123456",
"Type": "JobEvent",
"CreatedAt": 123.0,
"EventName": "QueryFailed",
"EventData": {}
}])
table.update_item.assert_called_with(
Key={
'Id': "job123",
'Sk': "job123",
},
ConditionExpression="#Id = :Id AND #Sk = :Sk",
UpdateExpression="set #qt = if_not_exists(#qt, :z) + :qt, "
"#qs = if_not_exists(#qs, :z) + :qs, "
"#qf = if_not_exists(#qf, :z) + :qf, "
"#qb = if_not_exists(#qb, :z) + :qb, "
"#qm = if_not_exists(#qm, :z) + :qm, "
"#ou = if_not_exists(#ou, :z) + :ou, "
"#of = if_not_exists(#of, :z) + :of, "
"#or = if_not_exists(#or, :z) + :or",
ExpressionAttributeNames={
"#Id": "Id",
"#Sk": "Sk",
'#qt': 'TotalQueryCount',
'#qs': 'TotalQuerySucceededCount',
'#qf': 'TotalQueryFailedCount',
'#qb': 'TotalQueryScannedInBytes',
'#qm': 'TotalQueryTimeInMillis',
'#ou': 'TotalObjectUpdatedCount',
'#of': 'TotalObjectUpdateFailedCount',
'#or': 'TotalObjectRollbackFailedCount',
},
ExpressionAttributeValues={
":Id": "job123",
":Sk": "job123",
':qt': 1,
':qs': 0,
':qf': 1,
':qb': 0,
':qm': 0,
':ou': 0,
':of': 0,
':or': 0,
':z': 0,
},
ReturnValues="ALL_NEW")