def test_service_counts(mock_data_client):
# Count services
mock_data_client.find_draft_services_iter.return_value = [
{'status': 'submitted', 'lot': 'saas'},
{'status': 'submitted', 'lot': 'saas'},
{'status': 'submitted', 'lot': 'paas'},
{'status': 'not-submitted', 'lot': 'scs'},
{'status': 'not-submitted', 'lot': 'scs'},
{'status': 'not-submitted', 'lot': 'paas'},
{'status': 'not-submitted', 'lot': 'paas'},
{'status': 'published', 'lot': 'paas'}, # anything not submitted or failed is considered draft
]
draft_counts_adder = export_g8_suppliers.add_draft_counts(mock_data_client, 'framework-slug')
record = draft_counts_adder({'supplier': {'id': 1}})
counts = export_g8_suppliers.service_counts(record)
assert counts == [
('completed_saas', 2),
('completed_paas', 1),
('completed_iaas', 0),
('completed_scs', 0),
('draft_saas', 0),
('draft_paas', 3),
('draft_iaas', 0),
('draft_scs', 2),
]
def test_add_draft_counts(mock_data_client):
mock_data_client.find_draft_services_iter.return_value = [
{'status': 'submitted', 'lot': 'saas'},
{'status': 'submitted', 'lot': 'saas'},
{'status': 'submitted', 'lot': 'paas'},
{'status': 'not-submitted', 'lot': 'saas'},
{'status': 'not-submitted', 'lot': 'paas'},
{'status': 'not-submitted', 'lot': 'paas'},
{'status': 'not-submitted', 'lot': 'paas'},
{'status': 'published', 'lot': 'scs'}, # anything not submitted or failed is considered draft
]
draft_counts_adder = export_g8_suppliers.add_draft_counts(mock_data_client, 'framework-slug')
record = draft_counts_adder({'supplier': {'id': 1}})
print("COUNTS: {}".format(record['counts']))
assert record['counts'] == {
'completed': {
'saas': 2, 'paas': 1,
'iaas': 0, 'scs': 0,
},
'draft': {
'saas': 1, 'paas': 3,
'iaas': 0, 'scs': 1,
},
}