def test_get_linked_datasets_for_form_many_selected_ids(self):
user = core_factories.User(id='user_selected_ids',
name='user_selected_ids')
container1 = factories.DataContainer(title='container1', users=[user])
container2 = factories.DataContainer(title='container2', users=[user])
dataset1 = factories.Dataset(id='id1',
title='dataset1',
owner_org=container1['id'])
dataset2 = factories.Dataset(id='id2',
title='dataset2',
owner_org=container2['id'])
context = {'model': model, 'user': user['name']}
linked_datasets = get_linked_datasets_for_form(context=context,
user_id=user['id'],
selected_ids=['id2'])
assert_equals(linked_datasets, [
{
'text': 'container1',
'children': [{
'text': 'dataset1',
'value': 'id1'
}]
},
{
'text':
'container2',
'children': [{
'text': 'dataset2',
'value': 'id2',
'selected': 'selected'
}]
},
])
def _get_allowed_linked_datasets():
datasets = []
for container in helpers.get_linked_datasets_for_form():
for dataset in container['children']:
datasets.append(dataset['value'])
return datasets
def test_get_linked_datasets_for_form_none(self):
user = core_factories.User()
context = {'model': model, 'user': user['name']}
linked_datasets = get_linked_datasets_for_form(context=context,
user_id=user['id'])
assert_equals(linked_datasets, [])
