def setUp(self):
     self.devices = DeviceManager(mock.Mock())
     self.device = mock.Mock(id='fakeid')
Esempio n. 2
0
 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])
Esempio n. 4
0
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])