test_assets.py

"""
Unit tests for the asset upload endpoint.
"""

import json
from datetime import datetime
from io import BytesIO
from pytz import UTC
from unittest import TestCase, skip
from .utils import CourseTestCase
from django.core.urlresolvers import reverse
from contentstore.views import assets


class AssetsTestCase(CourseTestCase):
    def setUp(self):
        super(AssetsTestCase, self).setUp()
        self.url = reverse("asset_index", kwargs={
            'org': self.course.location.org,
            'course': self.course.location.course,
            'name': self.course.location.name,
        })

    def test_basic(self):
        resp = self.client.get(self.url)
        self.assertEquals(resp.status_code, 200)

    def test_json(self):
        resp = self.client.get(
            self.url,
            HTTP_ACCEPT="application/json",
            HTTP_X_REQUESTED_WITH="XMLHttpRequest",
        )
        self.assertEquals(resp.status_code, 200)
        content = json.loads(resp.content)
        self.assertIsInstance(content, list)


class UploadTestCase(CourseTestCase):
    """
    Unit tests for uploading a file
    """
    def setUp(self):
        super(UploadTestCase, self).setUp()
        self.url = reverse("upload_asset", kwargs={
            'org': self.course.location.org,
            'course': self.course.location.course,
            'coursename': self.course.location.name,
        })

    @skip("CorruptGridFile error on continuous integration server")
    def test_happy_path(self):
        f = BytesIO("sample content")
        f.name = "sample.txt"
        resp = self.client.post(self.url, {"name": "my-name", "file": f})
        self.assert2XX(resp.status_code)

    def test_no_file(self):
        resp = self.client.post(self.url, {"name": "file.txt"})
        self.assert4XX(resp.status_code)

    def test_get(self):
        resp = self.client.get(self.url)
        self.assertEquals(resp.status_code, 405)


class AssetsToJsonTestCase(TestCase):
    """
    Unit tests for transforming the results of a database call into something
    we can send out to the client via JSON.
    """
    def test_basic(self):
        upload_date = datetime(2013, 6, 1, 10, 30, tzinfo=UTC)
        asset = {
            "displayname": "foo",
            "chunkSize": 512,
            "filename": "foo.png",
            "length": 100,
            "uploadDate": upload_date,
            "_id": {
                "course": "course",
                "org": "org",
                "revision": 12,
                "category": "category",
                "name": "name",
                "tag": "tag",
            }
        }
        output = assets.assets_to_json_dict([asset])
        self.assertEquals(len(output), 1)
        compare = output[0]
        self.assertEquals(compare["name"], "foo")
        self.assertEquals(compare["path"], "foo.png")
        self.assertEquals(compare["uploaded"], upload_date.isoformat())
        self.assertEquals(compare["id"], "/tag/org/course/12/category/name")