class ServiceAdapterTestCase(TestCase): @setup def setup_adapter(self): self.service = mock.MagicMock() self.adapter = ServiceAdapter(self.service) def test_repr(self): result = self.adapter.get_repr() assert_equal(result['name'], self.service.name) assert_equal(result['node_pool'], self.service.config.node)
class ServiceAdapterTestCase(TestCase): @setup def setup_adapter(self): self.service = mock.MagicMock() self.adapter = ServiceAdapter(self.service) @mock.patch('tron.api.adapter.NodePoolAdapter', autospec=True) def test_repr(self, mock_node_pool_adapter): result = self.adapter.get_repr() assert_equal(result['name'], self.service.name) assert_equal(result['node_pool'], mock_node_pool_adapter.return_value.get_repr.return_value)
def setup_adapter(self): self.service = mock.MagicMock() self.adapter = ServiceAdapter(self.service)