def get_unpropagated_chunks():
"""
Get a list of unpropagated chunks
:return: List of chunk ids
"""
results = db_manager.get_first_n_built_chunk_ids(number_of_chunks)
temp = []
for chunk in results:
temp.append(chunk['chunk_id'])
return jsonify(chunks=temp)
def test_get_first_n_built_chunk_ids():
# Setup test env
db_manager = DatabaseManager()
db_manager.operate_on_chunk_relation('INSERT', chunk_id='101c')
db_manager.operate_on_host_relation('INSERT', host='101.101.101.101:101', type='Test Server')
db_manager.operate_on_index_builder_relation('INSERT', chunk_id='101c', host='101.101.101.101:101', task='built')
# Test
result = db_manager.get_first_n_built_chunk_ids(1)
if len(result) == 1:
print('> PASSED | get_first_n_built_chunk_ids()')
else:
print('> FAILED | get_first_n_built_chunk_ids()')
# Clean up test env
db_manager.operate_on_chunk_relation('DELETE', chunk_id='101c')
db_manager.operate_on_host_relation('DELETE', host='101.101.101.101:101')
db_manager.operate_on_index_builder_relation('DELETE', chunk_id='101c')
