def test_new_task(): """Test create new task.""" gid.create_db(HOME_DIR, DB_PATH) connection = gid.connect_db(DB_PATH) gid.new_task(connection, "Buy milk.") response = gid.read_db(connection, "SELECT name from tasks") for task in response: task_name = task[0] assert len(str(task_name)) == len("Buy milk.") connection.close() os.remove(DB_PATH)
def test_connect_db(): """Test connection to db.""" gid.create_db(HOME_DIR, DB_PATH) try: connection = gid.connect_db(DB_PATH) conn = "success" except gid.sqlite3.Error: conn = "failed" assert conn == "success" connection.close() os.remove(DB_PATH)
def test_list_tasks(): """List all tasks.""" gid.create_db(HOME_DIR, DB_PATH) connection = gid.connect_db(DB_PATH) gid.new_task(connection, "Buy milk.") gid.new_task(connection, "Buy bread.") response = gid.read_db(connection, "SELECT name from tasks") taskList = [] for task in response: taskList.append(task[0]) assert len(taskList) == 2 connection.close() os.remove(DB_PATH)
def test_delete_task(monkeypatch): """Test delete task.""" gid.create_db(HOME_DIR, DB_PATH) connection = gid.connect_db(DB_PATH) taskName = "Buy milk." gid.new_task(connection, taskName) monkeypatch.setattr('builtins.input', lambda _: "y") gid.delete_task(connection, taskName) response = gid.read_db(connection, "SELECT name from tasks") for task in response: taskName = task[0] if taskName is None: assert True connection.close() os.remove(DB_PATH)
def test_update_task(monkeypatch): """Test update of task.""" gid.create_db(HOME_DIR, DB_PATH) connection = gid.connect_db(DB_PATH) oldTaskName = "Buy milk." gid.new_task(connection, oldTaskName) newTaskName = "Buy full cream milk." monkeypatch.setattr('builtins.input', lambda _: "y") gid.update_task(connection, oldTaskName, newTaskName) response = gid.read_db(connection, "SELECT name from tasks") taskName = "" for task in response: taskName = task[0] assert len(str(taskName)) == len(newTaskName) connection.close() os.remove(DB_PATH)
def test_create_db(): """Test database creation.""" gid.create_db(HOME_DIR, DB_PATH) assert os.path.isfile(DB_PATH) os.remove(DB_PATH)