def handle_noargs(self, **options): hizard = Company.objects.create(name="Hizard", subdomain="hizard", website="http://hizard.com", description="") setup_company(hizard) hizard.opening_set.all().update(published_date=datetime.datetime.now())
def form_valid(self, form): company = form.save() self.request.user.company = company self.request.user.save() utils.setup_company(company) return super(CompanyCreateView, self).form_valid(form)
def test_should_add_default_data(self): company = CompanyFactory() utils.setup_company(company) # Not asserting exact number since it can vary if we want to add some self.assertTrue( InterviewStage.objects.filter(company=company).exists() ) self.assertTrue( InterviewStage.objects.filter(company=company, tag="RECEIVED").exists() ) self.assertTrue( InterviewStage.objects.filter(company=company, tag="HIRED").exists() ) self.assertTrue( InterviewStage.objects.filter(company=company, tag="REJECTED").exists() ) self.assertTrue(Opening.objects.filter(company=company).exists()) self.assertEqual(EmailTemplate.objects.all().count(), 3)