Make sure that some ops don't start concurrently with Finish

test/cpp/end2end/test_service_impl.cc

@@ -502,12 +502,20 @@ CallbackTestServiceImpl::ResponseStream(
       if (num_msgs_sent_ == server_responses_to_send_ - 1 &&
           server_coalescing_api_ != 0) {
         num_msgs_sent_++;
-        StartWriteLast(&response_, WriteOptions());
+        {
+          std::lock_guard<std::mutex> l(finish_mu_);
+          if (!finished_) {
+            StartWriteLast(&response_, WriteOptions());
+          }
+        }
         // If we use WriteLast, we shouldn't wait before attempting Finish
         FinishOnce(Status::OK);
       } else {
         num_msgs_sent_++;
-        StartWrite(&response_);
+        std::lock_guard<std::mutex> l(finish_mu_);
+        if (!finished_) {
+          StartWrite(&response_);
+        }
       }
     }
     experimental::CallbackServerContext* const ctx_;
@@ -571,12 +579,15 @@ CallbackTestServiceImpl::BidiStream(
       if (ok) {
         num_msgs_read_++;
         response_.set_message(request_.message());
-        if (num_msgs_read_ == server_write_last_) {
+        std::lock_guard<std::mutex> l(finish_mu_);
-          StartWriteLast(&response_, WriteOptions());
+        if (!finished_) {
-          // If we use WriteLast, we shouldn't wait before attempting Finish
+          if (num_msgs_read_ == server_write_last_) {
-        } else {
+            StartWriteLast(&response_, WriteOptions());
-          StartWrite(&response_);
+            // If we use WriteLast, we shouldn't wait before attempting Finish
-          return;
+          } else {
+            StartWrite(&response_);
+            return;
+          }
         }
       }