def test_query_fetch_data_filter(self): entity1 = fake_cloud_client.make_entity(("abc", "1")) entity1["k1"] = "v1" entity2 = fake_cloud_client.make_entity(("abc", "2")) entity2["k1"] = "v2" entity2["k2"] = "v2" entity3 = fake_cloud_client.make_entity(("abc", "3")) entity3["k2"] = "v3" client = fake_cloud_client.FakeDatastoreClient( [entity1, entity2, entity3]) assertCountEqual(self, [entity1], client.query_fetch(filters=[("k1", "=", "v1")])) assertCountEqual(self, [entity2], client.query_fetch(filters=[("k1", ">", "v1")])) assertCountEqual(self, [entity1, entity2], client.query_fetch(filters=[("k1", ">=", "v1")])) assertCountEqual(self, [entity2], client.query_fetch(filters=[("k2", "<", "v3")])) assertCountEqual(self, [entity2, entity3], client.query_fetch(filters=[("k2", "<=", "v3")])) assertCountEqual( self, [entity2], client.query_fetch(filters=[("k1", ">=", "v1"), ("k2", "<=", "v3")]), )
def test_write_to_datastore(self): self.reset_work_pieces() self.work_pieces.work["w1"] = self.work1 self.work_pieces.work["w2"] = self.work2 self.work_pieces.write_all_to_datastore() # verify content of the datastore parent_key = fake_cloud_client.FakeDatastoreKey( work_data.KIND_WORK_TYPE, TEST_WORK_TYPE_ENTITY_ID) assertCountEqual( self, [fake_cloud_client.make_entity(parent_key)], self.datastore_client.query_fetch(kind=work_data.KIND_WORK_TYPE), ) entity1 = fake_cloud_client.make_entity( fake_cloud_client.FakeDatastoreKey(work_data.KIND_WORK, "w1", parent=parent_key)) entity1.update(self.work1) entity2 = fake_cloud_client.make_entity( fake_cloud_client.FakeDatastoreKey(work_data.KIND_WORK, "w2", parent=parent_key)) entity2.update(self.work2) assertCountEqual( self, [entity1, entity2], self.datastore_client.query_fetch(kind=work_data.KIND_WORK), )
def test_query_fetch_all(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('abc', '1', 'def', '2')) entity2['k2'] = 'v2' client = fake_cloud_client.FakeDatastoreClient([entity1, entity2]) assertCountEqual(self, [entity1, entity2], client.query_fetch())
def test_query_fetch_all(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('abc', '1', 'def', '2')) entity2['k2'] = 'v2' client = fake_cloud_client.FakeDatastoreClient([entity1, entity2]) assertCountEqual(self, [entity1, entity2], client.query_fetch())
def test_query_fetch_all(self): entity1 = fake_cloud_client.make_entity(("abc", "1")) entity1["k1"] = "v1" entity2 = fake_cloud_client.make_entity(("abc", "1", "def", "2")) entity2["k2"] = "v2" client = fake_cloud_client.FakeDatastoreClient([entity1, entity2]) assertCountEqual(self, [entity1, entity2], client.query_fetch())
def test_equality_keys(self): entity1a = fake_cloud_client.make_entity(('abc', '1')) entity1b = fake_cloud_client.make_entity(('abc', '1')) entity2 = fake_cloud_client.make_entity(('abc', '2')) self.assertFalse(entity1a == entity2) self.assertTrue(entity1a != entity2) self.assertTrue(entity1a == entity1b) self.assertFalse(entity1b != entity1b)
def test_equality_keys(self): entity1a = fake_cloud_client.make_entity(("abc", "1")) entity1b = fake_cloud_client.make_entity(("abc", "1")) entity2 = fake_cloud_client.make_entity(("abc", "2")) self.assertFalse(entity1a == entity2) self.assertTrue(entity1a != entity2) self.assertTrue(entity1a == entity1b) self.assertFalse(entity1b != entity1b)
def test_equality_keys(self): entity1a = fake_cloud_client.make_entity(('abc', '1')) entity1b = fake_cloud_client.make_entity(('abc', '1')) entity2 = fake_cloud_client.make_entity(('abc', '2')) self.assertFalse(entity1a == entity2) self.assertTrue(entity1a != entity2) self.assertTrue(entity1a == entity1b) self.assertFalse(entity1b != entity1b)
def test_query_fetch_ancestor_filter(self): entity1 = fake_cloud_client.make_entity(("abc", "1", "def", "2")) entity1["k1"] = "v1" entity2 = fake_cloud_client.make_entity(("xyz", "3", "qwe", "4")) entity2["k2"] = "v2" client = fake_cloud_client.FakeDatastoreClient([entity1, entity2]) assertCountEqual(self, [entity1], client.query_fetch(ancestor=client.key("abc", "1"))) assertCountEqual(self, [entity2], client.query_fetch(ancestor=client.key("xyz", "3")))
def test_query_fetch_ancestor_filter(self): entity1 = fake_cloud_client.make_entity(('abc', '1', 'def', '2')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('xyz', '3', 'qwe', '4')) entity2['k2'] = 'v2' client = fake_cloud_client.FakeDatastoreClient([entity1, entity2]) assertCountEqual(self, [entity1], client.query_fetch(ancestor=client.key('abc', '1'))) assertCountEqual(self, [entity2], client.query_fetch(ancestor=client.key('xyz', '3')))
def test_query_fetch_ancestor_filter(self): entity1 = fake_cloud_client.make_entity(('abc', '1', 'def', '2')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('xyz', '3', 'qwe', '4')) entity2['k2'] = 'v2' client = fake_cloud_client.FakeDatastoreClient([entity1, entity2]) assertCountEqual(self, [entity1], client.query_fetch(ancestor=client.key('abc', '1'))) assertCountEqual(self, [entity2], client.query_fetch(ancestor=client.key('xyz', '3')))
def test_equality_dict(self): entity1 = fake_cloud_client.make_entity(("abc", "1")) entity1["k1"] = "v1" entity2 = fake_cloud_client.make_entity(("abc", "1")) entity2["k1"] = "v2" entity3 = fake_cloud_client.make_entity(("abc", "1")) entity1["k1"] = "v1" entity1["k2"] = "v2" # compare to self self.assertTrue(entity1 == entity1) self.assertFalse(entity1 != entity1) self.assertTrue(entity2 == entity2) self.assertFalse(entity2 != entity2) self.assertTrue(entity3 == entity3) self.assertFalse(entity3 != entity3) # compare to others self.assertFalse(entity1 == entity2) self.assertTrue(entity1 != entity2) self.assertFalse(entity1 == entity3) self.assertTrue(entity1 != entity3) self.assertFalse(entity2 == entity3) self.assertTrue(entity2 != entity3)
def test_equality_dict(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('abc', '1')) entity2['k1'] = 'v2' entity3 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity1['k2'] = 'v2' # compare to self self.assertTrue(entity1 == entity1) self.assertFalse(entity1 != entity1) self.assertTrue(entity2 == entity2) self.assertFalse(entity2 != entity2) self.assertTrue(entity3 == entity3) self.assertFalse(entity3 != entity3) # compare to others self.assertFalse(entity1 == entity2) self.assertTrue(entity1 != entity2) self.assertFalse(entity1 == entity3) self.assertTrue(entity1 != entity3) self.assertFalse(entity2 == entity3) self.assertTrue(entity2 != entity3)
def test_equality_dict(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('abc', '1')) entity2['k1'] = 'v2' entity3 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity1['k2'] = 'v2' # compare to self self.assertTrue(entity1 == entity1) self.assertFalse(entity1 != entity1) self.assertTrue(entity2 == entity2) self.assertFalse(entity2 != entity2) self.assertTrue(entity3 == entity3) self.assertFalse(entity3 != entity3) # compare to others self.assertFalse(entity1 == entity2) self.assertTrue(entity1 != entity2) self.assertFalse(entity1 == entity3) self.assertTrue(entity1 != entity3) self.assertFalse(entity2 == entity3) self.assertTrue(entity2 != entity3)
def test_query_fetch_data_filter(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('abc', '2')) entity2['k1'] = 'v2' entity2['k2'] = 'v2' entity3 = fake_cloud_client.make_entity(('abc', '3')) entity3['k2'] = 'v3' client = fake_cloud_client.FakeDatastoreClient([entity1, entity2, entity3]) assertCountEqual(self, [entity1], client.query_fetch(filters=[('k1', '=', 'v1')])) assertCountEqual(self, [entity2], client.query_fetch(filters=[('k1', '>', 'v1')])) assertCountEqual(self, [entity1, entity2], client.query_fetch(filters=[('k1', '>=', 'v1')])) assertCountEqual(self, [entity2], client.query_fetch(filters=[('k2', '<', 'v3')])) assertCountEqual(self, [entity2, entity3], client.query_fetch(filters=[('k2', '<=', 'v3')])) assertCountEqual(self, [entity2], client.query_fetch(filters=[('k1', '>=', 'v1'), ('k2', '<=', 'v3')]))
def test_write_to_datastore(self): self.reset_work_pieces() self.work_pieces.work['w1'] = self.work1 self.work_pieces.work['w2'] = self.work2 self.work_pieces.write_all_to_datastore() # verify content of the datastore parent_key = fake_cloud_client.FakeDatastoreKey(work_data.KIND_WORK_TYPE, TEST_WORK_TYPE_ENTITY_ID) assertCountEqual( self, [fake_cloud_client.make_entity(parent_key)], self.datastore_client.query_fetch(kind=work_data.KIND_WORK_TYPE)) entity1 = fake_cloud_client.make_entity( fake_cloud_client.FakeDatastoreKey( work_data.KIND_WORK, 'w1', parent=parent_key)) entity1.update(self.work1) entity2 = fake_cloud_client.make_entity( fake_cloud_client.FakeDatastoreKey( work_data.KIND_WORK, 'w2', parent=parent_key)) entity2.update(self.work2) assertCountEqual( self, [entity1, entity2], self.datastore_client.query_fetch(kind=work_data.KIND_WORK))
def test_query_fetch_data_filter(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = 'v1' entity2 = fake_cloud_client.make_entity(('abc', '2')) entity2['k1'] = 'v2' entity2['k2'] = 'v2' entity3 = fake_cloud_client.make_entity(('abc', '3')) entity3['k2'] = 'v3' client = fake_cloud_client.FakeDatastoreClient( [entity1, entity2, entity3]) assertCountEqual(self, [entity1], client.query_fetch(filters=[('k1', '=', 'v1')])) assertCountEqual(self, [entity2], client.query_fetch(filters=[('k1', '>', 'v1')])) assertCountEqual(self, [entity1, entity2], client.query_fetch(filters=[('k1', '>=', 'v1')])) assertCountEqual(self, [entity2], client.query_fetch(filters=[('k2', '<', 'v3')])) assertCountEqual(self, [entity2, entity3], client.query_fetch(filters=[('k2', '<=', 'v3')])) assertCountEqual( self, [entity2], client.query_fetch(filters=[('k1', '>=', 'v1'), ('k2', '<=', 'v3')]))
def test_deep_copy(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = ['v1'] self.assertEqual(entity1.key, fake_cloud_client.FakeDatastoreKey('abc', '1')) self.assertEqual(dict(entity1), {'k1': ['v1']}) entity2 = copy.deepcopy(entity1) entity2['k1'].append('v2') entity2['k3'] = 'v3' self.assertIsInstance(entity2, fake_cloud_client.FakeDatastoreEntity) self.assertEqual(entity1.key, fake_cloud_client.FakeDatastoreKey('abc', '1')) self.assertEqual(dict(entity1), {'k1': ['v1']}) self.assertEqual(entity2.key, fake_cloud_client.FakeDatastoreKey('abc', '1')) self.assertEqual(dict(entity2), {'k1': ['v1', 'v2'], 'k3': 'v3'})
def test_deep_copy(self): entity1 = fake_cloud_client.make_entity(("abc", "1")) entity1["k1"] = ["v1"] self.assertEqual(entity1.key, fake_cloud_client.FakeDatastoreKey("abc", "1")) self.assertEqual(dict(entity1), {"k1": ["v1"]}) entity2 = copy.deepcopy(entity1) entity2["k1"].append("v2") entity2["k3"] = "v3" self.assertIsInstance(entity2, fake_cloud_client.FakeDatastoreEntity) self.assertEqual(entity1.key, fake_cloud_client.FakeDatastoreKey("abc", "1")) self.assertEqual(dict(entity1), {"k1": ["v1"]}) self.assertEqual(entity2.key, fake_cloud_client.FakeDatastoreKey("abc", "1")) self.assertEqual(dict(entity2), {"k1": ["v1", "v2"], "k3": "v3"})
def test_deep_copy(self): entity1 = fake_cloud_client.make_entity(('abc', '1')) entity1['k1'] = ['v1'] self.assertEqual(entity1.key, fake_cloud_client.FakeDatastoreKey('abc', '1')) self.assertEqual(dict(entity1), {'k1': ['v1']}) entity2 = copy.deepcopy(entity1) entity2['k1'].append('v2') entity2['k3'] = 'v3' self.assertIsInstance(entity2, fake_cloud_client.FakeDatastoreEntity) self.assertEqual(entity1.key, fake_cloud_client.FakeDatastoreKey('abc', '1')) self.assertEqual(dict(entity1), {'k1': ['v1']}) self.assertEqual(entity2.key, fake_cloud_client.FakeDatastoreKey('abc', '1')) self.assertEqual(dict(entity2), {'k1': ['v1', 'v2'], 'k3': 'v3'})
def test_key(self): entity = fake_cloud_client.make_entity(('abc', '1')) self.assertEqual(entity.key, fake_cloud_client.FakeDatastoreKey('abc', '1'))
def test_key(self): entity = fake_cloud_client.make_entity(("abc", "1")) self.assertEqual(entity.key, fake_cloud_client.FakeDatastoreKey("abc", "1"))
def test_key(self): entity = fake_cloud_client.make_entity(('abc', '1')) self.assertEqual(entity.key, fake_cloud_client.FakeDatastoreKey('abc', '1'))