def test_zip_based_artifact_s3_bucket_present_resolve_s3_present(self, template_artifacts_mock):
# implicitly artifacts are zips
template_artifacts_mock.return_value = [ZIP]
mock_params = {"t": MagicMock(), "template_file": MagicMock(), "template": MagicMock(), "s3_bucket": True}
mock_default_map = {"s3_bucket": False}
with self.assertRaises(PackageResolveS3AndS3SetError):
resolve_s3_callback(
ctx=MockContext(info_name="test", parent=None, params=mock_params, default_map=mock_default_map),
param=MagicMock(),
provided_value=True,
artifact=ZIP,
exc_set=PackageResolveS3AndS3SetError,
exc_not_set=PackageResolveS3AndS3NotSetError,
)
# Option is set in the configuration file.
mock_default_map["s3_bucket"] = True
mock_params["s3_bucket"] = False
with self.assertRaises(PackageResolveS3AndS3SetError):
resolve_s3_callback(
ctx=MockContext(info_name="test", parent=None, params=mock_params, default_map=mock_default_map),
param=MagicMock(),
provided_value=True,
artifact=ZIP,
exc_set=PackageResolveS3AndS3SetError,
exc_not_set=PackageResolveS3AndS3NotSetError,
)
def test_image_and_zip_based_artifact_s3_bucket_not_present_resolve_s3_not_present(self, template_artifacts_mock):
template_artifacts_mock.return_value = [IMAGE, ZIP]
mock_params = {"t": MagicMock(), "template_file": MagicMock(), "template": MagicMock(), "s3_bucket": False}
mock_default_map = {"s3_bucket": False}
with self.assertRaises(PackageResolveS3AndS3NotSetError):
resolve_s3_callback(
ctx=MockContext(info_name="test", parent=None, params=mock_params, default_map=mock_default_map),
param=MagicMock(),
provided_value=False,
artifact=ZIP,
exc_set=PackageResolveS3AndS3SetError,
exc_not_set=PackageResolveS3AndS3NotSetError,
)
def test_image_based_artifact_resolve_s3_present(self,
template_artifacts_mock):
template_artifacts_mock.return_value = [IMAGE]
mock_params = {
"t": MagicMock(),
"template_file": MagicMock(),
"template": MagicMock(),
"s3_bucket": False
}
mock_default_map = {"s3_bucket": False}
# No exception thrown if option is provided or not provided as --s3-bucket or --resolve-s3 is not required.
for provided_option_value in [True, False]:
self.assertEqual(
resolve_s3_callback(
ctx=MockContext(info_name="test",
parent=None,
params=mock_params,
default_map=mock_default_map),
param=MagicMock(),
provided_value=provided_option_value,
artifact=ZIP,
exc_set=PackageResolveS3AndS3SetError,
exc_not_set=PackageResolveS3AndS3NotSetError,
),
provided_option_value,
)
def test_image_and_zip_based_artifact_s3_bucket_present_resolve_s3_not_present(
self, template_artifacts_mock):
template_artifacts_mock.return_value = [IMAGE, ZIP]
mock_params = {
"t": MagicMock(),
"template_file": MagicMock(),
"template": MagicMock(),
"s3_bucket": True
}
mock_default_map = {"s3_bucket": False}
# No exception thrown, there is --s3-bucket option set.
self.assertEqual(
resolve_s3_callback(
ctx=MockContext(info_name="test",
parent=None,
params=mock_params,
default_map=mock_default_map),
param=MagicMock(),
provided_value=False,
artifact=ZIP,
exc_set=PackageResolveS3AndS3SetError,
exc_not_set=PackageResolveS3AndS3NotSetError,
),
False,
)
def test_zip_based_artifact_s3_bucket_not_present_resolve_s3_present(
self, template_artifacts_mock):
# implicitly artifacts are zips
template_artifacts_mock.return_value = [ZIP]
mock_params = {
"t": MagicMock(),
"template_file": MagicMock(),
"template": MagicMock(),
"s3_bucket": False
}
mock_default_map = {"s3_bucket": False}
self.assertEqual(
resolve_s3_callback(
ctx=MockContext(info_name="test",
parent=None,
params=mock_params,
default_map=mock_default_map),
param=MagicMock(),
provided_value=True,
artifact=ZIP,
exc_set=PackageResolveS3AndS3SetError,
exc_not_set=PackageResolveS3AndS3NotSetError,
),
True,
)
