From 7e29a5a65c8d492676ba0ca509a0aab14538d292 Mon Sep 17 00:00:00 2001
From: Usman Khalid <2200617@gmail.com>
Date: Mon, 22 Sep 2014 18:34:27 +0500
Subject: [PATCH] Added tests for parse_range_header() in contentserver.

PLAT-104
---
 common/djangoapps/contentserver/tests/test.py | 62 ++++++++++++++++++-
 1 file changed, 60 insertions(+), 2 deletions(-)

diff --git a/common/djangoapps/contentserver/tests/test.py b/common/djangoapps/contentserver/tests/test.py
index 63a2a0f7cb..36411842ae 100644
--- a/common/djangoapps/contentserver/tests/test.py
+++ b/common/djangoapps/contentserver/tests/test.py
@@ -3,20 +3,22 @@ Tests for StaticContentServer
 """
 import copy
 import logging
+import unittest
 from uuid import uuid4
 
 from django.conf import settings
 from django.test.client import Client
 from django.test.utils import override_settings
 
-from student.models import CourseEnrollment
-
 from xmodule.contentstore.django import contentstore
 from xmodule.modulestore.django import modulestore
 from opaque_keys.edx.locations import SlashSeparatedCourseKey
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
 from xmodule.modulestore.xml_importer import import_from_xml
 
+from contentserver.middleware import parse_range_header
+from student.models import CourseEnrollment
+
 log = logging.getLogger(__name__)
 
 TEST_DATA_CONTENTSTORE = copy.deepcopy(settings.CONTENTSTORE)
@@ -208,3 +210,59 @@ class ContentStoreToyCourseTest(ModuleStoreTestCase):
         )
 
         self.assertEqual(resp.status_code, 416)
+
+
+class ParseRangeHeaderTestCase(unittest.TestCase):
+    """
+    Tests for the parse_range_header function.
+    """
+
+    def setUp(self):
+        self.content_length = 10000
+
+    def test_bytes_unit(self):
+        unit, __ = parse_range_header('bytes=100-', self.content_length)
+        self.assertEqual(unit, 'bytes')
+
+    def test_invalid_syntax(self):
+        self.assertRaisesRegexp(ValueError, 'Invalid syntax', parse_range_header, 'bytes', self.content_length)
+        self.assertRaisesRegexp(ValueError, 'Invalid syntax', parse_range_header, 'bytes=', self.content_length)
+        self.assertRaisesRegexp(
+            ValueError, 'too many values to unpack', parse_range_header, 'bytes=0=', self.content_length
+        )
+        self.assertRaisesRegexp(ValueError, 'Invalid syntax', parse_range_header, 'bytes=0', self.content_length)
+        self.assertRaisesRegexp(ValueError, 'Invalid syntax', parse_range_header, 'bytes=0-10,0', self.content_length)
+
+    def test_byte_range_spec(self):
+        __, ranges = parse_range_header('bytes=100-', self.content_length)
+        self.assertEqual(len(ranges), 1)
+        self.assertEqual(ranges[0], (100, 9999))
+
+        __, ranges = parse_range_header('bytes=1000-', self.content_length)
+        self.assertEqual(len(ranges), 1)
+        self.assertEqual(ranges[0], (1000, 9999))
+
+        __, ranges = parse_range_header('bytes=100-199, 200-', self.content_length)
+        self.assertEqual(len(ranges), 2)
+        self.assertEqual(ranges, [(100, 199), (200, 9999)])
+
+        __, ranges = parse_range_header('bytes=100-199, 200-499', self.content_length)
+        self.assertEqual(len(ranges), 2)
+        self.assertEqual(ranges, [(100, 199), (200, 499)])
+
+        self.assertRaisesRegexp(
+            ValueError, 'invalid literal for int()', parse_range_header, 'bytes=one-20', self.content_length
+        )
+
+    def test_suffix_byte_range_spec(self):
+        __, ranges = parse_range_header('bytes=-100', self.content_length)
+        self.assertEqual(len(ranges), 1)
+        self.assertEqual(ranges[0], (9900, 9999))
+
+        __, ranges = parse_range_header('bytes=-100, -200', self.content_length)
+        self.assertEqual(len(ranges), 2)
+        self.assertEqual(ranges, [(9900, 9999), (9800, 9999)])
+
+        self.assertRaisesRegexp(
+            ValueError, 'invalid literal for int()', parse_range_header, 'bytes=-one', self.content_length
+        )