Add appengine/gcloud/mustache dependencies.

pkg/appengine/test/integration/raw_datastore_test_impl.dart

Index: pkg/appengine/test/integration/raw_datastore_test_impl.dart
diff --git a/pkg/appengine/test/integration/raw_datastore_test_impl.dart b/pkg/appengine/test/integration/raw_datastore_test_impl.dart
new file mode 100644
index 0000000000000000000000000000000000000000..bb3a21c9bcbf263c9321495ab259c72124a9c3f0
--- /dev/null
+++ b/pkg/appengine/test/integration/raw_datastore_test_impl.dart
@@ -0,0 +1,1072 @@
+// Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
+// for details. All rights reserved. Use of this source code is governed by a
+// BSD-style license that can be found in the LICENSE file.
+
+library raw_datastore_test_impl;
+
+import 'dart:async';
+
+import 'package:gcloud/datastore.dart';
+import 'package:gcloud/src/datastore_impl.dart' as datastore_impl;
+import 'package:gcloud/common.dart';
+import 'package:unittest/unittest.dart';
+
+import '../utils/error_matchers.dart';
+import '../utils/raw_datastore_test_utils.dart';
+
+
+Future sleep(Duration duration) {
+  var completer = new Completer();
+  new Timer(duration, completer.complete);
+  return completer.future;
+}
+
+Future<List<Entity>> consumePages(FirstPageProvider provider) {
+  return new StreamFromPages(provider).stream.toList();
+}
+
+runTests(Datastore datastore) {
+  Future withTransaction(Function f, {bool xg: false}) {
+    return datastore.beginTransaction(crossEntityGroup: xg).then(f);
+  }
+
+  Future<List<Key>> insert(List<Entity> entities,
+                           List<Entity> autoIdEntities,
+                           {bool transactional: true}) {
+    if (transactional) {
+      return withTransaction((Transaction transaction) {
+        return datastore.commit(inserts: entities,
+                                autoIdInserts: autoIdEntities,
+                                transaction: transaction).then((result) {
+          if (autoIdEntities != null && autoIdEntities.length > 0) {
+            expect(result.autoIdInsertKeys.length,
+                   equals(autoIdEntities.length));
+          }
+          return result.autoIdInsertKeys;
+        });
+      }, xg: true);
+    } else {
+      return datastore.commit(inserts: entities, autoIdInserts: autoIdEntities)
+          .then((result) {
+            if (autoIdEntities != null && autoIdEntities.length > 0) {
+              expect(result.autoIdInsertKeys.length,
+                     equals(autoIdEntities.length));
+            }
+            return result.autoIdInsertKeys;
+          });
+    }
+  }
+
+  Future delete(List<Key> keys, {bool transactional: true}) {
+    if (transactional) {
+      return withTransaction((Transaction t) {
+        return datastore.commit(deletes: keys, transaction: t)
+            .then((result) => null);
+      }, xg: true);
+    } else {
+      return datastore.commit(deletes: keys).then((_) => _);
+    }
+  }
+
+  Future<List<Entity>> lookup(List<Key> keys, {bool transactional: true}) {
+    if (transactional) {
+      return withTransaction((Transaction transaction) {
+        return datastore.lookup(keys, transaction: transaction);
+      }, xg: true);
+    } else {
+      return datastore.lookup(keys);
+    }
+  }
+
+  bool isValidKey(Key key, {bool ignoreIds: false}) {
+    if (key.elements.length == 0) return false;
+
+    for (var element in key.elements) {
+      if (element.kind == null || element.kind is! String) return false;
+      if (!ignoreIds) {
+        if (element.id == null ||
+            (element.id is! String && element.id is! int)) {
+          return false;
+        }
+      }
+    }
+    return true;
+  }
+
+  bool compareKey(Key a, Key b, {bool ignoreIds: false}) {
+    if (a.partition != b.partition) return false;
+    if (a.elements.length != b.elements.length) return false;
+    for (int i = 0; i < a.elements.length; i++) {
+      if (a.elements[i].kind != b.elements[i].kind) return false;
+      if (!ignoreIds && a.elements[i].id != b.elements[i].id) return false;
+    }
+    return true;
+  }
+
+  bool compareEntity(Entity a, Entity b, {bool ignoreIds: false}) {
+    if (!compareKey(a.key, b.key, ignoreIds: ignoreIds)) return false;
+    if (a.properties.length != b.properties.length) return false;
+    for (var key in a.properties.keys) {
+      if (!b.properties.containsKey(key)) return false;
+      if (a.properties[key] != null && a.properties[key] is List) {
+        var aList = a.properties[key];
+        var bList = b.properties[key];
+        if (aList.length != bList.length) return false;
+        for (var i = 0; i < aList.length; i++) {
+          if (aList[i] != bList[i]) return false;
+        }
+      } else if (a.properties[key] is BlobValue) {
+        if (b.properties[key] is BlobValue) {
+          var b1 = (a.properties[key] as BlobValue).bytes;
+          var b2 = (b.properties[key] as BlobValue).bytes;
+          if (b1.length != b2.length) return false;
+          for (var i = 0; i < b1.length; i++) {
+            if (b1[i] != b2[i]) return false;
+          }
+          return true;
+        }
+        return false;
+      } else {
+        if (a.properties[key] != b.properties[key]) {
+          return false;
+        }
+      }
+    }
+    return true;
+  }
+
+  group('e2e_datastore', () {
+    group('insert', () {
+      Future<List<Key>> testInsert(List<Entity> entities,
+          {bool transactional: false, bool xg: false, bool unnamed: true}) {
+        Future<List<Key>> test(Transaction transaction) {
+          return datastore.commit(autoIdInserts: entities,
+                                  transaction: transaction)
+              .then((CommitResult result) {
+            expect(result.autoIdInsertKeys.length, equals(entities.length));
+
+            for (var i = 0; i < result.autoIdInsertKeys.length; i++) {
+              var key = result.autoIdInsertKeys[i];
+              expect(isValidKey(key), isTrue);
+              if (unnamed) {
+                expect(compareKey(key, entities[i].key, ignoreIds: true),
+                       isTrue);
+              } else {
+                expect(compareKey(key, entities[i].key), isTrue);
+              }
+            }
+            return result.autoIdInsertKeys;
+          });
+        }
+
+        if (transactional) {
+          return withTransaction(test, xg: xg);
+        }
+        return test(null);
+      }
+
+      Future<List<Key>> testInsertNegative(List<Entity> entities,
+          {bool transactional: false, bool xg: false}) {
+        test(Transaction transaction) {
+          expect(datastore.commit(inserts: entities,
+                                  transaction: transaction),
+                                  throwsA(isApplicationError));
+        }
+
+        if (transactional) {
+          return withTransaction(test, xg: xg);
+        }
+        return test(null);
+      }
+
+      var unnamedEntities1 = buildEntities(42, 43);
+      var unnamedEntities5 = buildEntities(1, 6);
+      var unnamedEntities20 = buildEntities(6, 26);
+      var named20000 = buildEntities(
+          1000, 21001, idFunction: (i) => 'named_${i}_of_10000');
+
+      test('insert', () {
+        return testInsert(unnamedEntities5, transactional: false).then((keys) {
+          return delete(keys).then((_) {
+            return lookup(keys).then((List<Entity> entities) {
+              entities.forEach((Entity e) => expect(e, isNull));
+            });
+          });
+        });
+      });
+
+      test('insert_transactional', () {
+        return testInsert(unnamedEntities1, transactional: true).then((keys) {
+          return delete(keys).then((_) {
+            return lookup(keys).then((List<Entity> entities) {
+              entities.forEach((Entity e) => expect(e, isNull));
+            });
+          });
+        });
+      });
+
+      test('insert_transactional_xg', () {
+        return testInsert(
+            unnamedEntities5, transactional: true, xg: true).then((keys) {
+          return delete(keys).then((_) {
+            return lookup(keys).then((List<Entity> entities) {
+              entities.forEach((Entity e) => expect(e, isNull));
+            });
+          });
+        });
+      });
+
+      // Does not work with cloud datastore REST api, why?
+      test('negative_insert_transactional', () {
+        return testInsertNegative(unnamedEntities5, transactional: true);
+      });
+
+      // Does not work with cloud datastore REST api, why?
+      test('negative_insert_transactional_xg', () {
+        return testInsertNegative(
+            unnamedEntities20, transactional: true, xg: true);
+      });
+
+      test('negative_insert_20000_entities', () {
+        // Maybe it should not be a [DataStoreError] here?
+        // FIXME/TODO: This was adapted
+        expect(datastore.commit(inserts: named20000),
+               throws);
+      });
+
+      // TODO: test invalid inserts (like entities without key, ...)
+    });
+
+    group('allocate_ids', () {
+      test('allocate_ids_query', () {
+        compareResult(List<Key> keys, List<Key> completedKeys) {
+          expect(completedKeys.length, equals(keys.length));
+          for (int i = 0; i < keys.length; i++) {
+            var insertedKey = keys[i];
+            var completedKey = completedKeys[i];
+
+            expect(completedKey.elements.length,
+                   equals(insertedKey.elements.length));
+            for (int j = 0; j < insertedKey.elements.length - 1; j++) {
+              expect(completedKey.elements[j], equals(insertedKey.elements[j]));
+            }
+            for (int j = insertedKey.elements.length - 1;
+                 j < insertedKey.elements.length;
+                 j++) {
+              expect(completedKey.elements[j].kind,
+                     equals(insertedKey.elements[j].kind));
+              expect(completedKey.elements[j].id, isNotNull);
+              expect(completedKey.elements[j].id, isInt);
+            }
+          }
+        }
+
+        var keys = buildKeys(1, 4);
+        return datastore.allocateIds(keys).then((List<Key> completedKeys) {
+          compareResult(keys, completedKeys);
+          // TODO: Make sure we can insert these keys
+          // FIXME: Insert currently doesn't through if entities already exist!
+        });
+      });
+    });
+
+    group('lookup', () {
+      Future testLookup(List<Key> keysToLookup,
+                        List<Entity> entitiesToLookup,
+                        {bool transactional: false,
+                         bool xg: false,
+                         bool negative: false,
+                         bool named: false}) {
+        expect(keysToLookup.length, equals(entitiesToLookup.length));
+        for (var i = 0; i < keysToLookup.length; i++) {
+          expect(compareKey(keysToLookup[i],
+                            entitiesToLookup[i].key,
+                            ignoreIds: !named), isTrue);
+        }
+
+        Future test(Transaction transaction) {
+          return datastore.lookup(keysToLookup)
+              .then((List<Entity> entities) {
+            expect(entities.length, equals(keysToLookup.length));
+            if (negative) {
+              for (int i = 0; i < entities.length; i++) {
+                expect(entities[i], isNull);
+              }
+            } else {
+              for (var i = 0; i < entities.length; i++) {
+                expect(compareKey(entities[i].key, keysToLookup[i]), isTrue);
+                expect(compareEntity(entities[i],
+                                     entitiesToLookup[i],
+                                     ignoreIds: !named), isTrue);
+              }
+            }
+            if (transaction != null) {
+              return
+                  datastore.commit(transaction: transaction).then((_) => null);
+            }
+          });
+        }
+
+        if (transactional) {
+          return withTransaction(test, xg: xg);
+        }
+        return test(null);
+      }
+
+      var unnamedEntities1 = buildEntities(42, 43);
+      var unnamedEntities5 = buildEntities(1, 6);
+      var unnamedEntities20 = buildEntities(6, 26);
+      var entitiesWithAllPropertyTypes = buildEntityWithAllProperties(1, 6);
+
+      test('lookup', () {
+        return insert([], unnamedEntities20, transactional: false).then((keys) {
+          keys.forEach((key) => expect(isValidKey(key), isTrue));
+          return testLookup(keys, unnamedEntities20).then((_) {
+            return delete(keys, transactional: false);
+          });
+        });
+      });
+
+      test('lookup_with_all_properties', () {
+        return insert(entitiesWithAllPropertyTypes, [], transactional: false)
+            .then((_) {
+          var keys = entitiesWithAllPropertyTypes.map((e) => e.key).toList();
+          return testLookup(keys, entitiesWithAllPropertyTypes).then((_) {
+            return delete(keys, transactional: false);
+          });
+        });
+      });
+
+      test('lookup_transactional', () {
+        return insert([], unnamedEntities1).then((keys) {
+          keys.forEach((key) => expect(isValidKey(key), isTrue));
+          return testLookup(keys, unnamedEntities1, transactional: true)
+              .then((_)  => delete(keys));
+        });
+      });
+
+      test('lookup_transactional_xg', () {
+        return insert([], unnamedEntities5).then((keys) {
+          keys.forEach((key) => expect(isValidKey(key), isTrue));
+          return testLookup(
+              keys, unnamedEntities5, transactional: true, xg: true).then((_) {
+            return delete(keys);
+          });
+        });
+      });
+
+      // TODO: ancestor lookups, string id lookups
+    });
+
+    group('delete', () {
+      Future testDelete(List<Key> keys,
+                       {bool transactional: false, bool xg: false}) {
+        Future test(Transaction transaction) {
+          return datastore.commit(deletes: keys).then((_) {
+            if (transaction != null) {
+              return datastore.commit(transaction: transaction);
+            }
+          });
+        }
+
+        if (transactional) {
+          return withTransaction(test, xg: xg);
+        }
+        return test(null);
+      }
+
+      var unnamedEntities1 = buildEntities(42, 43);
+      var unnamedEntities5 = buildEntities(1, 6);
+      var unnamedEntities99 = buildEntities(6, 106);
+
+      test('delete', () {
+        return insert([], unnamedEntities99, transactional: false).then((keys) {
+          keys.forEach((key) => expect(isValidKey(key), isTrue));
+          return lookup(keys, transactional: false).then((entities) {
+            entities.forEach((e) => expect(e, isNotNull));
+            return testDelete(keys).then((_) {
+              return lookup(keys, transactional: false).then((entities) {
+               entities.forEach((e) => expect(e, isNull));
+              });
+            });
+          });
+        });
+      });
+
+      // This should not work with [unamedEntities20], but is working!
+      // FIXME TODO FIXME : look into this.
+      test('delete_transactional', () {
+        return insert([], unnamedEntities99, transactional: false).then((keys) {
+          keys.forEach((key) => expect(isValidKey(key), isTrue));
+          return lookup(keys, transactional: false).then((entities) {
+            entities.forEach((e) => expect(e, isNotNull));
+            return testDelete(keys, transactional: true).then((_) {
+              return lookup(keys, transactional: false).then((entities) {
+                entities.forEach((e) => expect(e, isNull));
+              });
+            });
+          });
+        });
+      });
+
+      test('delete_transactional_xg', () {
+        return insert([], unnamedEntities99, transactional: false).then((keys) {
+          keys.forEach((key) => expect(isValidKey(key), isTrue));
+          return lookup(keys, transactional: false).then((entities) {
+            expect(entities.length, equals(unnamedEntities99.length));
+            entities.forEach((e) => expect(e, isNotNull));
+            return testDelete(keys, transactional: true, xg: true).then((_) {
+              return lookup(keys, transactional: false).then((entities) {
+                expect(entities.length, equals(unnamedEntities99.length));
+                entities.forEach((e) => expect(e, isNull));
+              });
+            });
+          });
+        });
+      });
+
+      // TODO: ancestor deletes, string id deletes
+    });
+
+    group('rollback', () {
+      Future testRollback(List<Key> keys, {bool xg: false}) {
+        return withTransaction((Transaction transaction) {
+          return datastore.lookup(keys, transaction: transaction)
+              .then((List<Entity> entitites) {
+            return datastore.rollback(transaction);
+          });
+        }, xg: xg);
+      }
+
+      var namedEntities1 = buildEntities(42, 43, idFunction: (i) => "i$i");
+      var namedEntities5 = buildEntities(1, 6, idFunction: (i) => "i$i");
+
+      var namedEntities1Keys = namedEntities1.map((e) => e.key).toList();
+      var namedEntities5Keys = namedEntities5.map((e) => e.key).toList();
+
+      test('rollback', () {
+        return testRollback(namedEntities1Keys);
+      });
+
+      test('rollback_xg', () {
+        return testRollback(namedEntities5Keys, xg: true);
+      });
+    });
+
+    group('empty_commit', () {
+      Future testEmptyCommit(
+          List<Key> keys, {bool transactional: false, bool xg: false}) {
+        Future test(Transaction transaction) {
+          return datastore.lookup(keys, transaction: transaction)
+              .then((List<Entity> entitites) {
+            return datastore.commit(transaction: transaction);
+          });
+        }
+
+        if (transactional) {
+          return withTransaction(test, xg: xg);
+        } else {
+          return test(null);
+        }
+      }
+
+      var namedEntities1 = buildEntities(42, 43, idFunction: (i) => "i$i");
+      var namedEntities5 = buildEntities(1, 6, idFunction: (i) => "i$i");
+      var namedEntities20 = buildEntities(6, 26, idFunction: (i) => "i$i");
+
+      var namedEntities1Keys = namedEntities1.map((e) => e.key).toList();
+      var namedEntities5Keys = namedEntities5.map((e) => e.key).toList();
+      var namedEntities20Keys = namedEntities20.map((e) => e.key).toList();
+
+      test('empty_commit', () {
+        return testEmptyCommit(namedEntities20Keys);
+      });
+
+      test('empty_commit_transactional', () {
+        return testEmptyCommit(namedEntities1Keys);
+      });
+
+      test('empty_commit_transactional_xg', () {
+        return testEmptyCommit(namedEntities5Keys);
+      });
+
+      test('negative_empty_commit_xg', () {
+        expect(testEmptyCommit(
+               namedEntities20Keys, transactional: true, xg: true),
+               throwsA(isApplicationError));
+      });
+    });
+
+    group('conflicting_transaction', () {
+      Future testConflictingTransaction(
+          List<Entity> entities, {bool xg: false}) {
+        Future test(
+            List<Entity> entities, Transaction transaction, value) {
+
+          // Change entities:
+          var changedEntities = new List<Entity>(entities.length);
+          for (int i = 0; i < entities.length; i++) {
+            var entity = entities[i];
+            var newProperties = new Map.from(entity.properties);
+            for (var prop in newProperties.keys) {
+              newProperties[prop] = "${newProperties[prop]}conflict$value";
+            }
+            changedEntities[i] =
+                new Entity(entity.key, newProperties);
+          }
+          return datastore.commit(inserts: changedEntities,
+                                  transaction: transaction);
+        }
+
+        // Insert first
+        return insert(entities, [], transactional: true).then((_) {
+          var keys = entities.map((e) => e.key).toList();
+
+          var NUM_TRANSACTIONS = 10;
+
+          // Start transactions
+          var transactions = [];
+          for (var i = 0; i < NUM_TRANSACTIONS; i++) {
+            transactions.add(datastore.beginTransaction(crossEntityGroup: xg));
+          }
+          return Future.wait(transactions)
+              .then((List<Transaction> transactions) {
+            // Do a lookup for the entities in every transaction
+            var lookups = [];
+            for (var transaction in transactions) {
+              lookups.add(
+                  datastore.lookup(keys, transaction: transaction));
+            }
+            return Future.wait(lookups).then((List<List<Entity>> results) {
+              // Do a conflicting commit in every transaction.
+              var commits = [];
+              for (var i = 0; i < transactions.length; i++) {
+                var transaction = transactions[i];
+                commits.add(test(results[i], transaction, i));
+              }
+              return Future.wait(commits);
+            });
+          });
+        });
+      }
+
+      var namedEntities1 = buildEntities(42, 43, idFunction: (i) => "i$i");
+      var namedEntities5 = buildEntities(1, 6, idFunction: (i) => "i$i");
+
+      test('conflicting_transaction', () {
+        expect(testConflictingTransaction(namedEntities1),
+               throwsA(isTransactionAbortedError));
+      });
+
+      test('conflicting_transaction_xg', () {
+        expect(testConflictingTransaction(namedEntities5, xg: true),
+               throwsA(isTransactionAbortedError));
+      });
+    });
+
+    group('query', () {
+      Future testQuery(String kind,
+                       {List<Filter> filters,
+                        List<Order> orders,
+                        bool transactional: false,
+                        bool xg: false,
+                        int offset,
+                        int limit}) {
+        Future<List<Entity>> test(Transaction transaction) {
+          var query = new Query(
+              kind: kind, filters: filters, orders: orders,
+              offset: offset, limit: limit);
+          return consumePages((_) => datastore.query(query))
+              .then((List<Entity> entities) {
+            if (transaction != null) {
+              return datastore.commit(transaction: transaction)
+                  .then((_) => entities);
+            }
+            return entities;
+          });
+        }
+
+        if (transactional) {
+          return withTransaction(test, xg: xg);
+        }
+        return test(null);
+      }
+
+      Future testQueryAndCompare(String kind,
+                                 List<Entity> expectedEntities,
+                                 {List<Filter> filters,
+                                  List<Order> orders,
+                                  bool transactional: false,
+                                  bool xg: false,
+                                  bool correctOrder: true,
+                                  int offset,
+                                  int limit}) {
+        return testQuery(kind,
+                         filters: filters,
+                         orders: orders,
+                         transactional: transactional,
+                         xg: xg,
+                         offset: offset,
+                         limit: limit).then((List<Entity> entities) {
+          expect(entities.length, equals(expectedEntities.length));
+
+          if (correctOrder) {
+            for (int i = 0; i < entities.length; i++) {
+              expect(compareEntity(entities[i], expectedEntities[i]), isTrue);
+            }
+          } else {
+            for (int i = 0; i < entities.length; i++) {
+              bool found = false;
+              for (int j = 0; j < expectedEntities.length; j++) {
+               if (compareEntity(entities[i], expectedEntities[i])) {
+                 found = true;
+               }
+              }
+              expect(found, isTrue);
+            }
+          }
+        });
+      }
+      Future testOffsetLimitQuery(String kind,
+                                  List<Entity> expectedEntities,
+                                  {List<Order> orders,
+                                   bool transactional: false,
+                                   bool xg: false}) {
+        // We query for all subsets of expectedEntities
+        // NOTE: This is O(0.5 * n^2) queries, but n is currently only 6.
+        List<Function> queryTests = [];
+        for (int start = 0; start < expectedEntities.length; start++) {
+          for (int end = start; end < expectedEntities.length; end++) {
+            int offset = start;
+            int limit = end - start;
+            var entities = expectedEntities.sublist(offset, offset + limit);
+            queryTests.add(() {
+              return testQueryAndCompare(
+                   kind, entities, transactional: transactional,
+                   xg: xg, orders: orders,
+                   offset: offset, limit: limit);
+            });
+          }
+        }
+        // Query with limit higher than the number of results.
+        queryTests.add(() {
+          return testQueryAndCompare(
+               kind, expectedEntities, transactional: transactional,
+               xg: xg, orders: orders,
+               offset: 0, limit: expectedEntities.length * 10);
+        });
+
+        return Future.forEach(queryTests, (f) => f());
+      }
+
+      const TEST_QUERY_KIND = 'TestQueryKind';
+      var stringNamedEntities = buildEntities(
+          1, 6, idFunction: (i) => 'str$i', kind: TEST_QUERY_KIND);
+      var stringNamedKeys = stringNamedEntities.map((e) => e.key).toList();
+
+      var QUERY_KEY = TEST_PROPERTY_KEY_PREFIX;
+      var QUERY_UPPER_BOUND = "${TEST_PROPERTY_VALUE_PREFIX}4";
+      var QUERY_LOWER_BOUND = "${TEST_PROPERTY_VALUE_PREFIX}1";
+      var QUERY_LIST_ENTRY = '${TEST_LIST_VALUE}2';
+      var QUERY_INDEX_VALUE = '${TEST_INDEXED_PROPERTY_VALUE_PREFIX}1';
+
+      var reverseOrderFunction = (Entity a, Entity b) {
+        // Reverse the order
+        return -1 * (a.properties[QUERY_KEY] as String)
+            .compareTo(b.properties[QUERY_KEY]);
+      };
+
+      var filterFunction = (Entity entity) {
+        var value = entity.properties[QUERY_KEY];
+        return value.compareTo(QUERY_UPPER_BOUND) == -1 &&
+               value.compareTo(QUERY_LOWER_BOUND) == 1;
+      };
+      var listFilterFunction = (Entity entity) {
+        var values = entity.properties[TEST_LIST_PROPERTY];
+        return values.contains(QUERY_LIST_ENTRY);
+      };
+      var indexFilterMatches = (Entity entity) {
+        return entity.properties[TEST_INDEXED_PROPERTY] == QUERY_INDEX_VALUE;
+      };
+
+      var sorted = stringNamedEntities.toList()..sort(reverseOrderFunction);
+      var filtered = stringNamedEntities.where(filterFunction).toList();
+      var sortedAndFiltered = sorted.where(filterFunction).toList();
+      var sortedAndListFiltered = sorted.where(listFilterFunction).toList();
+      var indexedEntity = sorted.where(indexFilterMatches).toList();
+      expect(indexedEntity.length, equals(1));
+
+      var filters = [
+          new Filter(FilterRelation.GreatherThan, QUERY_KEY, QUERY_LOWER_BOUND),
+          new Filter(FilterRelation.LessThan, QUERY_KEY, QUERY_UPPER_BOUND),
+      ];
+      var listFilters = [
+          new Filter(FilterRelation.In, TEST_LIST_PROPERTY, [QUERY_LIST_ENTRY])
+      ];
+      var indexedPropertyFilter = [
+        new Filter(FilterRelation.Equal,
+                   TEST_INDEXED_PROPERTY,
+                   QUERY_INDEX_VALUE),
+        new Filter(FilterRelation.Equal,
+                   TEST_BLOB_INDEXED_PROPERTY,
+                   TEST_BLOB_INDEXED_VALUE)
+      ];
+      var unIndexedPropertyFilter = [
+        new Filter(FilterRelation.Equal,
+                   TEST_UNINDEXED_PROPERTY,
+                   QUERY_INDEX_VALUE)
+      ];
+
+      var orders = [new Order(OrderDirection.Decending, QUERY_KEY)];
+
+      test('query', () {
+        return insert(stringNamedEntities, []).then((keys) {
+          return waitUntilEntitiesReady(datastore, stringNamedKeys).then((_) {
+            var tests = [
+              // EntityKind query
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, stringNamedEntities, transactional: false,
+                  correctOrder: false),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, stringNamedEntities, transactional: true,
+                  correctOrder: false),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, stringNamedEntities, transactional: true,
+                  correctOrder: false, xg: true),
+
+              // EntityKind query with order
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sorted, transactional: false,
+                  orders: orders),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sorted, transactional: true,
+                  orders: orders),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sorted, transactional: false, xg: true,
+                  orders: orders),
+
+              // EntityKind query with filter
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, filtered, transactional: false,
+                  filters: filters),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, filtered, transactional: true,
+                  filters: filters),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, filtered, transactional: false, xg: true,
+                  filters: filters),
+
+              // EntityKind query with filter + order
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sortedAndFiltered, transactional: false,
+                  filters: filters, orders: orders),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sortedAndFiltered, transactional: true,
+                  filters: filters, orders: orders),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sortedAndFiltered, transactional: false,
+                  xg: true, filters: filters, orders: orders),
+
+              // EntityKind query with IN filter + order
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sortedAndListFiltered, transactional: false,
+                  filters: listFilters, orders: orders),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sortedAndListFiltered, transactional: true,
+                  filters: listFilters, orders: orders),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, sortedAndListFiltered, transactional: false,
+                  xg: true, filters: listFilters, orders: orders),
+
+              // Limit & Offset test
+              () => testOffsetLimitQuery(
+                  TEST_QUERY_KIND, sorted, transactional: false,
+                  orders: orders),
+              () => testOffsetLimitQuery(
+                  TEST_QUERY_KIND, sorted, transactional: true, orders: orders),
+              () => testOffsetLimitQuery(
+                  TEST_QUERY_KIND, sorted, transactional: false,
+                  xg: true, orders: orders),
+
+              // Query for indexed property
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, indexedEntity, transactional: false,
+                  filters: indexedPropertyFilter),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, indexedEntity, transactional: true,
+                  filters: indexedPropertyFilter),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, indexedEntity, transactional: false,
+                  xg: true, filters: indexedPropertyFilter),
+
+              // Query for un-indexed property
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: false,
+                  filters: unIndexedPropertyFilter),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: true,
+                  filters: unIndexedPropertyFilter),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: false,
+                  xg: true, filters: unIndexedPropertyFilter),
+
+              // Delete results
+              () => delete(stringNamedKeys, transactional: true),
+
+              // Wait until the entity deletes are reflected in the indices.
+              () => waitUntilEntitiesGone(datastore, stringNamedKeys),
+
+              // Make sure queries don't return results
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: false),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: true),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: true, xg: true),
+              () => testQueryAndCompare(
+                  TEST_QUERY_KIND, [], transactional: false,
+                  filters: filters, orders: orders),
+            ];
+            return Future.forEach(tests, (f) => f());
+          });
+        });
+
+        // TODO: query by multiple keys, multiple sort oders, ...
+      });
+
+      test('ancestor_query', () {
+        /*
+         * This test creates an
+         * RootKind:1 -- This defines the entity group (no entity with that key)
+         *    + SubKind:1  -- This a subpath (no entity with that key)
+         *        + SubSubKind:1  -- This is a real entity of kind SubSubKind
+         *        + SubSubKind2:1 -- This is a real entity of kind SubSubKind2
+         */
+        var rootKey = new Key.fromParent('RootKind', 1);
+        var subKey = new Key.fromParent('SubKind', 1, parent: rootKey);
+        var subSubKey = new Key.fromParent('SubSubKind', 1, parent: subKey);
+        var subSubKey2 = new Key.fromParent('SubSubKind2', 1, parent: subKey);
+        var properties = { 'foo' : 'bar' };
+
+        var entity = new Entity(subSubKey, properties);
+        var entity2 = new Entity(subSubKey2, properties);
+
+        var orders = [new Order(OrderDirection.Ascending, '__key__')];
+
+        return datastore.commit(inserts: [entity, entity2]).then((_) {
+          var futures = [
+            // FIXME/TODO: Ancestor queries should be strongly consistent.
+            // We should not need to wait for them.
+            () {
+              return waitUntilEntitiesReady(datastore, [subSubKey, subSubKey2]);
+            },
+            // Test that lookup only returns inserted entities.
+            () {
+              return datastore.lookup([rootKey, subKey, subSubKey, subSubKey2])
+                  .then((List<Entity> entities) {
+                expect(entities.length, 4);
+                expect(entities[0], isNull);
+                expect(entities[1], isNull);
+                expect(entities[2], isNotNull);
+                expect(entities[3], isNotNull);
+                expect(compareEntity(entity, entities[2]), isTrue);
+                expect(compareEntity(entity2, entities[3]), isTrue);
+              });
+            },
+
+            // Query by ancestor.
+            // - by [rootKey]
+            () {
+              var ancestorQuery =
+                  new Query(ancestorKey: rootKey, orders: orders);
+              return consumePages((_) => datastore.query(ancestorQuery))
+                  .then((results) {
+                expect(results.length, 2);
+                expect(compareEntity(entity, results[0]), isTrue);
+                expect(compareEntity(entity2, results[1]), isTrue);
+              });
+            },
+            // - by [subKey]
+            () {
+              var ancestorQuery =
+                  new Query(ancestorKey: subKey, orders: orders);
+              return consumePages((_) => datastore.query(ancestorQuery))
+                  .then((results) {
+                expect(results.length, 2);
+                expect(compareEntity(entity, results[0]), isTrue);
+                expect(compareEntity(entity2, results[1]), isTrue);
+              });
+            },
+            // - by [subSubKey]
+            () {
+              var ancestorQuery = new Query(ancestorKey: subSubKey);
+              return consumePages((_) => datastore.query(ancestorQuery))
+                  .then((results) {
+                expect(results.length, 1);
+                expect(compareEntity(entity, results[0]), isTrue);
+              });
+            },
+            // - by [subSubKey2]
+            () {
+              var ancestorQuery = new Query(ancestorKey: subSubKey2);
+              return consumePages((_) => datastore.query(ancestorQuery))
+                  .then((results) {
+                expect(results.length, 1);
+                expect(compareEntity(entity2, results[0]), isTrue);
+              });
+            },
+
+            // Query by ancestor and kind.
+            // - by [rootKey] + 'SubSubKind'
+            () {
+              var query = new Query(ancestorKey: rootKey, kind: 'SubSubKind');
+              return consumePages((_) => datastore.query(query))
+                  .then((List<Entity> results) {
+                expect(results.length, 1);
+                expect(compareEntity(entity, results[0]), isTrue);
+              });
+            },
+            // - by [rootKey] + 'SubSubKind2'
+            () {
+              var query = new Query(ancestorKey: rootKey, kind: 'SubSubKind2');
+              return consumePages((_) => datastore.query(query))
+                  .then((List<Entity> results) {
+                expect(results.length, 1);
+                expect(compareEntity(entity2, results[0]), isTrue);
+              });
+            },
+            // - by [subSubKey] + 'SubSubKind'
+            () {
+              var query = new Query(ancestorKey: subSubKey, kind: 'SubSubKind');
+              return consumePages((_) => datastore.query(query))
+                  .then((List<Entity> results) {
+                expect(results.length, 1);
+                expect(compareEntity(entity, results[0]), isTrue);
+              });
+            },
+            // - by [subSubKey2] + 'SubSubKind2'
+            () {
+              var query =
+                  new Query(ancestorKey: subSubKey2, kind: 'SubSubKind2');
+              return consumePages((_) => datastore.query(query))
+                  .then((List<Entity> results) {
+                expect(results.length, 1);
+                expect(compareEntity(entity2, results[0]), isTrue);
+              });
+            },
+            // - by [subSubKey] + 'SubSubKind2'
+            () {
+              var query =
+                  new Query(ancestorKey: subSubKey, kind: 'SubSubKind2');
+              return consumePages((_) => datastore.query(query))
+                  .then((List<Entity> results) {
+                expect(results.length, 0);
+              });
+            },
+            // - by [subSubKey2] + 'SubSubKind'
+            () {
+              var query =
+                  new Query(ancestorKey: subSubKey2, kind: 'SubSubKind');
+              return consumePages((_) => datastore.query(query))
+                  .then((List<Entity> results) {
+                expect(results.length, 0);
+              });
+            },
+
+            // Cleanup
+            () {
+              return datastore.commit(deletes: [subSubKey, subSubKey2]);
+            }
+          ];
+          return Future.forEach(futures, (f) => f()).then(expectAsync((_) {}));
+        });
+      });
+    });
+  });
+}
+
+Future cleanupDB(Datastore db) {
+  Future<List<String>> getNamespaces() {
+    var q = new Query(kind: '__namespace__');
+    return consumePages((_) => db.query(q)).then((List<Entity> entities) {
+      return entities.map((Entity e) {
+        var id = e.key.elements.last.id;
+        if (id == 1) return null;
+        return id;
+      }).toList();
+    });
+  }
+
+  Future<List<String>> getKinds(String namespace) {
+    var partition = new Partition(namespace);
+    var q = new Query(kind: '__kind__');
+    return consumePages((_) => db.query(q, partition: partition))
+        .then((List<Entity> entities) {
+      return entities
+          .map((Entity e) => e.key.elements.last.id)
+          .where((String kind) => !kind.contains('__'))
+          .toList();
+    });
+  }
+
+  // cleanup() will call itself again as long as the DB is not clean.
+  cleanup(String namespace, String kind) {
+    var partition = new Partition(namespace);
+    var q = new Query(kind: kind, limit: 500);
+    return consumePages((_) => db.query(q, partition: partition))
+        .then((List<Entity> entities) {
+      if (entities.length == 0) return null;
+
+      print('[cleanupDB]: Removing left-over ${entities.length} entities');
+      var deletes = entities.map((e) => e.key).toList();
+      return db.commit(deletes: deletes).then((_) => cleanup(namespace, kind));
+    });
+  }
+
+  return getNamespaces().then((List<String> namespaces) {
+    return Future.forEach(namespaces, (String namespace) {
+      return getKinds(namespace).then((List<String> kinds) {
+        return Future.forEach(kinds, (String kind) {
+          return cleanup(namespace, kind);
+        });
+      });
+    });
+  });
+}
+
+Future waitUntilEntitiesReady(Datastore db, List<Key> keys) {
+  return waitUntilEntitiesHelper(db, keys, true);
+}
+
+Future waitUntilEntitiesGone(Datastore db, List<Key> keys) {
+  return waitUntilEntitiesHelper(db, keys, false);
+}
+
+Future waitUntilEntitiesHelper(Datastore db, List<Key> keys, bool positive) {
+  var keysByKind = {};
+  for (var key in keys) {
+    keysByKind.putIfAbsent(key.elements.last.kind, () => []).add(key);
+  }
+
+  Future waitForKeys(String kind, List<Key> keys) {
+    var q = new Query(kind: kind);
+    return consumePages((_) => db.query(q)).then((entities) {
+      for (var key in keys) {
+        bool found = false;
+        for (var entity in entities) {
+          if (key == entity.key) found = true;
+        }
+        if (positive) {
+          if (!found) return waitForKeys(kind, keys);
+        } else {
+          if (found) return waitForKeys(kind, keys);
+        }
+      }
+      return null;
+    });
+  }
+
+  return Future.forEach(keysByKind.keys.toList(), (String kind) {
+    return waitForKeys(kind, keysByKind[kind]);
+  });
+}