def test_get_reports(self, fake_accessor, mock_inspect):
"""Test get_reports for combinations of setup_complete and ingest override."""
initial_month_qty = Config.INITIAL_INGEST_NUM_MONTHS
test_matrix = [
{
"get_setup_complete": True,
"ingest_override": True,
"test_months": 5,
"expected_month_length": 5
},
{
"get_setup_complete": False,
"ingest_override": True,
"test_months": 5,
"expected_month_length": 5
},
{
"get_setup_complete": True,
"ingest_override": False,
"test_months": 5,
"expected_month_length": 2
},
{
"get_setup_complete": False,
"ingest_override": False,
"test_months": 5,
"expected_month_length": 5
},
]
for test in test_matrix:
test_months = test.get("test_months")
fake_accessor.return_value = test.get("get_setup_complete")
Config.INGEST_OVERRIDE = test.get("ingest_override")
Config.INITIAL_INGEST_NUM_MONTHS = test_months
orchestrator = Orchestrator()
months = orchestrator.get_reports(self.aws_provider_uuid)
self.assertEqual(test.get("expected_month_length"), len(months))
for i in range(1, len(months)):
self.assertLess(months[i], months[i - 1])
Config.INGEST_OVERRIDE = False
Config.INITIAL_INGEST_NUM_MONTHS = initial_month_qty
dh = DateHelper()
expected = [dh.this_month_start.date()]
orchestrator = Orchestrator(bill_date=dh.today)
result = orchestrator.get_reports(self.aws_provider_uuid)
self.assertEqual(result, expected)
def test_get_reports(self, fake_accessor, mock_inspect):
"""Test get_reports for combinations of setup_complete and ingest override."""
initial_month_qty = Config.INITIAL_INGEST_NUM_MONTHS
test_matrix = [
{
"get_setup_complete": True,
"ingest_override": True,
"test_months": 5,
"expected_month_length": 5
},
{
"get_setup_complete": False,
"ingest_override": True,
"test_months": 5,
"expected_month_length": 5
},
{
"get_setup_complete": True,
"ingest_override": False,
"test_months": 5,
"expected_month_length": 2
},
{
"get_setup_complete": False,
"ingest_override": False,
"test_months": 5,
"expected_month_length": 5
},
]
for test in test_matrix:
test_months = test.get("test_months")
fake_accessor.return_value = test.get("get_setup_complete")
Config.INGEST_OVERRIDE = test.get("ingest_override")
Config.INITIAL_INGEST_NUM_MONTHS = test_months
orchestrator = Orchestrator()
months = orchestrator.get_reports(self.aws_provider_uuid)
self.assertEqual(test.get("expected_month_length"), len(months))
Config.INGEST_OVERRIDE = False
Config.INITIAL_INGEST_NUM_MONTHS = initial_month_qty
def test_get_reports(self, fake_accessor):
"""Test get_reports for combinations of setup_complete and ingest override."""
initial_month_qty = Config.INITIAL_INGEST_NUM_MONTHS
test_matrix = [{
'get_setup_complete': True,
'ingest_override': True,
'test_months': 5,
'expected_month_length': 5
}, {
'get_setup_complete': False,
'ingest_override': True,
'test_months': 5,
'expected_month_length': 5
}, {
'get_setup_complete': True,
'ingest_override': False,
'test_months': 5,
'expected_month_length': 2
}, {
'get_setup_complete': False,
'ingest_override': False,
'test_months': 5,
'expected_month_length': 5
}]
for test in test_matrix:
test_months = test.get('test_months')
fake_accessor.return_value = test.get('get_setup_complete')
Config.INGEST_OVERRIDE = test.get('ingest_override')
Config.INITIAL_INGEST_NUM_MONTHS = test_months
orchestrator = Orchestrator()
months = orchestrator.get_reports(self.aws_provider_uuid)
self.assertEqual(test.get('expected_month_length'), len(months))
Config.INGEST_OVERRIDE = False
Config.INITIAL_INGEST_NUM_MONTHS = initial_month_qty
