class TestBehaviour(object): def setUp(self): self.mock = StdoutMock() Model.path = os.path.join(os.getcwd(), 'tests', 'data', 'tdtest1') def tearDown(self): self.mock.undo() def test_display_whole_list(self): self.mock.resetArgv() run() result = "{}{}{}{}{}{}\n".format( View.RESET, View.COLORS[3], View.BRIGHT, 1, '.', 'test' ) self.mock.assertEqual(result)
class TestArg(object): def setUp(self): self.mock = StdoutMock() self.model = ModelMock() self.mock.resetArgv() def tearDown(self): self.mock.undo() def test_view(self): Arg(self.model) assert self.model.modify_val is True def test_modify(self): self.mock.addArgs("m", "-s", "sp", "-p", "-d", "dp", "-D", "Dp") Arg(self.model) assert self.model.modifyInPlace_val is True def test_add(self): self.mock.addArgs("a", "1.1", "-n", "n", "-p", "1", "-c", "c") Arg(self.model) assert self.model.add_val is True def test_edit(self): self.mock.addArgs("e", "1.1", "--parent", "2.2", "-n", "n", "-p", "1", "-c", "c") Arg(self.model) assert self.model.edit_val is True def test_rm(self): self.mock.addArgs("r", "1.1") Arg(self.model) assert self.model.rm_val is True def test_done(self): self.mock.addArgs("d", "1.1") Arg(self.model) assert self.model.done_val is True def test_undone(self): self.mock.addArgs("D", "1.1") Arg(self.model) assert self.model.undone_val is True def test_options(self): self.mock.addArgs("o", "-g", "-s", "sp", "-p", "-d", "dp", "-D", "Dp") Arg(self.model) assert self.model.options_val is True def test_options_sort(self): self.mock.addArgs("o", "-s") Arg(self.model) assert self.model.options_val is True def test_options_purge(self): self.mock.addArgs("o", "-p") Arg(self.model) assert self.model.options_val is True def test_options_done(self): self.mock.addArgs("o", "-d") Arg(self.model) assert self.model.options_val is True def test_options_undone(self): self.mock.addArgs("o", "-D") Arg(self.model) assert self.model.options_val is True def test_getKwargs(self): arg = Arg.__new__(Arg) result = arg.getKwargs({"priority": 3}, {"comment": "test"}, GetMock()) assert result == {"name": "mock", "comment": "test", "priority": 3, "parent": "mock"}