1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
|
from unittest import mock
from django.core.management import call_command
from .base import SitemapTestsBase
@mock.patch('django.contrib.sitemaps.management.commands.ping_google.ping_google')
class PingGoogleTests(SitemapTestsBase):
def test_default(self, ping_google_func):
call_command('ping_google')
ping_google_func.assert_called_with(sitemap_url=None, sitemap_uses_https=True)
def test_args(self, ping_google_func):
call_command('ping_google', 'foo.xml', '--sitemap-uses-http')
ping_google_func.assert_called_with(sitemap_url='foo.xml', sitemap_uses_https=False)
|