def setUp(self):
loop = Mock()
self.handler = (lambda x: x)
self.exposer = Exposer(loop)
self.capability = Capability(
name='test',
method='GET',
endpoint='/test',
version='v1',
handler=self.handler,
wrapper=self.handler,
)
class TestExposer(TestCase):
def setUp(self):
loop = Mock()
self.handler = (lambda x: x)
self.exposer = Exposer(loop)
self.capability = Capability(
name='test',
method='GET',
endpoint='/test',
version='v1',
handler=self.handler,
wrapper=self.handler,
)
@patch('aiohttp.web_urldispatcher.UrlDispatcher.add_route')
def test_001_register(self, add_route):
self.exposer.register(self.capability)
self.assertRaises(ValueError, self.exposer.register, self.capability)
add_route.assert_called_with('GET', '/v1/test', self.handler)
def test_002_find(self):
self.assertIsNone(self.exposer._find('hello'))
self.exposer.register(self.capability)
self.assertEqual(self.exposer._find('test'), self.capability)
