def setUp(self): self.devices = DeviceManager(mock.Mock()) self.device = mock.Mock(id='fakeid')
class TestDeviceManager(unittest2.TestCase): def setUp(self): self.devices = DeviceManager(mock.Mock()) self.device = mock.Mock(id='fakeid') @mock.patch('balancerclient.common.base.Manager._list', autospec=True) def test_list(self, mock_list): self.devices.list() expected = mock.call(self.devices, '/devices', 'devices') self.assertTrue(mock_list.called) self.assertEqual(mock_list.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._create', autospec=True) def test_create(self, mock_create): self.devices.create('device1', 'FAKE', '1.0.0', '10.0.0.1', 22, 'fakeuser', 'fakepassword') body = {'name': 'device1', 'type': 'FAKE', 'version': '1.0.0', 'ip': '10.0.0.1', 'port': 22, 'user': '******', 'password': '******'} expected = mock.call(self.devices, '/devices', body, 'device') self.assertTrue(mock_create.called) self.assertEqual(mock_create.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._delete', autospec=True) def test_delete(self, mock_delete): self.devices.delete(self.device) expected = mock.call(self.devices, '/devices/fakeid') self.assertTrue(mock_delete.called) self.assertEqual(mock_delete.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._get', autospec=True) def test_get(self, mock_get): self.devices.get(self.device) expected = mock.call(self.devices, '/devices/fakeid', 'device') self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._get') def test_list_algorithms(self, mock_get): self.devices.list_algoritms() expected = mock.call('/algorithms', 'algorithms', return_raw=True) self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._get') def test_list_protocols(self, mock_get): self.devices.list_protocols() expected = mock.call('/protocols', 'protocols', return_raw=True) self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected])
class TestDeviceManager(unittest2.TestCase): def setUp(self): self.devices = DeviceManager(mock.Mock()) self.device = mock.Mock(id='fakeid') @mock.patch('balancerclient.common.base.Manager._list', autospec=True) def test_list(self, mock_list): self.devices.list() expected = mock.call(self.devices, '/devices', 'devices') self.assertTrue(mock_list.called) self.assertEqual(mock_list.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._create', autospec=True) def test_create(self, mock_create): self.devices.create('device1', 'FAKE', '1.0.0', '10.0.0.1', 22, 'fakeuser', 'fakepassword') body = { 'name': 'device1', 'type': 'FAKE', 'version': '1.0.0', 'ip': '10.0.0.1', 'port': 22, 'user': '******', 'password': '******' } expected = mock.call(self.devices, '/devices', body, 'device') self.assertTrue(mock_create.called) self.assertEqual(mock_create.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._delete', autospec=True) def test_delete(self, mock_delete): self.devices.delete(self.device) expected = mock.call(self.devices, '/devices/fakeid') self.assertTrue(mock_delete.called) self.assertEqual(mock_delete.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._get', autospec=True) def test_get(self, mock_get): self.devices.get(self.device) expected = mock.call(self.devices, '/devices/fakeid', 'device') self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._get') def test_list_algorithms(self, mock_get): self.devices.list_algoritms() expected = mock.call('/algorithms', 'algorithms', return_raw=True) self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._get') def test_list_protocols(self, mock_get): self.devices.list_protocols() expected = mock.call('/protocols', 'protocols', return_raw=True) self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected])