def test_get_at_least_editable(self) -> None:
editable_collection_summary_model = (
collection_models.CollectionSummaryModel(
id='id0',
title='title',
category='category',
objective='objective',
language_code='language_code',
tags=['tags'],
status=constants.ACTIVITY_STATUS_PRIVATE,
community_owned=False,
owner_ids=['a'],
editor_ids=['editor_ids'],
viewer_ids=['viewer_ids'],
contributor_ids=[''],
contributors_summary={},
version=0,
node_count=0,
collection_model_last_updated=None,
collection_model_created_on=None,
))
editable_collection_summary_model.update_timestamps()
editable_collection_summary_model.put()
uneditable_collection_summary_model = (
collection_models.CollectionSummaryModel(
id='id1',
title='title',
category='category',
objective='objective',
language_code='language_code',
tags=['tags'],
status=constants.ACTIVITY_STATUS_PRIVATE,
community_owned=False,
owner_ids=['owner_ids'],
editor_ids=['editor_ids'],
viewer_ids=['viewer_ids'],
contributor_ids=[''],
contributors_summary={},
version=0,
node_count=0,
collection_model_last_updated=None,
collection_model_created_on=None,
))
uneditable_collection_summary_model.update_timestamps()
uneditable_collection_summary_model.put()
collection_summary_models = (
collection_models.CollectionSummaryModel
.get_at_least_editable('a'))
self.assertEqual(1, len(collection_summary_models))
self.assertEqual('id0', collection_summary_models[0].id)
collection_summary_models = (
collection_models.CollectionSummaryModel
.get_at_least_editable('viewer_ids'))
self.assertEqual(0, len(collection_summary_models))
def test_get_non_private(self):
# type: () -> None
public_collection_summary_model = (
collection_models.CollectionSummaryModel(
id='id0',
title='title',
category='category',
objective='objective',
language_code='language_code',
tags=['tags'],
status=constants.ACTIVITY_STATUS_PUBLIC,
community_owned=False,
owner_ids=['owner_ids'],
editor_ids=['editor_ids'],
viewer_ids=['viewer_ids'],
contributor_ids=[''],
contributors_summary={},
version=0,
node_count=0,
collection_model_last_updated=None,
collection_model_created_on=None,
))
public_collection_summary_model.update_timestamps()
public_collection_summary_model.put()
private_collection_summary_model = (
collection_models.CollectionSummaryModel(
id='id1',
title='title',
category='category',
objective='objective',
language_code='language_code',
tags=['tags'],
status=constants.ACTIVITY_STATUS_PRIVATE,
community_owned=False,
owner_ids=['owner_ids'],
editor_ids=['editor_ids'],
viewer_ids=['viewer_ids'],
contributor_ids=[''],
contributors_summary={},
version=0,
node_count=0,
collection_model_last_updated=None,
collection_model_created_on=None,
))
private_collection_summary_model.update_timestamps()
private_collection_summary_model.put()
collection_summary_models = (
collection_models.CollectionSummaryModel.get_non_private())
self.assertEqual(1, len(collection_summary_models))
def test_has_reference_to_user_id(self) -> None:
collection_models.CollectionSummaryModel(
id='id0',
title='title',
category='category',
objective='objective',
language_code='language_code',
community_owned=False,
owner_ids=['owner_id'],
editor_ids=['editor_id'],
viewer_ids=['viewer_id'],
contributor_ids=['contributor_id'],
).put()
self.assertTrue(
collection_models.CollectionSummaryModel.has_reference_to_user_id(
'owner_id'))
self.assertTrue(
collection_models.CollectionSummaryModel.has_reference_to_user_id(
'editor_id'))
self.assertTrue(
collection_models.CollectionSummaryModel.has_reference_to_user_id(
'viewer_id'))
self.assertTrue(
collection_models.CollectionSummaryModel.has_reference_to_user_id(
'contributor_id'))
self.assertFalse(
collection_models.CollectionSummaryModel.has_reference_to_user_id(
'x_id'))
