def setUp(self, db=None):
self.db = db
with Using(db, [Item], with_transaction=False):
parent = self.parent = Item.create(
title="Foo",
description="Bar",
available=True,
kind="Topic",
id="1",
slug="foo",
path="foopath"
)
self.available_item = Item.create(
title="available_item",
description="Bingo",
available=True,
kind="Topic",
id="2",
slug="avail",
path="avail",
parent=parent
)
self.unavailable_item = Item.create(
title="Unavailable item",
description="baz",
available=False,
kind="Topic",
id="3",
slug="unavail",
path="unavail",
parent=parent
)
def setup_content_paths(context, db):
"""
Creaters available content items and adds their urls to the context object.
:param context: A behave context, to which the attributes "available_content_path" and "unavailable_content_path"
will be added.
:return: None
"""
# These paths are "magic" -- the success or failure of actually visiting the content items in the browser
# depends on these specific values.
context.unavailable_content_path, context.available_content_path = (
"khan/foo/bar/unavail",
"khan/math/arithmetic/addition-subtraction/basic_addition/addition_1/",
)
# This function uses 'iterator_content_items' function to return a list of path, update dict pairs
# It then updates the items with these paths with their update dicts, and then propagates
# availability changes up the topic tree - this means that we can alter the availability of one item
# and make all its parent topics available so that it is navigable to in integration tests.
def iterator_content_items(ids=None, channel="khan", language="en"):
return [(context.available_content_path, {"available": True})]
annotate_content_models(db=db, iterator_content_items=iterator_content_items)
with Using(db, [Item], with_transaction=False):
context._unavailable_item = Item.create(
title="Unavailable item",
description="baz",
available=False,
kind="Video",
id="3",
slug="unavail",
path=context.unavailable_content_path
)
def setup_content_paths(context, db):
"""
Creaters available content items and adds their urls to the context object.
:param context: A behave context, to which the attributes "available_content_path" and "unavailable_content_path"
will be added.
:return: None
"""
# These paths are "magic" -- the success or failure of actually visiting the content items in the browser
# depends on these specific values.
context.unavailable_content_path, context.available_content_path = (
"khan/foo/bar/unavail",
"khan/math/arithmetic/addition-subtraction/basic_addition/addition_1/",
)
# This function uses 'iterator_content_items' function to return a list of path, update dict pairs
# It then updates the items with these paths with their update dicts, and then propagates
# availability changes up the topic tree - this means that we can alter the availability of one item
# and make all its parent topics available so that it is navigable to in integration tests.
def iterator_content_items(ids=None, channel="khan", language="en"):
return [(context.available_content_path, {"available": True})]
annotate_content_models(db=db,
iterator_content_items=iterator_content_items)
with Using(db, [Item], with_transaction=False):
context._unavailable_item = Item.create(
title="Unavailable item",
description="baz",
available=False,
kind="Video",
id="3",
slug="unavail",
path=context.unavailable_content_path)
def setUp(self, db=None):
self.db = db
with Using(db, [Item], with_transaction=False):
self.item = Item(
title=self.TITLE,
available=self.AVAILABLE,
kind=self.KIND,
description="test",
id="counting-out-1-20-objects",
slug="test",
path="thepath",
extra_fields={},
)
self.item.save()
self.version_path = contentload_settings.KHAN_ASSESSMENT_ITEM_VERSION_PATH
self.cleanup = False
if not os.path.exists(self.version_path):
with open(self.version_path, 'w') as f:
f.write("stuff")
self.cleanup = True
def add_test_content_videos(instance, db):
youtube_id = random.choice(TEST_YOUTUBE_IDS)
real_video = Item.create(
title="A real video from KA/Youtube",
description="This video exists",
available=False,
kind="Video",
format="mp4",
id=youtube_id,
youtube_id=youtube_id,
slug="youtube-vid",
path="khan/topic0/topic0-3/real-video",
parent=random.choice(instance.content_subsubtopics).pk,
)
instance.real_video = real_video
instance.content_videos.append(real_video)
def setUp(self, db=None):
self.db = db
with Using(db, [Item], with_transaction=False):
self.item = Item(
title=self.TITLE,
available=self.AVAILABLE,
kind=self.KIND,
description="test",
id="counting-out-1-20-objects",
slug="test",
path="thepath",
extra_fields={},
)
self.item.save()
self.version_path = contentload_settings.KHAN_ASSESSMENT_ITEM_VERSION_PATH
self.cleanup = False
if not os.path.exists(self.version_path):
with open(self.version_path, 'w') as f:
f.write("stuff")
self.cleanup = True
class AnnotateTestCase(TestCase):
TITLE = "testing "
AVAILABLE = False
KIND = "Exercise"
@set_database
def setUp(self, db=None):
self.db = db
with Using(db, [Item], with_transaction=False):
self.item = Item(
title=self.TITLE,
available=self.AVAILABLE,
kind=self.KIND,
description="test",
id="counting-out-1-20-objects",
slug="test",
path="thepath",
extra_fields={},
)
self.item.save()
self.version_path = contentload_settings.KHAN_ASSESSMENT_ITEM_VERSION_PATH
self.cleanup = False
if not os.path.exists(self.version_path):
with open(self.version_path, 'w') as f:
f.write("stuff")
self.cleanup = True
def tearDown(self):
with Using(self.db, [Item], with_transaction=False):
self.item.delete_instance()
if self.cleanup:
try:
os.remove(self.version_path)
except OSError:
pass
def test_update_content_availability_true(self):
with Using(self.db, [Item]):
actual = dict(update_content_availability([unparse_model_data(model_to_dict(self.item))])).get("thepath")
assert actual.get("available")
def test_update_content_availability_false(self):
try:
os.rename(self.version_path, self.version_path + ".bak")
except OSError:
pass
with Using(self.db, [Item]):
actual = dict(update_content_availability([unparse_model_data(model_to_dict(self.item))])).get("thepath")
# Update is only generated if changed from False to True, not from False to False, so should return None.
assert not actual
try:
os.rename(self.version_path + ".bak", self.version_path)
except OSError:
pass
def get_content_items_by_kind(self, kind, **kwargs):
return Item.select().where(Item.kind == kind)
def setup_content_db(instance, db):
# Setup the content.db (defaults to the en version)
with Using(db, [Item], with_transaction=False):
# Root node
instance.content_root = Item.create(title="Khan Academy",
description="",
available=True,
files_complete=0,
total_files="1",
kind="Topic",
parent=None,
id="khan",
slug="khan",
path="khan/",
extra_fields="{}",
youtube_id=None,
remote_size=315846064333,
sort_order=0)
for _i in range(4):
slug = "topic{}".format(_i)
instance.content_subtopics.append(
Item.create(
title="Subtopic {}".format(_i),
description="A subtopic",
available=True,
files_complete=0,
total_files="4",
kind="Topic",
parent=instance.content_root,
id=slug,
slug=slug,
path="khan/{}/".format(slug),
extra_fields="{}",
remote_size=1,
sort_order=_i,
))
# Parts of the content recommendation system currently is hard-coded
# to look for 3rd level recommendations only and so will fail if we
# don't have this level of lookup
for subtopic in instance.content_subtopics:
for _i in range(4):
slug = "{}-{}".format(subtopic.id, _i)
instance.content_subsubtopics.append(
Item.create(
title="{} Subsubtopic {}".format(subtopic.title, _i),
description="A subsubtopic",
available=True,
files_complete=4,
total_files="4",
kind="Topic",
parent=subtopic,
id=slug,
slug=slug,
path="{}{}/".format(subtopic.path, slug),
youtube_id=None,
extra_fields="{}",
remote_size=1,
sort_order=_i,
))
# We need at least 10 exercises in some of the tests to generate enough
# data etc.
# ...and we need at least some exercises in each sub-subtopic
for parent in instance.content_subsubtopics:
# Make former created exercise the prerequisite of the next one
prerequisite = None
for _i in range(4):
slug = "{}-exercise-{}".format(parent.id, _i)
extra_fields = {}
if prerequisite:
extra_fields['prerequisites'] = [prerequisite.id]
new_exercise = Item.create(
title="Exercise {} in {}".format(_i, parent.title),
parent=parent,
description="Solve this",
available=True,
kind="Exercise",
id=slug,
slug=slug,
path="{}{}/".format(parent.path, slug),
sort_order=_i,
**extra_fields)
instance.content_exercises.append(new_exercise)
prerequisite = new_exercise
# Add some videos, too, even though files don't exist
for parent in instance.content_subsubtopics:
for _i in range(4):
slug = "{}-video-{}".format(parent.pk, _i)
instance.content_videos.append(
Item.create(
title="Video {} in {}".format(_i, parent.title),
parent=random.choice(instance.content_subsubtopics),
description="Watch this",
available=True,
kind="Video",
id=slug,
slug=slug,
path="{}{}/".format(parent.path, slug),
extra_fields={
"subtitle_urls": [],
"content_urls": {
"stream": "/foo",
"stream_type": "video/mp4"
},
},
sort_order=_i))
with Using(db, [Item], with_transaction=False):
instance.content_unavailable_item = Item.create(
title="Unavailable item",
description="baz",
available=False,
kind="Video",
id="unavail123",
slug="unavail",
path=instance.content_unavailable_content_path,
parent=random.choice(instance.content_subsubtopics).pk,
)
instance.content_available_content_path = random.choice(
instance.content_exercises).path
def setup_content_db(instance, db):
# Setup the content.db (defaults to the en version)
# Root node
instance.content_root = Item.create(
title="Khan Academy",
description="",
available=True,
files_complete=0,
total_files="1",
kind="Topic",
parent=None,
id="khan",
slug="khan",
path="khan/",
extra_fields="{}",
youtube_id=None,
remote_size=315846064333,
sort_order=0
)
for _i in range(4):
slug = "topic{}".format(_i)
instance.content_subtopics.append(
Item.create(
title="Subtopic {}".format(_i),
description="A subtopic",
available=True,
files_complete=0,
total_files="4",
kind="Topic",
parent=instance.content_root,
id=slug,
slug=slug,
path="khan/{}/".format(slug),
extra_fields="{}",
remote_size=1,
sort_order=_i,
)
)
# Parts of the content recommendation system currently is hard-coded
# to look for 3rd level recommendations only and so will fail if we
# don't have this level of lookup
for subtopic in instance.content_subtopics:
for _i in range(4):
slug = "{}-{}".format(subtopic.id, _i)
instance.content_subsubtopics.append(
Item.create(
title="{} Subsubtopic {}".format(subtopic.title, _i),
description="A subsubtopic",
available=True,
files_complete=4,
total_files="4",
kind="Topic",
parent=subtopic,
id=slug,
slug=slug,
path="{}{}/".format(subtopic.path, slug),
youtube_id=None,
extra_fields="{}",
remote_size=1,
sort_order=_i,
)
)
# We need at least 10 exercises in some of the tests to generate enough
# data etc.
# ...and we need at least some exercises in each sub-subtopic
for parent in instance.content_subsubtopics:
# Make former created exercise the prerequisite of the next one
prerequisite = None
for _i in range(4):
slug = "{}-exercise-{}".format(parent.id, _i)
extra_fields = {}
if prerequisite:
extra_fields['prerequisites'] = [prerequisite.id]
new_exercise = Item.create(
title="Exercise {} in {}".format(_i, parent.title),
parent=parent,
description="Solve this",
available=True,
kind="Exercise",
id=slug,
slug=slug,
path="{}{}/".format(parent.path, slug),
sort_order=_i,
**extra_fields
)
instance.content_exercises.append(new_exercise)
prerequisite = new_exercise
# Add some videos, too, even though files don't exist
for parent in instance.content_subsubtopics:
for _i in range(4):
slug = "{}-video-{}".format(parent.pk, _i)
instance.content_videos.append(
Item.create(
title="Video {} in {}".format(_i, parent.title),
parent=random.choice(instance.content_subsubtopics),
description="Watch this",
available=True,
kind="Video",
id=slug,
slug=slug,
path="{}{}/".format(parent.path, slug),
extra_fields={
"subtitle_urls": [],
"content_urls": {"stream": "/foo", "stream_type": "video/mp4"},
},
sort_order=_i
)
)
instance.content_unavailable_item = Item.create(
title="Unavailable item",
description="baz",
available=False,
kind="Video",
id="00000000000",
youtube_id="00000000000",
slug="unavail",
path=instance.content_unavailable_content_path,
parent=random.choice(instance.content_subsubtopics).pk,
)
instance.content_available_content_path = random.choice(instance.content_exercises).path
class AnnotateTestCase(TestCase):
TITLE = "testing "
AVAILABLE = False
KIND = "Exercise"
@set_database
def setUp(self, db=None):
self.db = db
with Using(db, [Item], with_transaction=False):
self.item = Item(
title=self.TITLE,
available=self.AVAILABLE,
kind=self.KIND,
description="test",
id="counting-out-1-20-objects",
slug="test",
path="thepath",
extra_fields={},
)
self.item.save()
self.version_path = contentload_settings.KHAN_ASSESSMENT_ITEM_VERSION_PATH
self.cleanup = False
if not os.path.exists(self.version_path):
with open(self.version_path, 'w') as f:
f.write("stuff")
self.cleanup = True
def tearDown(self):
with Using(self.db, [Item], with_transaction=False):
self.item.delete_instance()
if self.cleanup:
try:
os.remove(self.version_path)
except OSError:
pass
def test_update_content_availability_true(self):
with Using(self.db, [Item]):
actual = dict(
update_content_availability([
unparse_model_data(model_to_dict(self.item))
])).get("thepath")
assert actual.get("available")
def test_update_content_availability_false(self):
try:
os.rename(self.version_path, self.version_path + ".bak")
except OSError:
pass
with Using(self.db, [Item]):
actual = dict(
update_content_availability([
unparse_model_data(model_to_dict(self.item))
])).get("thepath")
# Update is only generated if changed from False to True, not from False to False, so should return None.
assert not actual
try:
os.rename(self.version_path + ".bak", self.version_path)
except OSError:
pass
def get_content_items_by_kind(self, kind, **kwargs):
return Item.select().where(Item.kind == kind)
