예제 #1
0
    def test_before_request_hook(self):
        momo = mock.Mock()

        before_request_hooks.append(no_hook)
        before_request_hooks.append(momo)

        balanced.http_client.get('hithere', )
        self.assertEqual(momo.call_count, 1)
        args, _ = momo.call_args
        self.assertEqual(args[0], balanced.http_client)
        self.assertIn('hithere', args[2])
예제 #2
0
    def test_before_request_hook(self):
        momo = mock.Mock()

        before_request_hooks.append(no_hook)
        before_request_hooks.append(momo)

        balanced.http_client.get(
            'hithere',
        )
        self.assertEqual(momo.call_count, 1)
        args, _ = momo.call_args
        self.assertEqual(args[0], balanced.http_client)
        self.assertIn('hithere', args[2])
예제 #3
0
 def setUp(self):
     before_request_hooks.append(no_hook)
예제 #4
0
 def setUp(self):
     before_request_hooks.append(no_hook)