def testFlushLogLargestItems(self): self.pool = context._MutationPool(max_entity_count=3) self.pool.put(TestEntity(tag='a'*1024*1024)) self.assertRaises(apiproxy_errors.RequestTooLargeError, self.pool.flush) self.assertTrue(self.pool.puts._largest) self.pool = context._MutationPool(max_entity_count=3) self.pool.ndb_put(NdbTestEntity(tag='a'*1024*1024)) self.assertRaises(apiproxy_errors.RequestTooLargeError, self.pool.flush) self.assertTrue(self.pool.ndb_puts._largest)
def testFlushLogLargestItems(self): self.pool = context._MutationPool(max_entity_count=3) self.pool.put(TestEntity(tag='a' * 1024 * 1024)) self.assertRaises(apiproxy_errors.RequestTooLargeError, self.pool.flush) self.assertTrue(self.pool.puts._largest) self.pool = context._MutationPool(max_entity_count=3) self.pool.ndb_put(NdbTestEntity(tag='a' * 1024 * 1024)) self.assertRaises(apiproxy_errors.RequestTooLargeError, self.pool.flush) self.assertTrue(self.pool.ndb_puts._largest)
def testPoolWithForceWrites(self): class MapreduceSpec: def __init__(self): self.params = {'force_ops_writes': True} pool = context._MutationPool(mapreduce_spec=MapreduceSpec()) self.assertTrue(pool.force_writes)
def testNdbFlush(self): """Combined test for all NDB implicit and explicit flushing.""" self.pool = context._MutationPool(max_entity_count=3) for i in range(8): self.pool.put(NdbTestEntity()) self.assertEqual(len(self.pool.ndb_puts.items), (i % 3) + 1) for i in range(5): self.pool.delete(ndb.Key(NdbTestEntity, 'x%d' % i)) self.assertEquals(len(self.pool.ndb_deletes.items), (i % 3) + 1) self.pool.flush() self.assertEquals(len(self.pool.ndb_puts.items), 0) self.assertEquals(len(self.pool.ndb_deletes.items), 0) self.assertEquals(len(self.pool.puts.items), 0) self.assertEquals(len(self.pool.deletes.items), 0)
def testNdbFlush(self): """Combined test for all NDB implicit and explicit flushing.""" self.pool = context._MutationPool(max_entity_count=3) for i in range(8): self.pool.put(NdbTestEntity()) self.assertEqual(len(self.pool.ndb_puts.items), (i%3) + 1) for i in range(5): self.pool.delete(ndb.Key(NdbTestEntity, 'x%d' % i)) self.assertEquals(len(self.pool.ndb_deletes.items), (i%3) + 1) self.pool.flush() self.assertEquals(len(self.pool.ndb_puts.items), 0) self.assertEquals(len(self.pool.ndb_deletes.items), 0) self.assertEquals(len(self.pool.puts.items), 0) self.assertEquals(len(self.pool.deletes.items), 0)
def testFlush(self): """Combined test for all db implicit and explicit flushing.""" self.pool = context._MutationPool(max_entity_count=3) for i in range(8): self.pool.put(TestEntity()) self.assertEqual(len(self.pool.puts.items), (i % 3) + 1) for i in range(5): e = TestEntity() e.put() self.pool.delete(e) self.assertEqual(len(self.pool.deletes.items), (i % 3) + 1) self.pool.flush() self.assertEquals(len(self.pool.puts.items), 0) self.assertEquals(len(self.pool.deletes.items), 0) self.assertEquals(len(self.pool.ndb_puts.items), 0) self.assertEquals(len(self.pool.ndb_deletes.items), 0)
def testFlush(self): """Combined test for all db implicit and explicit flushing.""" self.pool = context._MutationPool(max_entity_count=3) for i in range(8): self.pool.put(TestEntity()) self.assertEqual(len(self.pool.puts.items), (i%3) + 1) for i in range(5): e = TestEntity() e.put() self.pool.delete(e) self.assertEqual(len(self.pool.deletes.items), (i%3) + 1) self.pool.flush() self.assertEquals(len(self.pool.puts.items), 0) self.assertEquals(len(self.pool.deletes.items), 0) self.assertEquals(len(self.pool.ndb_puts.items), 0) self.assertEquals(len(self.pool.ndb_deletes.items), 0)
def setUp(self): super(MutationPoolTest, self).setUp() self.pool = context._MutationPool()
def testPoolWithForceWrites(self): class MapreduceSpec: def __init__(self): self.params = {'force_ops_writes':True} pool = context._MutationPool(mapreduce_spec=MapreduceSpec()) self.assertTrue(pool.force_writes)