class StepCommandTestCase(unittest.TestCase): cmd = None visitor = None def setUp(self): self.cmd = StepCommand() self.visitor = UtilityVisitor() def testGetValue(self): self.assertEquals('step\n', self.cmd.getValue()) def testAccept(self): self.cmd.accept(self.visitor) self.assertEquals(True, self.visitor.isStepCommandVisited())