예제 #1
0
파일: test_main.py 프로젝트: roryokane/td
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)
예제 #2
0
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"}