def test_office_add(self):
rep = Representative.add_rep(single_rep1)
db.session.commit()
o = Office.add_office(single_office)
rep.offices.append(o)
db.session.commit()
testoffice = Office.query.get(1)
# import pdb
# pdb.set_trace()
self.assertIsNotNone(testoffice)
self.assertEqual(testoffice.phone, '716-664-4603')
self.assertEqual(
testoffice.address,
'2-6 E. Second Street; Fenton Building, Suite 302; Jamestown, NY 14701'
)
self.assertEqual(testoffice.location, 'District Office')
def test_office_add_error(self):
with self.assertRaises(TypeError):
Office.add_office(error_office)
