def test_service_add_company(self): # create service s1 = Service(parent_id=1, title='plumbing') s2 = Service(parent_id=2, title='electrical') db.session.add(s1) db.session.add(s2) db.session.commit() # create company c1 = Company(company_name='Paint Team', company_address='123 Main St.', company_website='Paintteam.com', company_phone_number='2086783456', company_email='*****@*****.**') c2 = Company(company_name='Bob Team', company_address='123 Main St.', company_website='Paintteam.com', company_phone_number='2086783456', company_email='*****@*****.**') db.session.add(c1) db.session.add(c2) db.session.commit() # add company to service check1 = s1.add_company(c1) db.session.commit() check2 = s2.add_company(c1) db.session.commit() # check that the connection was made self.assertEqual(c1.services.count(), 2) self.assertEqual(s1.companies.first().company_name, 'Paint Team') self.assertEqual(c1.services.first().id, 1) print(c1.services.first().id) # remove companies check3 = s2.remove_company(c1) # check that it was removed self.assertEqual(c1.services.count(), 1) # add services to company check4 = c2.services.append(s1) # check that the connection was made self.assertEqual(c2.services.count(), 1) self.assertEqual(c2.services.first().title, 'plumbing')