def test_clean_pull_cleans_bootstrap(self):
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
plugin.clean_pull()
mock_clean = self.bootstrapper_mock.return_value.clean
mock_clean.assert_called_once_with()
def test_pull_fetches_ros2_underlay(self):
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
plugin.pull()
self.bootstrapper_mock.return_value.pull.assert_called_once_with()
self.bootstrapper_mock.return_value.build.assert_not_called()
def test_build_builds_ros2_underlay(self, mock_check_call):
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
self.bootstrapper_mock.return_value.build.return_value = 'bootstrap'
os.mkdir('bootstrap')
with open(os.path.join('bootstrap', 'test-file'), 'w') as f:
f.write('hello')
plugin.build()
self.bootstrapper_mock.return_value.pull.assert_not_called()
self.bootstrapper_mock.return_value.build.assert_called_once_with()
# Assert that the bootstrap dir was copied into the installdir
self.assertThat(
os.path.join(plugin.installdir, 'test-file'), FileExists())
self.assertThat(
os.path.join(plugin.installdir, 'test-file'),
FileContains('hello'))
class check_env():
def __eq__(self, env):
return (
env['PYTHONPATH'] == os.path.join(
os.path.sep, 'usr', 'lib', 'python3', 'dist-packages'))
# Verify that the source space was built as expected, and that the
# system's PYTHONPATH was included while building
mock_check_call.assert_called_once_with([
mock.ANY, mock.ANY, 'ament', 'build', plugin.sourcedir,
'--build-space', plugin.builddir, '--install-space',
plugin.installdir, '--cmake-args', '-DCMAKE_BUILD_TYPE=Release'],
cwd=mock.ANY, env=check_env())
def test_sources_include_ros_archive(self):
plugin = ament.AmentPlugin("test-part", self.properties, self.project)
self.assertThat(
plugin.PLUGIN_STAGE_SOURCES,
Contains("deb http://packages.ros.org/ros/ubuntu/ xenial main"),
)
def test_environment(self):
plugin = ament.AmentPlugin("test-part", self.properties, self.project)
python_path = os.path.join(plugin.installdir, "usr", "lib",
"python2.7", "dist-packages")
os.makedirs(python_path)
# Joining and re-splitting to get hacked script in there as well
environment = "\n".join(plugin.env(plugin.installdir)).split("\n")
self.assertThat(
environment,
Contains('PYTHONUSERBASE="{}"'.format(plugin.installdir)))
self.assertThat(
environment,
Contains('PYTHONPATH="{}:{}${{PYTHONPATH:+:$PYTHONPATH}}"'.format(
os.path.join(plugin.installdir, "usr", "lib", "python3",
"dist-packages"),
os.path.join(plugin.installdir, "lib", "python3",
"dist-packages"),
)),
)
self.assertThat(
environment,
Contains(" . {}/setup.sh".format(plugin.installdir)))
def test_build_builds_ros2_underlay(self, mock_run):
plugin = ament.AmentPlugin("test-part", self.properties, self.project)
self.bootstrapper_mock.return_value.build.return_value = "bootstrap"
os.mkdir("bootstrap")
with open(os.path.join("bootstrap", "test-file"), "w") as f:
f.write("hello")
plugin.build()
self.bootstrapper_mock.return_value.pull.assert_not_called()
self.bootstrapper_mock.return_value.build.assert_called_once_with()
# Assert that the bootstrap dir was copied into the installdir
self.assertThat(os.path.join(plugin.installdir, "test-file"),
FileExists())
self.assertThat(os.path.join(plugin.installdir, "test-file"),
FileContains("hello"))
class check_env:
def __eq__(self, env):
return env["PYTHONPATH"] == os.path.join(
os.path.sep, "usr", "lib", "python3", "dist-packages")
# Verify that the source space was built as expected, and that the
# system's PYTHONPATH was included while building
mock_run.assert_called_once_with(
[
"ament",
"build",
plugin.sourcedir,
"--build-space",
plugin.builddir,
"--install-space",
plugin.installdir,
"--cmake-args",
"-DCMAKE_BUILD_TYPE=Release",
],
cwd=mock.ANY,
env=check_env(),
)
def test_environment(self):
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
python_path = os.path.join(
plugin.installdir, 'usr', 'lib', 'python2.7', 'dist-packages')
os.makedirs(python_path)
# Joining and re-splitting to get hacked script in there as well
environment = '\n'.join(plugin.env(plugin.installdir)).split('\n')
self.assertThat(environment, Contains(
'PYTHONUSERBASE="{}"'.format(plugin.installdir)))
self.assertThat(environment, Contains(
'PYTHONPATH="{}:{}${{PYTHONPATH:+:$PYTHONPATH}}"'.format(
os.path.join(
plugin.installdir, 'usr', 'lib', 'python3',
'dist-packages'),
os.path.join(
plugin.installdir, 'lib', 'python3', 'dist-packages'))))
self.assertThat(environment, Contains(
' . {}/setup.sh'.format(plugin.installdir)))
def test_prepare_build_rewrites_ament_current_prefix(self):
"""Make sure AMENT_CURRENT_PREFIX is rewritten to be the installdir
Otherwise ament embeds absolute paths in there, which results in a
non-relocatable snap.
"""
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
self.bootstrapper_mock.return_value.build.return_value = 'bootstrap'
os.mkdir('bootstrap')
with open(os.path.join('bootstrap', 'test-file'), 'w') as f:
f.write('${AMENT_CURRENT_PREFIX:=bootstrap}')
plugin._prepare_build()
self.assertThat(
os.path.join(plugin.installdir, 'test-file'), FileExists())
self.assertThat(
os.path.join(plugin.installdir, 'test-file'),
FileContains('${{AMENT_CURRENT_PREFIX:={}}}'.format(
plugin.installdir)))
def test_finish_build_rewrites_ament_current_prefix(self):
"""Make sure AMENT_CURRENT_PREFIX is rewritten to $SNAP
Otherwise ament keeps an absolute path in there, requiring us to define
it at runtime to a valid location. Also, without this, multiple parts
would clash.
"""
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
self.bootstrapper_mock.return_value.build.return_value = 'bootstrap'
os.mkdir('bootstrap')
with open(os.path.join('bootstrap', 'test-file'), 'w') as f:
f.write('${AMENT_CURRENT_PREFIX:=bootstrap}')
plugin._prepare_build()
plugin._finish_build()
self.assertThat(os.path.join(plugin.installdir, 'test-file'),
FileExists())
self.assertThat(os.path.join(plugin.installdir, 'test-file'),
FileContains('${AMENT_CURRENT_PREFIX:=$SNAP}'))
def test_bootstrap_stage_packages_are_included(self):
self.bootstrapper_mock.return_value.get_stage_packages.return_value = [
'foo']
plugin = ament.AmentPlugin('test-part', self.properties, self.project)
self.assertThat(plugin.stage_packages, Contains('foo'))
def test_bootstrap_build_packages_are_included(self):
self.bootstrapper_mock.return_value.get_build_packages.return_value = [
"foo"
]
plugin = ament.AmentPlugin("test-part", self.properties, self.project)
self.assertThat(plugin.build_packages, Contains("foo"))
