def test_lookup_test_helpers(self):
bouncer_client = BouncerClient('http://127.0.0.1:8888')
required_helpers = [u'http-return-json-headers', u'dns']
helpers = yield bouncer_client.lookupTestHelpers(required_helpers)
self.assertEqual(set(helpers.keys()), set(required_helpers + [u'default']))
self.assertTrue(helpers['http-return-json-headers']['address'].startswith('http'))
self.assertTrue(int(helpers['dns']['address'].split('.')[0]))
def test_bouncer_client_lookup_test_helpers(self):
bcc = BouncerClient('https://bouncer.ooni.io')
bcc.queryBackend = MagicMock()
bcc.queryBackend.return_value = defer.succeed({'spam': 'ham'})
yield bcc.lookupTestHelpers(["foo"])
bcc.queryBackend.assert_called_with("POST",
"/bouncer/test-helpers",
query={'test-helpers': ["foo"]})
def test_bouncer_client_lookup_test_helpers(self):
bcc = BouncerClient('https://bouncer.ooni.io')
bcc.queryBackend = MagicMock()
bcc.queryBackend.return_value = defer.succeed({'spam': 'ham'})
yield bcc.lookupTestHelpers(["foo"])
bcc.queryBackend.assert_called_with("POST",
"/bouncer/test-helpers",
query={'test-helpers': ["foo"]})
def test_lookup_test_helpers(self):
bouncer_client = BouncerClient('http://127.0.0.1:8888')
required_helpers = [u'http-return-json-headers', u'dns']
helpers = yield bouncer_client.lookupTestHelpers(required_helpers)
self.assertEqual(set(helpers.keys()),
set(required_helpers + [u'default']))
self.assertTrue(
helpers['http-return-json-headers']['address'].startswith('http'))
self.assertTrue(int(helpers['dns']['address'].split('.')[0]))
def test_lookup_no_test_helpers(self):
bouncer_client = BouncerClient('http://127.0.0.1:8888')
required_helpers = []
helpers = yield bouncer_client.lookupTestHelpers(required_helpers)
self.assertTrue('default' in helpers.keys())
def test_lookup_invalid_helpers(self):
bouncer_client = BouncerClient('http://127.0.0.1:8888')
return self.failUnlessFailure(
bouncer_client.lookupTestHelpers([
'sdadsadsa', 'dns'
]), e.CouldNotFindTestHelper)
def test_lookup_no_test_helpers(self):
bouncer_client = BouncerClient('http://127.0.0.1:8888')
required_helpers = []
helpers = yield bouncer_client.lookupTestHelpers(required_helpers)
self.assertTrue('default' in helpers.keys())
def test_lookup_invalid_helpers(self):
bouncer_client = BouncerClient('http://127.0.0.1:8888')
return self.failUnlessFailure(
bouncer_client.lookupTestHelpers(['sdadsadsa', 'dns']),
e.CouldNotFindTestHelper)
