def test_get_with_unprocessed_keys(self): _lv.batch_put_item(self.connection, "Aaa", [{"h": unicode(i), "xs": "x" * 300000} for i in range(100)]) # 300kB items ensure a single BatchGetItem will return at most 55 items r1 = self.connection(_lv.BatchGetItem().table("Aaa").keys({"h": unicode(i)} for i in range(100))) self.assertEqual(len(r1.unprocessed_keys["Aaa"]["Keys"]), 45) self.assertEqual(len(r1.responses["Aaa"]), 55)
def test_get_without_unprocessed_keys(self): _lv.batch_put_item(self.connection, "Aaa", [{"h": unicode(i)} for i in range(100)]) r = self.connection(_lv.BatchGetItem().table("Aaa").keys({"h": unicode(i)} for i in range(100))) self.assertEqual(r.unprocessed_keys, {}) self.assertEqual(len(r.responses["Aaa"]), 100)
def global_setup(): connection = _lv.Connection("us-west-2", _lv.EnvironmentCredentials()) table1 = "LowVoltage.Tests.Doc.1" table2 = "LowVoltage.Tests.Doc.2" try: connection(_lv.DescribeTable(table1)) except _lv.ResourceNotFoundException: connection( _lv.CreateTable(table) .hash_key("h", _lv.NUMBER).provisioned_throughput(1, 1) .global_secondary_index("gsi").hash_key("gh", _lv.NUMBER).range_key("gr", _lv.NUMBER).provisioned_throughput(1, 1).project_all() ) try: connection(_lv.DescribeTable(table2)) except _lv.ResourceNotFoundException: connection( _lv.CreateTable(table2) .hash_key("h", _lv.NUMBER).range_key("r1", _lv.NUMBER).provisioned_throughput(1, 1) .local_secondary_index("lsi").hash_key("h", _lv.NUMBER).range_key("r2", _lv.NUMBER).project_all() ) _lv.wait_for_table_activation(connection, table1) _lv.batch_put_item( connection, table1, [{"h": h, "gh": h * h, "gr": 10 - 2 * h} for h in range(7)], ) _lv.wait_for_table_activation(connection, table1) _lv.batch_put_item( connection, table1, [{"h": h, "a": 0} for h in range(7, 10)], ) _lv.wait_for_table_activation(connection, table2) _lv.batch_put_item( connection, table2, [{"h": h, "r1": 0, "r2": 0} for h in range(10)], ) _lv.batch_put_item( connection, table2, [{"h": 42, "r1": r1, "r2": 10 - r1} for r1 in range(6)], ) _lv.batch_put_item( connection, table2, [{"h": 42, "r1": r1} for r1 in range(6, 10)], ) return connection, table1, table2
def test_get_without_unprocessed_keys(self): _lv.batch_put_item(self.connection, "Aaa", [{ "h": str(i) } for i in range(100)]) r = self.connection(_lv.BatchGetItem().table("Aaa").keys( {"h": str(i)} for i in range(100))) self.assertEqual(r.unprocessed_keys, {}) self.assertEqual(len(r.responses["Aaa"]), 100)
def test_get_with_unprocessed_keys(self): _lv.batch_put_item( self.connection, "Aaa", [{ "h": str(i), "xs": "x" * 300000 } for i in range(100)] ) # 300kB items ensure a single BatchGetItem will return at most 55 items r1 = self.connection(_lv.BatchGetItem().table("Aaa").keys( {"h": str(i)} for i in range(100))) self.assertEqual(len(r1.unprocessed_keys["Aaa"]["Keys"]), 45) self.assertEqual(len(r1.responses["Aaa"]), 55)
def setUp(self): super(IterateBatchGetItemLocalIntegTests, self).setUp() _lv.batch_put_item(self.connection, "Aaa", [{"h": self.key(i), "xs": "x" * 300000} for i in range(250)]) # 300kB items ensure a single BatchGetItem will return at most 55 items
def setUp(self): super(BatchDeleteItemLocalIntegTests, self).setUp() _lv.batch_put_item(self.connection, "Aaa", [{ "h": self.key(i) } for i in range(100)])
def test(self): _lv.batch_put_item(self.connection, "Aaa", [{"h": self.key(i)} for i in range(100)]) self.assertEqual(len(list(_lv.iterate_scan(self.connection, _lv.Scan("Aaa")))), 100)
def setUp(self): super(IterateBatchGetItemLocalIntegTests, self).setUp() _lv.batch_put_item(self.connection, "Aaa", [{"h": self.key(i), "xs": "x" * 300000} for i in range(250)]) # 300kB items ensure a single BatchGetItem will return at most 55 items
def test(self): _lv.batch_put_item(self.connection, "Aaa", [{"h": self.key(i)} for i in range(100)]) self.assertEqual(len(list(_lv.iterate_scan(self.connection, _lv.Scan("Aaa")))), 100)
def global_setup(): connection = _lv.Connection("us-west-2", _lv.EnvironmentCredentials()) table1 = "LowVoltage.Tests.Doc.1" table2 = "LowVoltage.Tests.Doc.2" try: connection(_lv.DescribeTable(table1)) except _lv.ResourceNotFoundException: connection( _lv.CreateTable(table).hash_key( "h", _lv.NUMBER).provisioned_throughput( 1, 1).global_secondary_index("gsi").hash_key( "gh", _lv.NUMBER).range_key( "gr", _lv.NUMBER).provisioned_throughput( 1, 1).project_all()) try: connection(_lv.DescribeTable(table2)) except _lv.ResourceNotFoundException: connection( _lv.CreateTable(table2).hash_key("h", _lv.NUMBER).range_key( "r1", _lv.NUMBER).provisioned_throughput( 1, 1).local_secondary_index("lsi").hash_key( "h", _lv.NUMBER).range_key("r2", _lv.NUMBER).project_all()) _lv.wait_for_table_activation(connection, table1) _lv.batch_put_item( connection, table1, [{ "h": h, "gh": h * h, "gr": 10 - 2 * h } for h in range(7)], ) _lv.wait_for_table_activation(connection, table1) _lv.batch_put_item( connection, table1, [{ "h": h, "a": 0 } for h in range(7, 10)], ) _lv.wait_for_table_activation(connection, table2) _lv.batch_put_item( connection, table2, [{ "h": h, "r1": 0, "r2": 0 } for h in range(10)], ) _lv.batch_put_item( connection, table2, [{ "h": 42, "r1": r1, "r2": 10 - r1 } for r1 in range(6)], ) _lv.batch_put_item( connection, table2, [{ "h": 42, "r1": r1 } for r1 in range(6, 10)], ) return connection, table1, table2
def setUp(self): super(BatchDeleteItemLocalIntegTests, self).setUp() _lv.batch_put_item(self.connection, "Aaa", [{"h": self.key(i)} for i in range(100)])