From 7d474811ab068580919df1f9027ee774b4d24f46 Mon Sep 17 00:00:00 2001
From: ncteisen <ncteisen@gmail.com>
Date: Fri, 29 Jun 2018 14:59:59 -0700
Subject: [PATCH] Reviewer feedback

---
 .../grpcpp/impl/codegen/rpc_service_method.h  |   2 +-
 test/cpp/end2end/raw_end2end_test.cc          | 105 +-----------------
 2 files changed, 7 insertions(+), 100 deletions(-)

diff --git a/include/grpcpp/impl/codegen/rpc_service_method.h b/include/grpcpp/impl/codegen/rpc_service_method.h
index 92c66bbf15b..5cf88e216f9 100644
--- a/include/grpcpp/impl/codegen/rpc_service_method.h
+++ b/include/grpcpp/impl/codegen/rpc_service_method.h
@@ -95,7 +95,7 @@ class RpcServiceMethod : public RpcMethod {
 
  private:
   void* server_tag_;
-  RpcServiceMethod::AsyncType async_type_;
+  AsyncType async_type_;
   std::unique_ptr<MethodHandler> handler_;
 
   const char* TypeToString(RpcServiceMethod::AsyncType type) {
diff --git a/test/cpp/end2end/raw_end2end_test.cc b/test/cpp/end2end/raw_end2end_test.cc
index 37fdc423fd7..a413905ef7a 100644
--- a/test/cpp/end2end/raw_end2end_test.cc
+++ b/test/cpp/end2end/raw_end2end_test.cc
@@ -55,26 +55,11 @@ int detag(void* p) { return static_cast<int>(reinterpret_cast<intptr_t>(p)); }
 
 class Verifier {
  public:
-  Verifier() : lambda_run_(false) {}
+  Verifier() {}
+
   // Expect sets the expected ok value for a specific tag
   Verifier& Expect(int i, bool expect_ok) {
-    return ExpectUnless(i, expect_ok, false);
-  }
-  // ExpectUnless sets the expected ok value for a specific tag
-  // unless the tag was already marked seen (as a result of ExpectMaybe)
-  Verifier& ExpectUnless(int i, bool expect_ok, bool seen) {
-    if (!seen) {
-      expectations_[tag(i)] = expect_ok;
-    }
-    return *this;
-  }
-  // ExpectMaybe sets the expected ok value for a specific tag, but does not
-  // require it to appear
-  // If it does, sets *seen to true
-  Verifier& ExpectMaybe(int i, bool expect_ok, bool* seen) {
-    if (!*seen) {
-      maybe_expectations_[tag(i)] = MaybeExpect{expect_ok, seen};
-    }
+    expectations_[tag(i)] = expect_ok;
     return *this;
   }
 
@@ -88,69 +73,12 @@ class Verifier {
     return detag(got_tag);
   }
 
-  template <typename T>
-  CompletionQueue::NextStatus DoOnceThenAsyncNext(
-      CompletionQueue* cq, void** got_tag, bool* ok, T deadline,
-      std::function<void(void)> lambda) {
-    if (lambda_run_) {
-      return cq->AsyncNext(got_tag, ok, deadline);
-    } else {
-      lambda_run_ = true;
-      return cq->DoThenAsyncNext(lambda, got_tag, ok, deadline);
-    }
-  }
-
   // Verify keeps calling Next until all currently set
   // expected tags are complete
-  void Verify(CompletionQueue* cq) { Verify(cq, false); }
-
-  // This version of Verify allows optionally ignoring the
-  // outcome of the expectation
-  void Verify(CompletionQueue* cq, bool ignore_ok) {
-    GPR_ASSERT(!expectations_.empty() || !maybe_expectations_.empty());
+  void Verify(CompletionQueue* cq) {
+    GPR_ASSERT(!expectations_.empty());
     while (!expectations_.empty()) {
-      Next(cq, ignore_ok);
-    }
-  }
-
-  // This version of Verify stops after a certain deadline
-  void Verify(CompletionQueue* cq,
-              std::chrono::system_clock::time_point deadline) {
-    if (expectations_.empty()) {
-      bool ok;
-      void* got_tag;
-      EXPECT_EQ(cq->AsyncNext(&got_tag, &ok, deadline),
-                CompletionQueue::TIMEOUT);
-    } else {
-      while (!expectations_.empty()) {
-        bool ok;
-        void* got_tag;
-        EXPECT_EQ(cq->AsyncNext(&got_tag, &ok, deadline),
-                  CompletionQueue::GOT_EVENT);
-        GotTag(got_tag, ok, false);
-      }
-    }
-  }
-
-  // This version of Verify stops after a certain deadline, and uses the
-  // DoThenAsyncNext API
-  // to call the lambda
-  void Verify(CompletionQueue* cq,
-              std::chrono::system_clock::time_point deadline,
-              std::function<void(void)> lambda) {
-    if (expectations_.empty()) {
-      bool ok;
-      void* got_tag;
-      EXPECT_EQ(DoOnceThenAsyncNext(cq, &got_tag, &ok, deadline, lambda),
-                CompletionQueue::TIMEOUT);
-    } else {
-      while (!expectations_.empty()) {
-        bool ok;
-        void* got_tag;
-        EXPECT_EQ(DoOnceThenAsyncNext(cq, &got_tag, &ok, deadline, lambda),
-                  CompletionQueue::GOT_EVENT);
-        GotTag(got_tag, ok, false);
-      }
+      Next(cq, false);
     }
   }
 
@@ -162,31 +90,10 @@ class Verifier {
         EXPECT_EQ(it->second, ok);
       }
       expectations_.erase(it);
-    } else {
-      auto it2 = maybe_expectations_.find(got_tag);
-      if (it2 != maybe_expectations_.end()) {
-        if (it2->second.seen != nullptr) {
-          EXPECT_FALSE(*it2->second.seen);
-          *it2->second.seen = true;
-        }
-        if (!ignore_ok) {
-          EXPECT_EQ(it2->second.ok, ok);
-        }
-      } else {
-        gpr_log(GPR_ERROR, "Unexpected tag: %p", tag);
-        abort();
-      }
     }
   }
 
-  struct MaybeExpect {
-    bool ok;
-    bool* seen;
-  };
-
   std::map<void*, bool> expectations_;
-  std::map<void*, MaybeExpect> maybe_expectations_;
-  bool lambda_run_;
 };
 
 class RawEnd2EndTest : public ::testing::Test {