Revision d9492490 test/ganeti.rapi.client_unittest.py

b/test/ganeti.rapi.client_unittest.py
74 74
  def GetLastRequestData(self):
75 75
    return self._last_req_data
76 76

  
77
  def FetchResponse(self, path, method, request_body):
77
  def FetchResponse(self, path, method, headers, request_body):
78 78
    self._last_req_data = request_body
79 79

  
80 80
    try:
......
139 139
  def test(self):
140 140
    rapi = RapiMock()
141 141
    path = "/version"
142
    self.assertEqual((404, None), rapi.FetchResponse("/foo", "GET", None))
142
    self.assertEqual((404, None), rapi.FetchResponse("/foo", "GET", None, None))
143 143
    self.assertEqual((501, "Method not implemented"),
144
                     rapi.FetchResponse("/version", "POST", None))
144
                     rapi.FetchResponse("/version", "POST", None, None))
145 145
    rapi.AddResponse("2")
146
    code, response = rapi.FetchResponse("/version", "GET", None)
146
    code, response = rapi.FetchResponse("/version", "GET", None, None)
147 147
    self.assertEqual(200, code)
148 148
    self.assertEqual("2", response)
149 149
    self.failUnless(isinstance(rapi.GetLastHandler(), rlib2.R_version))

Also available in: Unified diff