def create_fake_project(creator, n_users, privacy, n_components, name, n_tags,
presentation_name, is_registration, is_preprint,
preprint_providers):
auth = Auth(user=creator)
project_title = name if name else fake.science_sentence()
if is_preprint:
providers_to_add = []
if preprint_providers:
providers = preprint_providers.split(',')
for provider in providers:
try:
preprint_provider = models.PreprintProvider.find_one(
Q('_id', 'eq', provider))
except NoResultsFound:
preprint_provider = PreprintProviderFactory(name=provider)
providers_to_add.append(preprint_provider)
privacy = 'public'
project = PreprintFactory(title=project_title,
description=fake.science_paragraph(),
creator=creator,
providers=providers_to_add)
elif is_registration:
project = RegistrationFactory(title=project_title,
description=fake.science_paragraph(),
creator=creator)
else:
project = ProjectFactory(title=project_title,
description=fake.science_paragraph(),
creator=creator)
project.set_privacy(privacy)
for _ in range(n_users):
contrib = create_fake_user()
project.add_contributor(contrib, auth=auth)
if isinstance(n_components, int):
for _ in range(n_components):
NodeFactory(project=project,
title=fake.science_sentence(),
description=fake.science_paragraph(),
creator=creator)
elif isinstance(n_components, list):
render_generations_from_node_structure_list(project, creator,
n_components)
for _ in range(n_tags):
project.add_tag(fake.science_word(), auth=auth)
if presentation_name is not None:
project.add_tag(presentation_name, auth=auth)
project.add_tag('poster', auth=auth)
project.save()
logger.info('Created project: {0}'.format(project.title))
return project
def create_fake_project(creator, n_users, privacy, n_components, name, n_tags, presentation_name, is_registration, is_preprint, preprint_providers):
auth = Auth(user=creator)
project_title = name if name else fake.science_sentence()
if is_preprint:
providers_to_add = []
if preprint_providers:
providers = preprint_providers.split(',')
for provider in providers:
try:
preprint_provider = models.PreprintProvider.find_one(Q('_id', 'eq', provider))
except NoResultsFound:
preprint_provider = PreprintProviderFactory(name=provider)
providers_to_add.append(preprint_provider)
privacy = 'public'
project = PreprintFactory(title=project_title, description=fake.science_paragraph(), creator=creator, providers=providers_to_add)
elif is_registration:
project = RegistrationFactory(title=project_title, description=fake.science_paragraph(), creator=creator)
else:
project = ProjectFactory(title=project_title, description=fake.science_paragraph(), creator=creator)
project.set_privacy(privacy)
for _ in range(n_users):
contrib = create_fake_user()
project.add_contributor(contrib, auth=auth)
if isinstance(n_components, int):
for _ in range(n_components):
NodeFactory(project=project, title=fake.science_sentence(), description=fake.science_paragraph(),
creator=creator)
elif isinstance(n_components, list):
render_generations_from_node_structure_list(project, creator, n_components)
for _ in range(n_tags):
project.add_tag(fake.science_word(), auth=auth)
if presentation_name is not None:
project.add_tag(presentation_name, auth=auth)
project.add_tag('poster', auth=auth)
project.save()
logger.info('Created project: {0}'.format(project.title))
return project
class TestPreprintFiltering(ApiTestCase):
def setUp(self):
super(TestPreprintFiltering, self).setUp()
self.user = AuthUserFactory()
self.provider = PreprintProviderFactory(name='wwe')
self.preprint = PreprintFactory(creator=self.user, providers=[self.provider])
self.preprint.add_tag('nature boy', Auth(self.user), save=False)
self.preprint.add_tag('ric flair', Auth(self.user), save=False)
self.preprint.save()
self.provider_two = PreprintProviderFactory(name='wcw')
self.preprint_two = PreprintFactory(creator=self.user, filename='woo.txt', providers=[self.provider_two])
self.preprint_two.add_tag('nature boy', Auth(self.user), save=False)
self.preprint_two.add_tag('woo', Auth(self.user), save=False)
self.preprint_two.save()
self.preprint_three = PreprintFactory(creator=self.user, filename='stonecold.txt', providers=[self.provider])
self.preprint_three.add_tag('stone', Auth(self.user), save=False)
self.preprint_two.add_tag('cold', Auth(self.user), save=False)
self.preprint_three.save()
def tearDown(self):
super(TestPreprintFiltering, self).tearDown()
Node.remove()
def test_filtering_tags(self):
# both preprint and preprint_two have nature boy
url = '/{}preprints/?filter[tags]={}'.format(API_BASE, 'nature boy')
res = self.app.get(url, auth=self.user.auth)
reg_json = res.json['data']
ids = [each['id'] for each in reg_json]
assert_in(self.preprint._id, ids)
assert_in(self.preprint_two._id, ids)
assert_not_in(self.preprint_three._id, ids)
# filtering two tags
# preprint has both tags; preprint_two only has one
url = '/{}preprints/?filter[tags]={}&filter[tags]={}'.format(API_BASE, 'nature boy', 'ric flair')
res = self.app.get(url, auth=self.user.auth)
reg_json = res.json['data']
ids = [each['id'] for each in reg_json]
assert_in(self.preprint._id, ids)
assert_not_in(self.preprint_two._id, ids)
assert_not_in(self.preprint_three._id, ids)
def test_filter_by_doi(self):
url = '/{}preprints/?filter[doi]={}'.format(API_BASE, self.preprint.preprint_doi)
res = self.app.get(url, auth=self.user.auth)
data = res.json['data']
assert_equal(len(data), 1)
for result in data:
assert_equal(self.preprint._id, result['id'])
