Produits Entr'ouvert » Hobo

        basepath = kwargs.get('basepath') or '/var/lib/%s/skeletons' % app_name

            raise CommandError('Unknown template (%r)' % template)

import pytest

import mock

from django.conf import settings

from django.core.management.base import CommandError

from django.test import override_settings

from hobo.agent.common.management.commands.import_template import Command

@pytest.fixture

def template():

    # build a template file to import

    basepath = '/tmp'

    template_name = 'import_me'

    with open('/%s/%s.json' % (basepath, template_name),'w') as handler:

        handler.write('some content to import')

    return template_name

@mock.patch('hobo.agent.common.management.commands.import_template.call_command')

@mock.patch('hobo.agent.common.management.commands.import_template.get_commands')

def test_import_template_ok(mocked_get_commands, mocked_call_command, template):

    """do import a template"""

    expected = [mock.call('import_site', '/tmp/import_me.json')]

    mocked_get_commands.return_value = 'import_site'

    mocked_call_command.return_value = mock.MagicMock()

    # simulate

    # $ XXX-manage import-template import_me --basepath /tmp

    command = Command()

    with override_settings(PROJECT_NAME='bobo'):

        command.handle(template_name=template, basepath='/tmp')

    # assert 'import_site' command is run

    assert mocked_call_command.mock_calls == expected

def test_import_template_default_path():

    """check default path where template should be found"""

    expected = "Unknown template ('/var/lib/bobo/skeletons/import_me.json')"

    # simulate

    # $ XXX-manage import-template import_me

    command = Command()

    with override_settings(PROJECT_NAME='bobo'):

        with pytest.raises(CommandError) as e_info:

            command.handle(template_name='import_me')

    assert str(e_info.value) == expected