Ejemplo n.º 1
0
    def test_Todo_add_task(self):
        todo = Todo()

        sys.stdin = io.StringIO('测试aaa')
        todo.add_task()

        sql = """select name from to_do_list where now = True"""
        data = todo.show_sql_data(sql)

        self.assertIn('测试aaa', data['name'].values)

        sql = """delete from to_do_list where name = '测试aaa' """
        todo.run_sql(sql)