diff --git a/tests/unit/test_clidriver.py b/tests/unit/test_clidriver.py index 193d16057462..3908d5f45e54 100644 --- a/tests/unit/test_clidriver.py +++ b/tests/unit/test_clidriver.py @@ -748,9 +748,10 @@ def test_name(self): self.assertEqual(self.cmd.name, 'bar') def test_lineage(self): + cmd = CLICommand() self.assertEqual(self.cmd.lineage, [self.cmd]) - self.cmd.lineage = ['foo'] - self.assertEqual(self.cmd.lineage, ['foo']) + self.cmd.lineage = [cmd] + self.assertEqual(self.cmd.lineage, [cmd]) def test_lineage_names(self): self.assertEqual(self.cmd.lineage_names, ['foo']) @@ -777,9 +778,10 @@ def test_name(self): self.assertEqual(self.cmd.name, 'bar') def test_lineage(self): + cmd = CLICommand() self.assertEqual(self.cmd.lineage, [self.cmd]) - self.cmd.lineage = ['foo'] - self.assertEqual(self.cmd.lineage, ['foo']) + self.cmd.lineage = [cmd] + self.assertEqual(self.cmd.lineage, [cmd]) def test_lineage_names(self): self.assertEqual(self.cmd.lineage_names, ['foo'])