Home | History | Annotate | Download | only in http2_test
      1 # Copyright 2016 gRPC authors.
      2 #
      3 # Licensed under the Apache License, Version 2.0 (the "License");
      4 # you may not use this file except in compliance with the License.
      5 # You may obtain a copy of the License at
      6 #
      7 #     http://www.apache.org/licenses/LICENSE-2.0
      8 #
      9 # Unless required by applicable law or agreed to in writing, software
     10 # distributed under the License is distributed on an "AS IS" BASIS,
     11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
     14 
     15 import http2_base_server
     16 
     17 class TestcaseRstStreamAfterHeader(object):
     18   """
     19     In response to an incoming request, this test sends headers, followed by
     20     a reset stream frame. Client asserts that the RPC failed.
     21   """
     22   def __init__(self):
     23     self._base_server = http2_base_server.H2ProtocolBaseServer()
     24     self._base_server._handlers['RequestReceived'] = self.on_request_received
     25 
     26   def get_base_server(self):
     27     return self._base_server
     28 
     29   def on_request_received(self, event):
     30     # send initial headers
     31     self._base_server.on_request_received_default(event)
     32     # send reset stream
     33     self._base_server.send_reset_stream()
     34