Method: NamedTests#custom_metadata

Defined in:

src/ruby/pb/test/client.rb

#custom_metadata ⇒ Object

# File 'src/ruby/pb/test/client.rb', line 584

def custom_metadata

  # Function wide constants
  req_size, wanted_response_size = 271_828, 314_159
  initial_metadata_key = "x-grpc-test-echo-initial"
  initial_metadata_value = "test_initial_metadata_value"
  trailing_metadata_key = "x-grpc-test-echo-trailing-bin"
  trailing_metadata_value = "\x0a\x0b\x0a\x0b\x0a\x0b"

  metadata = {
    initial_metadata_key => initial_metadata_value,
    trailing_metadata_key => trailing_metadata_value
  }

  # Testing with UnaryCall
  payload = Payload.new(type: :COMPRESSABLE, body: nulls(req_size))
  req = SimpleRequest.new(response_type: :COMPRESSABLE,
      response_size: wanted_response_size,
      payload: payload)

  op = @stub.unary_call(req, metadata: metadata, return_op: true)
  op.execute
  if not op.metadata.has_key?(initial_metadata_key)
    fail AssertionError, "Expected initial metadata. None received"
  elsif op.metadata[initial_metadata_key] != metadata[initial_metadata_key]
    fail AssertionError,
           "Expected initial metadata: #{metadata[initial_metadata_key]}. "\
           "Received: #{op.metadata[initial_metadata_key]}"
  end
  if not op.trailing_metadata.has_key?(trailing_metadata_key)
    fail AssertionError, "Expected trailing metadata. None received"
  elsif op.trailing_metadata[trailing_metadata_key] !=
        metadata[trailing_metadata_key]
    fail AssertionError,
          "Expected trailing metadata: #{metadata[trailing_metadata_key]}. "\
          "Received: #{op.trailing_metadata[trailing_metadata_key]}"
  end

  # Testing with FullDuplex
  req_cls, p_cls = StreamingOutputCallRequest, ResponseParameters
  duplex_req = req_cls.new(payload: Payload.new(body: nulls(req_size)),
                response_type: :COMPRESSABLE,
                response_parameters: [p_cls.new(size: wanted_response_size)])

  duplex_op = @stub.full_duplex_call([duplex_req], metadata: metadata,
                                      return_op: true)
  resp = duplex_op.execute
  resp.each { |r| } # ensures that the server sends trailing data
  duplex_op.wait
  if not duplex_op.metadata.has_key?(initial_metadata_key)
    fail AssertionError, "Expected initial metadata. None received"
  elsif duplex_op.metadata[initial_metadata_key] !=
        metadata[initial_metadata_key]
    fail AssertionError,
           "Expected initial metadata: #{metadata[initial_metadata_key]}. "\
           "Received: #{duplex_op.metadata[initial_metadata_key]}"
  end
  if not duplex_op.trailing_metadata[trailing_metadata_key]
    fail AssertionError, "Expected trailing metadata. None received"
  elsif duplex_op.trailing_metadata[trailing_metadata_key] !=
        metadata[trailing_metadata_key]
    fail AssertionError,
        "Expected trailing metadata: #{metadata[trailing_metadata_key]}. "\
        "Received: #{duplex_op.trailing_metadata[trailing_metadata_key]}"
  end

end