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'))
