def test_all_for_company_by_company_id(self): # Iterate over all tags for user red_tag = Tag.tag_companies("red", [self.company.id]) tags = Tag.find_all_for_company(company_id=self.company.id) for tag in tags: self.assertEqual(red_tag.id, tag.id) Tag.untag_companies("red", [self.company.id])
def test_company_counts_for_each_tag(self): # Get Company Tag Count Object Tag.tag_companies('blue', [self.company.id]) counts = Count.company_counts_for_each_tag Tag.untag_companies('blue', [self.company.id])
def test_untag_companies(self): # Untag companies tag = Tag.untag_companies("red", [self.user.companies[0].id]) self.assertEqual(tag.name, "red") company = Company.find(id=self.user.companies[0].id) self.assertEqual(0, len(company.tags))