@@ -1254,7 +1254,6 @@ def test_create_break(self, mock_set_meetings):
12541254 and returns a 201 response.
12551255 """
12561256 data = {
1257- "id" : 2 ,
12581257 "name" : "Afternoon Break" ,
12591258 "location_string" : "Lobby" ,
12601259 "meetings" : [], # No meetings provided
@@ -1263,7 +1262,7 @@ def test_create_break(self, mock_set_meetings):
12631262 self .assertEqual (response .status_code , status .HTTP_201_CREATED )
12641263
12651264 # Check that a new Break was created with the proper fields.
1266- new_break_id = response .data .get ("id " )
1265+ new_break_id = response .data .get ("break_id " )
12671266 self .assertIsNotNone (new_break_id )
12681267 new_break = Break .objects .get (id = new_break_id )
12691268 self .assertEqual (new_break .name , "Afternoon Break" )
@@ -1302,7 +1301,7 @@ def test_update_break_existing(self, mock_set_meetings):
13021301 self .assertEqual (self .break_obj .location_string , "New Location" )
13031302
13041303 self .assertEqual (response .data .get ("message" ), "success" )
1305- self .assertEqual (response .data .get ("id " ), self .break_obj .id )
1304+ self .assertEqual (response .data .get ("break_id " ), self .break_obj .id )
13061305
13071306 def test_update_break_nonexistent (self ):
13081307 """
@@ -1318,7 +1317,7 @@ def test_update_break_nonexistent(self):
13181317 self .assertEqual (response .status_code , status .HTTP_201_CREATED )
13191318 self .assertEqual (response .data .get ("message" ), "success" )
13201319
1321- new_break = Break .objects .get (id = response .data .get ("id " ))
1320+ new_break = Break .objects .get (id = response .data .get ("break_id " ))
13221321
13231322 self .assertEqual (new_break .name , "Nonexistent Break" )
13241323 self .assertEqual (new_break .location_string , "Nowhere" )
0 commit comments