def test_one_unprocessed_item(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"S": "a"}}}}, {"PutRequest": {"Item": {"h": {"S": "b"}}}}]}})
).andReturn(
_lv.BatchWriteItemResponse(UnprocessedItems={"Aaa": [{"PutRequest": {"Item": {"h": {"S": "c"}}}}]})
)
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"S": "c"}}}}]}})
).andReturn(
_lv.BatchWriteItemResponse()
)
batch_put_item(self.connection.object, "Aaa", {"h": "a"}, {"h": "b"})
def test_one_page(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"DeleteRequest": {"Key": {"h": {"S": "a"}}}}, {"DeleteRequest": {"Key": {"h": {"S": "b"}}}}]}})
).andReturn(
_lv.BatchWriteItemResponse()
)
batch_delete_item(self.connection.object, "Aaa", {"h": "a"}, {"h": "b"})
def test_several_pages(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(0, 25)]}})
).andReturn(
_lv.BatchWriteItemResponse()
)
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(25, 50)]}})
).andReturn(
_lv.BatchWriteItemResponse()
)
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(50, 60)]}})
).andReturn(
_lv.BatchWriteItemResponse()
)
batch_put_item(self.connection.object, "Aaa", ({"h": i} for i in range(60)))
def test_several_pages_of_unprocessed_item(self):
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(0, 25)]}})
).andReturn(
_lv.BatchWriteItemResponse(UnprocessedItems={"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(100, 110)]})
)
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(25, 35)]}})
).andReturn(
_lv.BatchWriteItemResponse(UnprocessedItems={"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(110, 120)]})
)
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(100, 120)]}})
).andReturn(
_lv.BatchWriteItemResponse(UnprocessedItems={"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(120, 130)]})
)
self.connection.expect._call_.withArguments(
self.ActionChecker("BatchWriteItem", {"RequestItems": {"Aaa": [{"PutRequest": {"Item": {"h": {"N": str(i)}}}} for i in range(120, 130)]}})
).andReturn(
_lv.BatchWriteItemResponse()
)
batch_put_item(self.connection.object, "Aaa", [{"h": i} for i in range(35)])