def mocked_dsg(dataset_file, configuration_data, mocker):
with mock_sts():
config = Config()
config.config = configuration_data
dataset_group = config.dataset_group(dataset_file)
dsg = DatasetGroup(
dataset_group_name=dataset_group.dataset_group_name,
dataset_domain=dataset_group.dataset_group_domain,
)
dsg.cli = mocker.MagicMock()
dsg.cli.describe_dataset_group.return_value = {
"DatasetArns": ["arn::1", "arn::2", "arn::3"]
}
dsg.cli.describe_dataset.return_value = {"DatasetArn": 'arn::1', "Status": "ACTIVE", 'LastModificationTime': datetime.now()}
dsg.cli.get_paginator().paginate.return_value = [{
'DatasetImportJobs': [
{
"DatasetImportJobArn": f"arn::{i}",
"Status": "ACTIVE",
"LastModificationTime": datetime.now()
}
for i in range(3)
]
}]
yield dsg
def test_status(dataset_file, configuration_data, mock_forecast_dsg_exists):
config = Config()
config.config = configuration_data
dataset_group = config.dataset_group(dataset_file)
dsg = DatasetGroup(
dataset_group_name=dataset_group.dataset_group_name,
dataset_domain=dataset_group.dataset_group_domain,
)
dsg.cli = mock_forecast_dsg_exists
assert dsg.status == Status.ACTIVE