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)