pungi/tests/test_extra_files_phase.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import mock
import os
import sys
import unittest

sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))

from pungi.phases import extra_files
from tests import helpers


class TestExtraFilePhase(helpers.PungiTestCase):

    @mock.patch('pungi.phases.extra_files.copy_extra_files')
    def test_runs_copy_files_for_each_variant(self, copy_extra_files):
        cfg = mock.Mock()
        pkgset_phase = mock.Mock()
        compose = helpers.DummyCompose(self.topdir, {
            'extra_files': [
                ('^.+$', {'*': cfg})
            ]
        })

        phase = extra_files.ExtraFilesPhase(compose, pkgset_phase)
        phase.run()

        self.assertItemsEqual(
            copy_extra_files.call_args_list,
            [mock.call(compose, 'x86_64', compose.variants['Server'], pkgset_phase.package_sets),
             mock.call(compose, 'amd64', compose.variants['Server'], pkgset_phase.package_sets),
             mock.call(compose, 'amd64', compose.variants['Client'], pkgset_phase.package_sets),
             mock.call(compose, 'x86_64', compose.variants['Everything'], pkgset_phase.package_sets),
             mock.call(compose, 'amd64', compose.variants['Everything'], pkgset_phase.package_sets)]
        )


class TestCopyFiles(helpers.PungiTestCase):

    @mock.patch('pungi.phases.extra_files.run')
    @mock.patch('pungi.phases.extra_files.get_file_from_scm')
    @mock.patch('pungi.phases.extra_files.get_dir_from_scm')
    def test_run_without_config(self, get_dir_from_scm, get_file_from_scm, run):
        compose = helpers.DummyCompose(self.topdir, {})

        extra_files.copy_extra_files(compose, 'x86_64', compose.variants['Server'], mock.Mock())

        self.assertEqual(run.call_args_list, [])
        self.assertEqual(get_file_from_scm.call_args_list, [])
        self.assertEqual(get_dir_from_scm.call_args_list, [])

    def test_copy_local_file(self):
        tgt = os.path.join(self.topdir, 'file')
        helpers.touch(tgt)
        compose = helpers.DummyCompose(self.topdir, {
            'extra_files': [
                ('^.+$', {'*': {
                    'scm': 'file',
                    'file': tgt,
                    'repo': None,
                }})
            ]
        })

        extra_files.copy_extra_files(compose, 'x86_64', compose.variants['Server'], mock.Mock())

        self.assertTrue(os.path.isfile(os.path.join(
            self.topdir, 'compose', 'Server', 'x86_64', 'os', 'file')))

    def test_copy_local_dir(self):
        helpers.touch(os.path.join(self.topdir, 'src', 'file'))
        helpers.touch(os.path.join(self.topdir, 'src', 'another'))
        compose = helpers.DummyCompose(self.topdir, {
            'extra_files': [
                ('^.+$', {'*': {
                    'scm': 'file',
                    'dir': os.path.join(self.topdir, 'src'),
                    'repo': None,
                    'target': 'subdir',
                }})
            ]
        })

        extra_files.copy_extra_files(compose, 'x86_64', compose.variants['Server'], mock.Mock())

        self.assertTrue(os.path.isfile(os.path.join(
            self.topdir, 'compose', 'Server', 'x86_64', 'os', 'subdir', 'file')))
        self.assertTrue(os.path.isfile(os.path.join(
            self.topdir, 'compose', 'Server', 'x86_64', 'os', 'subdir', 'another')))

    @mock.patch('pungi.phases.extra_files.get_file_from_scm')
    @mock.patch('pungi.phases.extra_files.get_dir_from_scm')
    def test_copy_from_external_rpm(self, get_dir_from_scm, get_file_from_scm):
        compose = helpers.DummyCompose(self.topdir, {
            'extra_files': [
                ('^.+$', {'*': {
                    'scm': 'rpm',
                    'file': 'file.txt',
                    'repo': 'http://example.com/package.rpm',
                }})
            ]
        })

        get_file_from_scm.side_effect = self.fake_get_file

        extra_files.copy_extra_files(compose, 'x86_64', compose.variants['Server'], mock.Mock())

        self.assertEqual(len(get_file_from_scm.call_args_list), 1)
        self.assertEqual(get_dir_from_scm.call_args_list, [])
        self.assertEqual(self.scm_dict,
                         {'scm': 'rpm', 'file': 'file.txt', 'repo': 'http://example.com/package.rpm'})

        self.assertTrue(os.path.isfile(os.path.join(
            self.topdir, 'compose', 'Server', 'x86_64', 'os', 'file.txt')))

    @mock.patch('pungi.phases.extra_files.get_file_from_scm')
    @mock.patch('pungi.phases.extra_files.get_dir_from_scm')
    def test_copy_from_rpm_in_compose(self, get_dir_from_scm, get_file_from_scm):
        compose = helpers.DummyCompose(self.topdir, {
            'extra_files': [
                ('^.+$', {'*': {
                    'scm': 'rpm',
                    'file': 'file.txt',
                    'repo': '%(variant_uid)s-data*',
                }})
            ]
        })
        server_po, client_po, src_po = mock.Mock(), mock.Mock(), mock.Mock()
        server_po.configure_mock(name='Server-data-1.1-1.fc24.x86_64.rpm',
                                 file_path='/server/location',
                                 arch='x86_64')
        client_po.configure_mock(name='Client-data-1.1-1.fc24.x86_64.rpm',
                                 file_path='/client/location',
                                 arch='x86_64')
        src_po.configure_mock(name='extra-data-1.1-1.fc24.src.rpm',
                              file_path='/src/location',
                              arch='src')
        package_sets = {
            'x86_64': {server_po.name: server_po,
                       client_po.name: client_po,
                       src_po.name: src_po}
        }

        get_file_from_scm.side_effect = self.fake_get_file

        extra_files.copy_extra_files(compose, 'x86_64', compose.variants['Server'], package_sets)

        self.assertEqual(len(get_file_from_scm.call_args_list), 1)
        self.assertEqual(get_dir_from_scm.call_args_list, [])

        self.assertEqual(self.scm_dict,
                         {'scm': 'rpm', 'file': 'file.txt', 'repo': ['/server/location']})

        self.assertTrue(os.path.isfile(os.path.join(
            self.topdir, 'compose', 'Server', 'x86_64', 'os', 'file.txt')))

    def fake_get_file(self, scm_dict, dest, logger):
        self.scm_dict = scm_dict
        helpers.touch(os.path.join(dest, scm_dict['file']))


if __name__ == "__main__":
    unittest.main()