add test_del_register_error for unit test

Change-Id: I3c2ffda9bc861fab98fe0a8fa5b26a5fd8cef0ff
This commit is contained in:
gecong1973 2021-02-03 23:18:06 -08:00
parent e5f36293e5
commit f4bc498f04
1 changed files with 8 additions and 0 deletions

View File

@ -35,6 +35,14 @@ class TestUtils(unittest.TestCase):
ret = utils.do_register(self.client, args=None)
self.assertEqual(0, ret)
def test_del_register_error(self):
self.client.clients.delete = mock.Mock(side_effect=Exception(
'delete client error: bad request'))
with self.assertRaises(Exception) as cm: # noqa
utils.del_register(self.client)
the_exception = cm.exception
self.assertIn('delete client error', str(the_exception))
def test_find_config_files(self):
temp = tempfile.NamedTemporaryFile('wb', delete=True,
suffix='.conf')