class AddDirectoryCommandTestCase(unittest.TestCase): cmd = None visitor = None def setUp(self): self.cmd = AddDirectoryCommand('/mydirectory/dir') self.visitor = UtilityVisitor() def testGetValue(self): self.assertEquals('-environment-directory /mydirectory/dir\n', self.cmd.getValue()) def testAccept(self): self.cmd.accept(self.visitor) self.assertEquals(True, self.visitor.isAddDirectoryCommandVisited())