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_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()