Update linege unit tests

aws · Feb 18, 2015 · 9e40bfb · 9e40bfb
1 parent 2a02659
commit 9e40bfb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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'])