Ordering is not guaranted, make test more robust so it doesn't depend on

Kami · Kami · commit a3d21b6edcdf · 2019-08-19T15:17:29.000+02:00
the order.
diff --git a/datastore/tests/system/test_system.py b/datastore/tests/system/test_system.py
@@ -612,10 +612,15 @@ def test_put_multi_get_multi_pb(self):
         keys = [key1, key2]
         entity_retrieved_pbs = Config.CLIENT.get_multi_entity_pb(keys=keys)
         self.assertEqual(len(entity_retrieved_pbs), 2)
-        self.assertEqual(entity_retrieved_pbs[0].key.path[0].id, 1111)
-        self.assertEqual(entity_retrieved_pbs[1].key.path[0].id, 2222)
 
-    def test_fetch_entity_pbs(self):
+        retrieved_keys = []
+        for entity_retrieved_pb in entity_retrieved_pbs:
+            retrieved_keys.append(entity_retrieved_pb.key.path[0].id)
+
+        self.assertTrue(1111 in retrieved_keys)
+        self.assertTrue(2222 in retrieved_keys)
+
+    def test_fetch_entity_pb(self):
         # Store multiple entities
         key1 = Config.CLIENT.key("TestEntityPB", 1111)
         entity1_pb = entity_pb2.Entity()
