Python assert_called_with Examples

Example #1

 def test_validate_models(self, patching):
     f = self.Fixup(self.app)
     f.django_setup = Mock(name='django.setup')
     patching.modules('django.core.checks')
     from django.core.checks import run_checks
     f.validate_models()
     f.django_setup.assert_called_with()
     run_checks.assert_called_with()

Example #2

File: test_django.py Project: celery/celery

 def test_validate_models(self, patching):
     with mock.module('django', 'django.db', 'django.core',
                      'django.core.cache', 'django.conf',
                      'django.db.utils'):
         f = self.Fixup(self.app)
         f.django_setup = Mock(name='django.setup')
         patching.modules('django.core.checks')
         from django.core.checks import run_checks
         f.validate_models()
         f.django_setup.assert_called_with()
         run_checks.assert_called_with()