[extra-files] Add tests
Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
This commit is contained in:
Lubomír Sedlář
parent
b4fc97be03
commit
d3b2fbe387
@ -82,7 +82,7 @@ class DummyCompose(object):
|
|||||||
result = set()
|
result = set()
|
||||||
for variant in self.variants.itervalues():
|
for variant in self.variants.itervalues():
|
||||||
result |= set(variant.arches)
|
result |= set(variant.arches)
|
||||||
return result
|
return sorted(result)
|
||||||
|
|
||||||
|
|
||||||
def touch(path, content=None):
|
def touch(path, content=None):
|
||||||
|
|||||||
165
tests/test_extra_files_phase.py
Executable file
165
tests/test_extra_files_phase.py
Executable file
@ -0,0 +1,165 @@
|
|||||||
|
#!/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()
|
||||||
Loading…
Reference in New Issue
Block a user