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)