class AdvanceCommandTestCase(unittest.TestCase): cmd = None visitor = None def setUp(self): self.cmd = AdvanceCommand('function') self.visitor = UtilityVisitor() def testGetValue(self): self.assertEquals('advance function\n', self.cmd.getValue()) def testAccept(self): self.cmd.accept(self.visitor) self.assertEquals(True, self.visitor.isAdvanceCommandVisited())