def test_ocp_inputs_insights_upload_no_envs(self):
"""
Test where user passes an invalid ocp argument combination.
"""
with self.assertRaises(SystemExit):
options = {'ocp': True, 'insights_upload': 'true', 'ocp_cluster_id': '123'}
_validate_provider_inputs(self.parser, options)
def test_no_provider_type_fail(self):
"""
Test that parser fails when no provider type is passed.
"""
with self.assertRaises(SystemExit):
options = {}
_validate_provider_inputs(self.parser, options)
def test_invalid_s3_inputs(self):
"""
Test where user passes an invalid s3 argument combination.
"""
with self.assertRaises(SystemExit):
options = {'aws': True, 'aws_bucket_name': 'mybucket'}
_validate_provider_inputs(self.parser, options)
def test_ocp_no_cluster_id(self):
"""
Test where user passes ocp without cluster id combination.
"""
with self.assertRaises(SystemExit):
options = {'ocp': True}
_validate_provider_inputs(self.parser, options)
def test_invalid_gcp_inputs_aws_args(self):
"""
Test where user aws arg when creating gcp data.
"""
with self.assertRaises(SystemExit):
options = {'gcp': True, 'aws_bucket_name': 'my-bucket'}
_validate_provider_inputs(self.parser, options)
def test_invalid_azure_inputs(self):
"""
Test where user passes an invalid azure argument combination.
"""
with self.assertRaises(SystemExit):
args = [
"report", "azure", "--start-date",
str(date.today()), "--azure-report-name", "report"
]
options = vars(self.parser.parse_args(args))
_validate_provider_inputs(self.parser, options)
with self.assertRaises(SystemExit):
args = [
"report", "azure", "--start-date",
str(date.today()), "--aws-s3-bucket-name", "mybucket"
]
self.parser.parse_args(args)
with self.assertRaises(SystemExit):
args = [
"report", "azure", "--start-date",
str(date.today()), "--ocp-cluster-id", "123"
]
self.parser.parse_args(args)
with self.assertRaises(SystemExit):
args = [
"report", "azure", "--start-date",
str(date.today()), "--gcp-report-prefix", "gcp-report"
]
self.parser.parse_args(args)
def test_invalid_s3_inputs(self):
"""
Test where user passes an invalid s3 argument combination.
"""
with self.assertRaises(SystemExit):
args = ["report", "aws", "--start-date", str(date.today()), "--aws-s3-bucket-name", "mybucket"]
options = vars(self.parser.parse_args(args))
_validate_provider_inputs(self.parser, options)
def test_ocp_no_cluster_id(self):
"""
Test where user passes ocp without cluster id combination.
"""
with self.assertRaises(SystemExit):
args = ["report", "ocp", "--start-date", str(date.today())]
options = vars(self.parser.parse_args(args))
_validate_provider_inputs(self.parser, options)
def test_with_empty_args(self):
"""
User passes no args, should fail with SystemExit
"""
with self.assertRaises(SystemExit):
self.parser.parse_args([])
with self.assertRaises(SystemExit):
options = {'aws': False, 'ocp': False, 'azure': False}
_validate_provider_inputs(self.parser, options)
def test_invalid_ocp_inputs(self):
"""
Test where user passes an invalid ocp argument combination.
"""
with self.assertRaises(SystemExit):
options = {
'ocp': True,
'aws_bucket_name': 'mybucket',
'ocp_cluster_id': '123'
}
_validate_provider_inputs(self.parser, options)
def test_valid_s3_no_input(self):
"""
Test where user passes no s3 argument combination.
"""
options = {'aws': True}
valid = _validate_provider_inputs(self.parser, options)
self.assertTrue(valid)
def test_valid_s3_no_input(self):
"""
Test where user passes no s3 argument combination.
"""
args = ["report", "aws", "--start-date", str(date.today())]
options = vars(self.parser.parse_args(args))
self.assertTrue(_validate_provider_inputs(self.parser, options))
def test_ocp_inputs_insights_upload_user_pass(self):
"""
Test where user passes an invalid ocp argument combination.
"""
options = {'ocp': True, 'insights_upload': 'true', 'ocp_cluster_id': '123'}
is_valid, _ = _validate_provider_inputs(self.parser, options)
self.assertTrue(is_valid)
def test_ocp_inputs_insights_upload_no_envs(self):
"""
Test where user passes an invalid ocp argument combination.
"""
with self.assertRaises(SystemExit):
args = [
"report",
"ocp",
"--start-date",
str(date.today()),
"--insights-upload",
"true",
"--ocp-cluster-id",
"123",
]
options = vars(self.parser.parse_args(args))
_validate_provider_inputs(self.parser, options)
def test_ocp_no_insights_upload(self):
"""
Test where user passes ocp without insights upload.
"""
args = ["report", "ocp", "--start-date", str(date.today()), "--ocp-cluster-id", "132"]
options = vars(self.parser.parse_args(args))
is_valid, _ = _validate_provider_inputs(self.parser, options)
self.assertTrue(is_valid)
def test_valid_s3_both_inputs(self):
"""
Test where user passes a valid s3 argument combination.
"""
options = {'aws': True,
'aws_bucket_name': 'mybucket',
'aws_report_name': 'cur'}
valid = _validate_provider_inputs(self.parser, options)
self.assertTrue(valid)
def test_valid_azure_inputs(self):
"""
Test where user passes a valid s3 argument combination.
"""
options = {'azure': True,
'azure_container_name': 'storage',
'azure_report_name': 'report',
'azure_prefix_name': 'value'}
valid = _validate_provider_inputs(self.parser, options)
self.assertTrue(valid)
def test_run_for_azure_enddates(self):
"""That that fix_dates corrects the azure end_date."""
start = date.today().replace(day=1)
end = date.today().replace(day=1)
args = ["report", "azure", "-s", str(start), "-e", str(end)]
parsed_args = self.parser.parse_args(args)
options = vars(parsed_args)
_, provider_type = _validate_provider_inputs(self.parser, options)
self.assertEqual(provider_type, "azure")
with patch("nise.__main__.azure_create_report"):
run(provider_type, options)
self.assertEqual(
options.get("end_date").date(), end + timedelta(days=1))
def test_invalid_azure_inputs(self):
"""
Test where user passes an invalid azure argument combination.
"""
with self.assertRaises(SystemExit):
options = {'azure': True, 'azure_report_name': 'report'}
_validate_provider_inputs(self.parser, options)
with self.assertRaises(SystemExit):
options = {'azure': True, 'aws_bucket_name': 'mybucket'}
_validate_provider_inputs(self.parser, options)
with self.assertRaises(SystemExit):
options = {'azure': True, 'ocp_cluster_id': '123'}
_validate_provider_inputs(self.parser, options)
with self.assertRaises(SystemExit):
options = {'azure': True, 'gcp_report_prefix': 'gcp-report'}
_validate_provider_inputs(self.parser, options)
def test_valid_s3_both_inputs(self):
"""
Test where user passes a valid s3 argument combination.
"""
args = [
"report",
"aws",
"--start-date",
str(date.today()),
"--aws-s3-bucket-name",
"mybucket",
"--aws-s3-report-name",
"cur",
]
options = vars(self.parser.parse_args(args))
self.assertTrue(_validate_provider_inputs(self.parser, options))
def test_ocp_inputs_insights_upload_user_pass(self):
"""
Test where user passes an invalid ocp argument combination.
"""
args = [
"report",
"ocp",
"--start-date",
str(date.today()),
"--insights-upload",
"true",
"--ocp-cluster-id",
"123",
]
options = vars(self.parser.parse_args(args))
is_valid, _ = _validate_provider_inputs(self.parser, options)
self.assertTrue(is_valid)
def test_invalid_gcp_inputs(self):
"""
Test where user args from azure, ocp, and aws when creating gcp data.
"""
with self.assertRaises(SystemExit):
options = {'gcp': True, 'azure_report_name': 'report'}
_validate_provider_inputs(self.parser, options)
with self.assertRaises(SystemExit):
options = {'gcp': True, 'aws_bucket_name': 'mybucket'}
_validate_provider_inputs(self.parser, options)
with self.assertRaises(SystemExit):
options = {'gcp': True, 'ocp_cluster_id': '123'}
_validate_provider_inputs(self.parser, options)
def test_valid_azure_inputs(self):
"""
Test where user passes a valid s3 argument combination.
"""
args = [
"report",
"azure",
"--start-date",
str(date.today()),
"--azure-container-name",
"storage",
"--azure-report-name",
"report",
"--azure-report-prefix",
"value",
"--azure-account-name",
"account",
]
options = vars(self.parser.parse_args(args))
valid = _validate_provider_inputs(self.parser, options)
self.assertTrue(valid)
