changed compression request streaming enumerable to set write flag on…

… every message
dexteryuen · Jul 12, 2016 · 535b71d · 535b71d
1 parent 3403565
commit 535b71d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ruby/pb/test/client.rb b/src/ruby/pb/test/client.rb
@@ -244,8 +244,7 @@ def initialize(requests_and_write_flags)
 
   def each
     @requests_and_write_flags.each do |request_and_flag|
-      @call_op.write_flag = request_and_flag[:write_flag] if
-        request_and_flag[:write_flag]
+      @call_op.write_flag = request_and_flag[:write_flag]
       yield request_and_flag[:request]
     end
   end
@@ -411,7 +410,8 @@ def client_compressed_streaming
     # Create the requests messages and the corresponding write flags
     # for each message
     requests = WriteFlagSettingStreamingInputEnumerable.new([
-      { request: first_request },
+      { request: first_request,
+        write_flag: 0 },
       { request: second_request,
         write_flag: GRPC::Core::WriteFlags::NO_COMPRESS }
     ])