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, [])