fills out wave 6 to have the update tasks at post /goals/:id/tasks route

Author: tildeee

ada-project-docs/wave_06.md

@@ -44,6 +44,32 @@ After learning the strategy for creating a one-to-many relationship, in the Task
 - Setting the foreign key to `goal`'s primary key column
 - Setting the `nullable` to `True`
 
+Remember to run `flask db migrate` and `flask db upgrade` whenever there is a change to the model.
+
+### Sending a List of Task IDs to a Goal
+
+Given:
+
+- a goal that has the ID `1`
+- three tasks with the IDs `1`, `2`, and `3`
+
+When I send a `POST` request to `/goals/1/tasks` with this request body:
+
+```json
+{
+  "task_ids": [1, 2, 3]
+}
+```
+
+Then the three `Task`s belong to the `Goal` and it gets updated in the database, and we get back a `200 OK` with the following response body:
+
+```json
+{
+  "id": 1,
+  "task_ids": [1, 2, 3]
+}
+```
+
 ### Getting Tasks of One Goal
 
 Given a goal that has:

tests/test_wave_06.py

@@ -1,3 +1,44 @@
+from app.models.goal import Goal
+
+
+def test_post_task_ids_to_goal(client, one_goal, three_tasks):
+    # Act
+    response = client.post("/goals/1/tasks", json={
+        "task_ids": [1, 2, 3]
+    })
+    response_body = response.get_json()
+
+    # Assert
+    assert response.status_code == 200
+    assert "id" in response_body
+    assert "task_ids" in response_body
+    assert response_body == {
+        "id": 1,
+        "task_ids": [1, 2, 3]
+    }
+
+    # Check that Goal was updated in the db
+    assert len(Goal.query.get(1).tasks) == 3
+
+
+def test_post_task_ids_to_goal_already_with_goals(client, one_task_belongs_to_one_goal, three_tasks):
+    # Act
+    response = client.post("/goals/1/tasks", json={
+        "task_ids": [1, 4]
+    })
+    response_body = response.get_json()
+
+    # Assert
+    assert response.status_code == 200
+    assert "id" in response_body
+    assert "task_ids" in response_body
+    assert response_body == {
+        "id": 1,
+        "task_ids": [1, 4]
+    }
+    assert len(Goal.query.get(1).tasks) == 2
+
+
 def test_get_tasks_for_specific_goal_no_goal(client):
     # Act
     response = client.get("/goals/1/tasks")