Ejemplo n.º 1
0
class TodoTestCase(object):
    def __init__(self):
        self._todo = Todo()

    def set_up(self):
        self._todo.clear()

    def test_add_todo_adds_pending_item(self):
        self._todo.add("Sandwich")

        items = self._todo.items()
        assert_equal(("Sandwich", Todo.PENDING), items)

    def test_add_return_value(self):
        add_return_index = self._todo.add("Sandwich")
        assert_equal(0, add_return_index)
Ejemplo n.º 2
0
class TodoTestCase(object):

    def __init__(self):
        self._todo = Todo()

    def set_up(self):
        self._todo.clear()

    def test_add_todo_adds_pending_item(self):
        self._todo.add("Sandwich")

        items = self._todo.items()
        assert_equal(("Sandwich", Todo.PENDING), items)

    def test_add_return_value(self):
        add_return_index = self._todo.add("Sandwich")
        assert_equal(0, add_return_index)
Ejemplo n.º 3
0
def todo():
    todos = Todo()
    if len(sys.argv) == 1:
        print('Incorrect usage!')
    else:
        cmd = sys.argv[1]
        if cmd == 'add':
            name = sys.argv[2]
            todos.add(name)
        elif cmd == 'remove':
            id = sys.argv[2]
            todos.remove(id)
        elif cmd == 'show':
            todos.show()
        elif cmd == 'done':
            id = sys.argv[2]
            todos.done(id)
        elif cmd == 'undone':
            id = sys.argv[2]
            todos.undone(id)
        elif cmd == 'clear':
            todos.clear()
    todos.save()