def test_get_slug(self): company = ProductionCompanyFactory(name='Company Name') production = ProductionFactory( start_date=datetime(2017, 1, 3), play__title='Test Play Title', production_company=company, ) self.assertEqual(production.get_slug(), u'20170103-test-play-title-by-company-name')
def test_save(self): production = ProductionFactory(pk=None, slug=None) with patch('django.db.models.Model.save') as mock_save: production.save() self.assertEqual(production.slug, production.get_slug()) mock_save.assert_called_once_with()