def test_create_rserver_3(self, mock_func):
"""Exception"""
self.rs['parent_id'] = None
cmd.create_rserver(self.ctx, self.rs)
rollback_fn = self.ctx.add_rollback.call_args[0][0]
rollback_fn(False)
self.assertTrue(self.ctx.device.delete_real_server.called)
self.assertTrue(mock_func.called)
def test_create_rserver_1(self, mock_func):
""" parent_id is None """
self.rs['parent_id'] = None
cmd.create_rserver(self.ctx, self.rs)
self.assertTrue(self.ctx.device.create_real_server.called)
self.assertTrue(mock_func.called)
mock_func.assert_called_once_with(self.ctx.conf,
self.rs['id'], self.rs)
def test_create_rserver_2(self, mock_func):
""" parent_id is not None or 0, no exception """
self.rs['parent_id'] = 1
cmd.create_rserver(self.ctx, self.rs)
self.assertFalse(self.ctx.device.create_real_server.called)
self.assertFalse(mock_func.called)
