tests: Add download manager test

This is a basic smoketest that the collection downloader can be run with
any collection specified.

Fixes: #778

kolibri_explore_plugin/test/test_collectionviews.py

@@ -1,10 +1,14 @@
 import json
+from itertools import product
 
 import pytest
 from django.urls import reverse
+from kolibri.core.content.models import ChannelMetadata
+from kolibri.core.content.models import ContentNode
 from rest_framework.test import APIClient
 
 from .utils import COLLECTIONSDIR
+from .utils import wait_for_background_tasks
 from kolibri_explore_plugin import collectionviews
 
 
@@ -125,3 +129,43 @@ def test_get_should_resume():
         "grade": "artist",
         "name": "0001",
     }
+
+
+@pytest.mark.usefixtures("channel_import_db", "worker", "content_server")
+@pytest.mark.django_db
+@pytest.mark.parametrize(
+    ("grade", "name"),
+    product(
+        collectionviews.COLLECTION_GRADES, collectionviews.COLLECTION_NAMES
+    ),
+)
+def test_download_manager_clean(facility_user, grade, name):
+    """Test collections downloads from no content state"""
+    manifest = collectionviews._content_manifests_by_grade_name[grade][name]
+    manager = collectionviews.CollectionDownloadManager()
+
+    # Drive the manager until the foreground tasks complete.
+    status = manager.get_status()
+    manager.start(manifest, facility_user)
+    while status["stage"] != collectionviews.DownloadStage.COMPLETED.name:
+        manager.update(facility_user)
+        status = manager.get_status()
+
+    # Wait for the background download tasks to complete. Those can take a
+    # while to complete, so give it a full minute.
+    wait_for_background_tasks(60)
+
+    # Check that the correct number of channels, nodes and tags are present.
+    # Each pack has 2 channels with 2 included nodes each. Each channel has 6
+    # nodes. The included nodes from each channel end up downloading 4 nodes -
+    # root topic, video, subtopic and document. So, each pack will have 8
+    # available nodes. The extra packs don't cause any nodes to become
+    # available since only the thumbnails are downloaded.
+    assert manifest.language in ("en", "es")
+    if manifest.language == "en":
+        num_packs = len(collectionviews.COLLECTION_GRADES_EN)
+    elif manifest.language == "es":
+        num_packs = len(collectionviews.COLLECTION_GRADES_ES)
+    assert ChannelMetadata.objects.count() == 2 * num_packs
+    assert ContentNode.objects.filter().count() == 12 * num_packs
+    assert ContentNode.objects.filter(available=True).count() == 8