Firestore: Add SCOPED_TRACE to Await() calls in firestore_integration_test.h/cc (#1051)

Author: dconeybe

firestore/integration_test_internal/src/firestore_integration_test.cc

@@ -237,6 +237,8 @@ DocumentReference FirestoreIntegrationTest::DocumentWithData(
 
 void FirestoreIntegrationTest::WriteDocument(DocumentReference reference,
                                              const MapFieldValue& data) const {
+  SCOPED_TRACE("FirestoreIntegrationTest::WriteDocument(" + reference.path() +
+               ")");
   Future<void> future = reference.Set(data);
   Stopwatch stopwatch;
   Await(future);
@@ -247,13 +249,17 @@ void FirestoreIntegrationTest::WriteDocument(DocumentReference reference,
 void FirestoreIntegrationTest::WriteDocuments(
     CollectionReference reference,
     const std::map<std::string, MapFieldValue>& data) const {
+  SCOPED_TRACE("FirestoreIntegrationTest::WriteDocuments(" +
+               std::to_string(data.size()) + " documents)");
   for (const auto& kv : data) {
     WriteDocument(reference.Document(kv.first), kv.second);
   }
 }
 
 DocumentSnapshot FirestoreIntegrationTest::ReadDocument(
     const DocumentReference& reference) const {
+  SCOPED_TRACE("FirestoreIntegrationTest::ReadDocument(" + reference.path() +
+               ")");
   Future<DocumentSnapshot> future = reference.Get();
   Stopwatch stopwatch;
   const DocumentSnapshot* result = Await(future);
@@ -267,6 +273,7 @@ DocumentSnapshot FirestoreIntegrationTest::ReadDocument(
 
 QuerySnapshot FirestoreIntegrationTest::ReadDocuments(
     const Query& reference) const {
+  SCOPED_TRACE("FirestoreIntegrationTest::ReadDocuments()");
   Future<QuerySnapshot> future = reference.Get();
   Stopwatch stopwatch;
   const QuerySnapshot* result = Await(future);
@@ -280,6 +287,8 @@ QuerySnapshot FirestoreIntegrationTest::ReadDocuments(
 
 void FirestoreIntegrationTest::DeleteDocument(
     DocumentReference reference) const {
+  SCOPED_TRACE("FirestoreIntegrationTest::DeleteDocument(" + reference.path() +
+               ")");
   Future<void> future = reference.Delete();
   Stopwatch stopwatch;
   Await(future);

firestore/integration_test_internal/src/firestore_integration_test.h

@@ -319,6 +319,8 @@ class FirestoreIntegrationTest : public testing::Test {
   // Update the specified document and wait for the update to complete.
   template <typename MapType>
   void UpdateDocument(DocumentReference reference, const MapType& data) const {
+    SCOPED_TRACE("FirestoreIntegrationTest::UpdateDocument(" +
+                 reference.path() + ")");
     Future<void> future = reference.Update(data);
     Await(future);
     EXPECT_EQ(FutureStatus::kFutureStatusComplete, future.status());
@@ -392,9 +394,15 @@ class FirestoreIntegrationTest : public testing::Test {
 
   static std::string DescribeFailedFuture(const FutureBase& future);
 
-  void DisableNetwork() { Await(TestFirestore()->DisableNetwork()); }
+  void DisableNetwork() {
+    SCOPED_TRACE("FirestoreIntegrationTest::DisableNetwork()");
+    Await(TestFirestore()->DisableNetwork());
+  }
 
-  void EnableNetwork() { Await(TestFirestore()->EnableNetwork()); }
+  void EnableNetwork() {
+    SCOPED_TRACE("FirestoreIntegrationTest::EnableNetwork()");
+    Await(TestFirestore()->EnableNetwork());
+  }
 
   static FirestoreInternal* GetFirestoreInternal(Firestore* firestore) {
     return firestore->internal_;