Solution aggregation

samtstern · samtstern · commit e26834e539e1 · 2020-09-02T12:12:44.000+01:00
diff --git a/firestore/test.solution-aggregation.js b/firestore/test.solution-aggregation.js
@@ -1,34 +1,35 @@
 // [START sample_doc]
-var arinellDoc = {
+const arinellDoc = {
   name: 'Arinell Pizza',
   avgRating: 4.65,
   numRatings: 683
 }
 // [END sample_doc]
 
 describe("firestore-solution-arrays", () => {
-    var db;
-    before(() => {
-        var config = {
-            apiKey: "AIzaSyArvVh6VSdXicubcvIyuB-GZs8ua0m0DTI",
-            authDomain: "firestorequickstarts.firebaseapp.com",
-            projectId: "firestorequickstarts",
-        };
-        var app = firebase.initializeApp(config, "solution-arrays");
-        db = firebase.firestore(app);
+    let db;
+    before(async () => {
+      const { initializeApp } = require("@firebase/app");
+      const { getFirestore, collection, doc, setDoc } = require("@firebase/firestore");
 
-        return db.collection("restaurants")
-            .doc("arinell-pizza")
-            .set(arinellDoc);
+      const config = {
+          apiKey: "AIzaSyArvVh6VSdXicubcvIyuB-GZs8ua0m0DTI",
+          authDomain: "firestorequickstarts.firebaseapp.com",
+          projectId: "firestorequickstarts",
+      };
+      const app = initializeApp(config, "solution-arrays");
+      db = getFirestore(app);
+
+      await setDoc(doc(collection(db, "restaurants"), "arinell-pizza"), arinellDoc);
     });
 
     describe("solution-arrays", () => {
-        it("should get a collection of ratings", () => {
+        it("should get a collection of ratings", async () => {
           // [START get_collection_ratings]
-          db.collection("restaurants")
-            .doc("arinell-pizza")
-            .collection("ratings")
-            .get()
+          const { collection, doc, getDocs } = require("@firebase/firestore");
+
+          const ratingsRef = collection(doc(collection(db, "restaurants"), "arinell-pizza"), "ratings");
+          const ratingsDocs = await getDocs(ratingsRef);
           // [END get_collection_ratings]
         })
     });
