class TestMultiplayer(BaseTestCase): inviter_number = os.environ.get('TEST_FROM_NUMBER') secondary_number = os.environ.get('TEST_TO_NUMBER') def setUp(self): super().setUp() self.mitch = AppUser(phone_number=self.inviter_number, username='******', timezone=US_TIMEZONES['b']) self.db.session.add(self.mitch) self.billy = AppUser(phone_number=self.secondary_number, username='******', timezone=US_TIMEZONES['b']) self.db.session.add(self.billy) self.team = Team(founder=self.mitch, name='The Cherrys') self.db.session.add(self.team) self.mitch_member = TeamMember(user=self.mitch, team=self.team, inviter=self.mitch, status=Statuses.ACTIVE) self.db.session.add(self.mitch_member) self.billy_member = TeamMember(user=self.billy, team=self.team, inviter=self.mitch, status=Statuses.PENDING) self.db.session.add(self.billy_member) self.db.session.commit() self.mitch = self.mitch.to_dict() self.billy = self.billy.to_dict() self.team = self.team.to_dict() def test_leave_team(self): valid_inbound = self.team['id'] invalid_inbound = '999' # test correct team id, and incorrect team id multiplayer.leave_team(self.mitch, valid_inbound) self.assertRaises(NoResultFound, multiplayer.leave_team, self.mitch, invalid_inbound) def test_join_team_existing_user(self): '''Test the ability to join a team as an existing user''' member = multiplayer.respond_to_invite(self.billy, 'yes') assert member.status == Statuses.ACTIVE
def teams_post(id): form = TeamForm() user = User.query.get(id) form['csrf_token'].data = request.cookies['csrf_token'] if form.validate_on_submit(): team = Team( teamName=form.data['teamName'], ) team.users.append(user) db.session.add(team) db.session.commit() return team.to_dict() return "Oops, something went wrong!"