def test_transaction_assign_recipient_volunteer(self): t = Transaction() r = Recipient(username='******') v = Volunteer(username='******') db.session.add_all([t, r, v]) db.session.commit() t.assign_recipient(r) self.assertEqual(t.recipient, r) self.assertEqual(t.recipient_id, r.id) self.assertFalse(t.claimed) self.assertEqual(r.transactions[0], t) t.assign_volunteer(v) db.session.add(t) db.session.commit() self.assertEqual(t.volunteer, v) self.assertEqual(t.volunteer_id, v.id) self.assertTrue(t.claimed) self.assertEqual(v.transactions[0], t)