def test_main_manifest_new_lock(self, mock_logging: Mock,
mock_to_file: Mock, mock_stable: Mock,
*mocks: Any) -> None:
main()
mock_stable.assert_called_with()
mock_to_file.assert_called_with(self.OPENSEARCH_MANIFEST + ".lock")
mock_logging.assert_called_with(
f"Creating {self.OPENSEARCH_MANIFEST}.lock")
def test_main_manifest_new_lock(self, mock_logging, mock_to_file,
mock_stable, *mocks):
with self.assertRaises(SystemExit):
main()
mock_stable.assert_called_with()
mock_to_file.assert_called_with(self.OPENSEARCH_MANIFEST + ".lock")
mock_logging.assert_called_with(
f"Creating {self.OPENSEARCH_MANIFEST}.lock")
def test_main_platform_darwin(self, mock_temp, mock_recorder, mock_builder,
*mocks):
mock_temp.return_value.__enter__.return_value.name = tempfile.gettempdir(
)
main()
self.assertNotEqual(mock_builder.return_value.build.call_count, 0)
self.assertEqual(mock_builder.return_value.build.call_count,
mock_builder.return_value.export_artifacts.call_count)
mock_recorder.return_value.write_manifest.assert_called()
def test_main_manifest_lock_without_changes(self, mock_logging,
mock_to_file, mock_stable,
*mocks):
with self.assertRaises(SystemExit):
main()
mock_stable.assert_called_with()
mock_to_file.assert_not_called()
mock_logging.assert_called_with(
f"No changes since {self.OPENSEARCH_MANIFEST}.lock")
def test_main_with_architecture(self, mock_temp, mock_recorder,
mock_builder, *mocks):
mock_temp.return_value.__enter__.return_value.name = tempfile.gettempdir(
)
main()
self.assertEqual(mock_builder.return_value.build.call_count, 11)
self.assertEqual(mock_builder.return_value.export_artifacts.call_count,
11)
mock_recorder.return_value.write_manifest.assert_called()
def test_main_manifest_lock_without_changes(self, mock_logging: Mock,
mock_to_file: Mock,
mock_stable: Mock,
*mocks: Any) -> None:
main()
mock_stable.assert_called_with()
mock_to_file.assert_not_called()
mock_logging.assert_called_with(
f"No changes since {self.OPENSEARCH_MANIFEST}.lock")
def test_main_platform_windows(self, mock_temp, mock_recorder,
mock_builder, *mocks):
mock_temp.return_value.__enter__.return_value.name = tempfile.gettempdir(
)
main()
# excludes performance analyzer and k-nn
for call_args in mock_builder.call_args:
if len(call_args) > 1:
component = call_args[0]
self.assertNotIn("k-nn", component.name.lower())
self.assertNotIn("analyzer", component.name.lower())
self.assertNotEqual(mock_builder.call_count, 0)
self.assertEqual(mock_builder.return_value.build.call_count,
mock_builder.call_count)
self.assertEqual(mock_builder.return_value.export_artifacts.call_count,
mock_builder.call_count)
mock_recorder.return_value.write_manifest.assert_called()
def test_usage(self):
with self.assertRaises(SystemExit):
main()
out, _ = self.capfd.readouterr()
self.assertTrue(out.startswith("usage:"))
def test_main_with_invalid_architecture(self, *mocks):
with self.assertRaises(SystemExit):
main()
def test_main_with_invalid_platform(self, *mocks):
with self.assertRaises(SystemExit):
main()