def test_add_company(self):
"""
Create environment to test for every possible case--
- Existing relationship but the name is different pk=10
- No existing relationship, but the company exists in the database (as
established by the BusinessUnit title matching a company name) pk=11
- No relationship and the company is not in the database pk=12
Start with 2 Company objects and 3 BusinessUnit objects
End up with 3 Company objects and 3 BusinessUnit objects
"""
for i in range(10, 4):
add_company(BusinessUnit.get(id=i))
# The names of the BU and the Co should be the same
self.assertEquals(BusinessUnit.get(id=i).title,
Company.get(id=i).name,
msg="Company names do not match")
# ensure the relationship was formed
self.assertIn(Company.objects.get(id=i),
BusinessUnit.objects.get(id=i).company_set.all(),
msg="Company is not related to job feed")
class DJCeleryAdminTestCase(DirectSEOBase):
fixtures= ["countries.json"]
def setUp(self):
super(type(self), self).setUp()
# User credentials
self.password = '******'
self.user = User.objects.create_superuser(password=self.password,
email='*****@*****.**')
self.user.save()
self.client.login(email=self.user.email,
password=self.password)
# Task to resend
self.id = 24974
self.date = datetime(year=2015, month=1, day=1, hour=1, minute=1, second=1, tzinfo=pytz.utc)
self.bu = BusinessUnit(id=self.id, date_updated=self.date, date_crawled=self.date)
self.bu.save()
self.etl_to_solr = TaskState(state="test", task_id="testing",
name="priority_etl_to_solr", tstamp=datetime.now(),
args="""(u'8f3dcec2-264d-49fb-98df-8c6f52fb3936', u'24974', u"Gold's Gym")""",
kwargs="{}")
self.etl_to_solr.save()
@override_settings(CELERY_ALWAYS_EAGER=True,
BROKER_BACKEND='memory')
def test_requeue_etl_task(self):
url = "http://www.my.jobs/admin/djcelery/taskstate/"
data = {'action': 'resend_task',
'_selected_action': [unicode(self.etl_to_solr.pk)]}
self.assertEqual(self.bu.date_updated, self.date)
self.client.post(url, data)
bu = BusinessUnit.objects.get(pk=self.id)
self.assertNotEqual(bu.date_updated, self.date)
class DJCeleryAdminTestCase(DirectSEOBase):
fixtures= ["countries.json"]
def setUp(self):
super(type(self), self).setUp()
# User credentials
self.password = '******'
self.user = User.objects.create_superuser(password=self.password,
email='*****@*****.**')
self.user.save()
self.client.login(email=self.user.email,
password=self.password)
# Task to resend
self.id = 25506
self.date = datetime(year=2015, month=1, day=1, hour=1, minute=1, second=1, tzinfo=pytz.utc)
self.bu = BusinessUnit(id=self.id, date_updated=self.date, date_crawled=self.date)
self.bu.save()
self.etl_to_solr = TaskState(state="test", task_id="testing",
name="priority_etl_to_solr", tstamp=datetime.now(),
args="""(u'442e8eab-a4d8-4c49-9ee1-c408c509c1ca', u'25506', u'Penske')""",
kwargs="{}")
self.etl_to_solr.save()
@override_settings(CELERY_ALWAYS_EAGER=True,
BROKER_BACKEND='memory')
def test_requeue_etl_task(self):
url = "http://www.my.jobs/admin/djcelery/taskstate/"
data = {'action': 'resend_task',
'_selected_action': [unicode(self.etl_to_solr.pk)]}
self.assertEqual(self.bu.date_updated, self.date)
self.client.post(url, data)
bu = BusinessUnit.objects.get(pk=self.id)
self.assertNotEqual(bu.date_updated, self.date)
def task_priority_etl_to_solr(guid, buid, name):
try:
import_jobs.update_job_source(guid, buid, name)
BusinessUnit.clear_cache(int(buid))
ImportRecord(buid=int(buid), success=True).save()
except Exception as e:
logging.error("Error loading jobs for jobsource: %s", guid)
logging.exception(e)
ImportRecord(buid=int(buid), success=False).save()
raise task_priority_etl_to_solr.retry(exc=e)
def task_priority_etl_to_solr(guid, buid, name):
try:
import_jobs.update_job_source(guid, buid, name)
BusinessUnit.clear_cache(int(buid))
ImportRecord(buid=int(buid), success=True).save()
except Exception as e:
logging.error("Error loading jobs for jobsource: %s", guid)
logging.exception(e)
ImportRecord(buid=int(buid), success=False).save()
raise task_priority_etl_to_solr.retry(exc=e)
def setUp(self):
super(MultiHostMiddlewareTestCase, self).setUp()
self.bu = BusinessUnit(id=42,
date_crawled=datetime.datetime.now(),
date_updated=datetime.datetime.now())
self.test_site = SeoSiteFactory(domain=u'buckconsultants.jobs',
name=u'buckconsultants')
def setUp(self):
super(type(self), self).setUp()
# User credentials
self.password = '******'
self.user = User.objects.create_superuser(password=self.password,
email='*****@*****.**')
self.user.save()
self.client.login(email=self.user.email,
password=self.password)
# Task to resend
self.id = 25506
self.date = datetime(year=2015, month=1, day=1, hour=1, minute=1, second=1, tzinfo=pytz.utc)
self.bu = BusinessUnit(id=self.id, date_updated=self.date, date_crawled=self.date)
self.bu.save()
self.etl_to_solr = TaskState(state="test", task_id="testing",
name="priority_etl_to_solr", tstamp=datetime.now(),
args="""(u'442e8eab-a4d8-4c49-9ee1-c408c509c1ca', u'25506', u'Penske')""",
kwargs="{}")
self.etl_to_solr.save()
def setUp(self):
super(type(self), self).setUp()
# User credentials
self.password = '******'
self.user = User.objects.create_superuser(password=self.password,
email='*****@*****.**')
self.user.save()
self.client.login(email=self.user.email,
password=self.password)
# Task to resend
self.id = 24974
self.date = datetime(year=2015, month=1, day=1, hour=1, minute=1, second=1, tzinfo=pytz.utc)
self.bu = BusinessUnit(id=self.id, date_updated=self.date, date_crawled=self.date)
self.bu.save()
self.etl_to_solr = TaskState(state="test", task_id="testing",
name="priority_etl_to_solr", tstamp=datetime.now(),
args="""(u'8f3dcec2-264d-49fb-98df-8c6f52fb3936', u'24974', u"Gold's Gym")""",
kwargs="{}")
self.etl_to_solr.save()
def task_clear_bu_cache(buid, **kwargs):
try:
BusinessUnit.clear_cache(buid)
except:
logging.error(traceback.format_exc(sys.exc_info()))
def task_clear_bu_cache(buid, **kwargs):
try:
BusinessUnit.clear_cache(buid)
except:
logging.error(traceback.format_exc(sys.exc_info()))
