def test_bundle_install_min_patch(self, mock_os_rename: Mock,
mock_check_call: Mock) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.1.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(BuildManifest.from_path(manifest_path),
artifacts_path, MagicMock())
bundle.install_min()
self.assertEqual(mock_check_call.call_count, 1)
mock_os_rename.assert_called_with(
os.path.join(bundle.tmp_dir.name, "opensearch-1.1.0"),
os.path.join(bundle.tmp_dir.name, "opensearch-1.1.1"))
mock_check_call.assert_has_calls([
call(
" ".join([
"bash",
ScriptFinder.find_install_script("OpenSearch"),
"-v 1.1.1-SNAPSHOT",
"-p linux",
"-a x64",
"-f",
artifacts_path,
"-o",
bundle.min_dist.archive_path,
]),
cwd=bundle.min_dist.archive_path,
shell=True,
),
])
def test_bundle_install_min_distribution(self) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-windows-1.3.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(BuildManifest.from_path(manifest_path),
artifacts_path, MagicMock())
with patch("subprocess.check_call") as mock_check_call:
bundle.install_min()
self.assertEqual(mock_check_call.call_count, 1)
mock_check_call.assert_has_calls([
call(
" ".join([
"bash",
ScriptFinder.find_install_script("OpenSearch"),
"-v 1.3.0",
"-p windows",
"-a x64",
"-d zip",
"-f",
artifacts_path,
"-o",
bundle.min_dist.archive_path,
]),
cwd=bundle.min_dist.archive_path,
shell=True,
),
])
def test_bundle_install_min_no_patch(self, mock_os_rename: Mock,
mock_check_call: Mock) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(BuildManifest.from_path(manifest_path),
artifacts_path, MagicMock())
bundle.install_min()
mock_os_rename.assert_not_called()
mock_check_call.assert_called()
def test_bundle_install_components(self,
bundle_install_plugin: Mock) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.0.yml")
bundle = BundleOpenSearch(
BuildManifest.from_path(manifest_path),
os.path.join(os.path.dirname(__file__), "data", "artifacts"),
MagicMock(),
)
bundle.install_components()
self.assertEqual(bundle_install_plugin.call_count, 12)
def test_bundle_install_plugin(self, path_isfile: Mock) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(BuildManifest.from_path(manifest_path),
artifacts_path, MagicMock())
plugin = bundle.components['job-scheduler']
with patch("shutil.copyfile") as mock_copyfile:
with patch("subprocess.check_call") as mock_check_call:
bundle.install_plugin(plugin)
self.assertEqual(mock_copyfile.call_count, 1)
self.assertEqual(mock_check_call.call_count, 2)
script = "opensearch-plugin.bat" if current_platform(
) == "windows" else "opensearch-plugin"
install_plugin_bin = os.path.join(bundle.min_dist.archive_path,
"bin", script)
mock_check_call.assert_has_calls([
call(
f'{install_plugin_bin} install --batch file:{os.path.join(bundle.tmp_dir.name, "opensearch-job-scheduler-1.1.0.0.zip")}',
cwd=bundle.min_dist.archive_path,
shell=True,
),
call(
" ".join([
"bash",
ScriptFinder.find_install_script(
"opensearch-job-scheduler"),
"-v 1.1.0",
"-p linux",
"-a x64",
"-f",
artifacts_path,
"-o",
bundle.min_dist.archive_path,
]),
cwd=bundle.min_dist.archive_path,
shell=True,
),
])
self.assertEqual(path_isfile.call_count, 2)
def test_bundle_package_tar(self) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(
BuildManifest.from_path(manifest_path),
artifacts_path,
MagicMock(package_name="opensearch.tar"),
)
with patch("tarfile.open") as mock_tarfile_open:
mock_tarfile_add = MagicMock()
mock_tarfile_open.return_value.__enter__.return_value.add = mock_tarfile_add
with patch("shutil.copyfile") as mock_copyfile:
bundle.package(os.path.dirname(__file__))
mock_tarfile_open.assert_called_with("opensearch.tar", "w:gz")
mock_tarfile_add.assert_called_with(os.path.join(
bundle.tmp_dir.name, "opensearch-1.1.0"),
arcname="opensearch-1.1.0")
self.assertEqual(mock_copyfile.call_count, 1)
def test_bundle_package_zip(self) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-windows-1.3.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(
BuildManifest.from_path(manifest_path),
artifacts_path,
MagicMock(package_name="opensearch.zip"),
)
with patch("assemble_workflow.dist.ZipFile") as mock_zipfile_open:
mock_zipfile_write = MagicMock()
mock_zipfile_open.return_value.__enter__.return_value.write = mock_zipfile_write
with patch("shutil.copyfile") as mock_copyfile:
bundle.package(os.path.dirname(__file__))
mock_zipfile_open.assert_called_with("opensearch.zip", "w",
zipfile.ZIP_DEFLATED)
mock_zipfile_write.assert_called_with(
os.path.join(bundle.tmp_dir.name, "opensearch-1.3.0",
"opensearch.txt"), "opensearch.txt")
self.assertEqual(mock_copyfile.call_count, 1)
def test_bundle_include_common_utils(self, mock_path_isile: Mock,
mock_check_call: Mock) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle_recorder = MagicMock()
bundle = BundleOpenSearch(BuildManifest.from_path(manifest_path),
artifacts_path, bundle_recorder)
with patch("shutil.copyfile"):
bundle.install_components()
bundle_recorder.record_component.assert_has_calls([
call(bundle.components["OpenSearch"],
"dist/opensearch-min-1.1.0-linux-x64.tar.gz"),
])
bundle_recorder.record_component.assert_has_calls([
call(bundle.components["common-utils"]),
call(bundle.components["job-scheduler"],
"plugins/opensearch-job-scheduler-1.1.0.0.zip"),
])
mock_path_isile.assert_called()
mock_check_call.assert_called()
def test_bundle_opensearch(self) -> None:
manifest_path = os.path.join(os.path.dirname(__file__), "data",
"opensearch-build-linux-1.1.0.yml")
artifacts_path = os.path.join(os.path.dirname(__file__), "data",
"artifacts")
bundle = BundleOpenSearch(BuildManifest.from_path(manifest_path),
artifacts_path, MagicMock())
self.assertEqual(bundle.min_dist.name, "OpenSearch")
self.assertEqual(len(bundle.components), 15)
self.assertEqual(bundle.artifacts_dir, artifacts_path)
self.assertIsNotNone(bundle.bundle_recorder)
self.assertEqual(bundle.installed_plugins, [])
self.assertTrue(
bundle.min_dist.path.endswith(
"opensearch-min-1.1.0-linux-x64.tar.gz"))
self.assertIsNotNone(bundle.min_dist.archive_path)