def test_stub_and_fetch(self): client = AsyncHTTPStubClient() stub("/hello").and_return(body="world") client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 200) self.assertEqual(response.body, "world")
def test_can_queue_stub_responses(self): client = AsyncHTTPStubClient() stub("/hello").and_return(body="hello")\ .and_return(body="beautiful")\ .and_return(body="world") # First Response client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 200) self.assertEqual(response.body, "hello") # Second Response client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 200) self.assertEqual(response.body, "beautiful") # Third Response client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 200) self.assertEqual(response.body, "world") # Fourth Response client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 200) self.assertEqual(response.body, "hello")
def test_with_syntax(self): client = AsyncHTTPStubClient() with stub("/hello").and_return(body=b"world"): client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 200) self.assertEqual(response.body, b"world") client.fetch("/hello", self.stop) response = self.wait() self.assertEqual(response.code, 404)