def test_clean(self, clean_mock):
text = '<strong><a href="#">foo</a></strong>'
tags = ["em", "strong"]
fact = FunFactFactory.create(public_text=text)
fact.clean()
clean_mock.assert_called_once_with(text, tags=tags, strip=True)
eq_(fact.public_text, "<strong>foo</strong>")
def test_clean(self, clean_mock):
text = '<strong><a href="#">foo</a></strong>'
tags = ['em', 'strong']
fact = FunFactFactory.create(public_text=text)
fact.clean()
clean_mock.assert_called_once_with(text, tags=tags, strip=True)
eq_(fact.public_text, '<strong>foo</strong>')
def test_cron_validator(self, validator_mock):
valid_fact_1 = FunFactFactory.create(published=True)
valid_fact_2 = FunFactFactory.create(
published=True,
divisor='UserProfile.objects.aggregate(number=Count("id"))')
invalid_fact_1 = FunFactFactory.create(published=True, number='invalid')
invalid_fact_2 = FunFactFactory.create(published=True, divisor='invalid')
validate_fun_facts()
call_list = [call(valid_fact_1.number),
call(valid_fact_2.number),
call(valid_fact_2.divisor),
call(invalid_fact_1.number),
call(invalid_fact_2.number),
call(invalid_fact_2.divisor)]
eq_(validator_mock.call_args_list, call_list)
invalid_fact_1 = FunFact.objects.get(id=invalid_fact_1.id)
invalid_fact_2 = FunFact.objects.get(id=invalid_fact_2.id)
eq_(invalid_fact_1.published, False)
eq_(invalid_fact_2.published, False)
def test_cron_validator(self, validator_mock):
valid_fact_1 = FunFactFactory.create(published=True)
valid_fact_2 = FunFactFactory.create(
published=True,
divisor='UserProfile.objects.aggregate(number=Count("id"))')
invalid_fact_1 = FunFactFactory.create(published=True, number='invalid')
invalid_fact_2 = FunFactFactory.create(published=True, divisor='invalid')
validate_fun_facts()
call_list = [call(valid_fact_1.number),
call(valid_fact_2.number),
call(valid_fact_2.divisor),
call(invalid_fact_1.number),
call(invalid_fact_2.number),
call(invalid_fact_2.divisor)]
eq_(validator_mock.call_args_list, call_list)
invalid_fact_1 = FunFact.objects.get(id=invalid_fact_1.id)
invalid_fact_2 = FunFact.objects.get(id=invalid_fact_2.id)
eq_(invalid_fact_1.published, False)
eq_(invalid_fact_2.published, False)
def test_execute_invalid_funfact(self, transaction_mock):
fact = FunFactFactory.create(number='number')
return_value = fact.execute()
ok_(return_value.startswith('Error'))
transaction_mock.atomic.assert_called_once_with()
def test_execute_valid_funfact(self, transaction_mock):
fact = FunFactFactory.create()
fact.execute()
transaction_mock.atomic.assert_called_once_with()
def setUp(self):
self.published_1 = FunFactFactory.create(published=True)
self.published_2 = FunFactFactory.create(published=True)
self.unpublished_1 = FunFactFactory.create(published=False)
def test_helper_returns_none(self):
"""Test helper returns None when no published FunFacts."""
FunFactFactory.create()
eq_(random_funfact(), None)
def test_execute_invalid_funfact(self, transaction_mock):
fact = FunFactFactory.create(number="number")
return_value = fact.execute()
ok_(return_value.startswith("Error"))
transaction_mock.commit_manually.assert_called_once_with()
transaction_mock.rollback.assert_called_once_with()
def test_execute_valid_funfact(self, transaction_mock):
fact = FunFactFactory.create()
fact.execute()
transaction_mock.commit_manually.assert_called_once_with()
transaction_mock.rollback.assert_called_once_with()
def setUp(self):
self.published_1 = FunFactFactory.create(published=True)
self.published_2 = FunFactFactory.create(published=True)
self.unpublished_1 = FunFactFactory.create(published=False)
def test_execute_invalid_funfact(self, transaction_mock):
fact = FunFactFactory.create(number='number')
return_value = fact.execute()
ok_(return_value.startswith('Error'))
transaction_mock.atomic.assert_called_once_with()
def test_execute_valid_funfact(self, transaction_mock):
fact = FunFactFactory.create()
fact.execute()
transaction_mock.atomic.assert_called_once_with()
def test_execute_valid_funfact(self, transaction_mock):
fact = FunFactFactory.create()
fact.execute()
transaction_mock.commit_manually.assert_called_once_with()
transaction_mock.rollback.assert_called_once_with()
