def test_when_team_has_no_responses_expect_empty_set(self):
# ARRANGE
section = Section()
team = section.get_team('1', 'team 1')
# ACT
result = section.responses_for_team(team)
# ASSERT
self.assertEqual(result, set())
def test_when_team_has_one_response_expect_one_response(self):
# ARRANGE
section = Section()
team = section.get_team('1', 'team 1')
response = Response(team=team)
section.add_response(response)
# ACT
result = section.responses_for_team(team)
# ASSERT
self.assertEqual(result, {response})
def test_when_team_has_multiple_responses_expect_all_responses(self):
# ARRANGE
section = Section()
team = section.get_team('1', 'team 1')
response1 = Response(team=team)
section.add_response(response1)
response2 = Response(team=team)
section.add_response(response2)
# ACT
result = section.responses_for_team(team)
# ASSERT
self.assertEqual(result, {response1, response2})
