Produits Entr'ouvert » Hobo

import pytest

import mock

from django.conf import settings

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 = '/var/lib/bobo/skeletons/import_me.json'

    # simulate

    # $ XXX-manage import-template import_me

    command = Command()

    with override_settings(PROJECT_NAME='bobo'):

        command.handle(template_name='import_me')

    # should raise error about the path where template was expected

    # should return the expected path in order to check it

    # in fact this path is wrong: /var/lib/%s/skeletons/import_me.json

    assert '/var/lib/%s/skeletons/import_me.json' == expected