test_contexts.py 2.06 KB
Newer Older
1
# pylint: disable=missing-docstring
2

3
import ddt
4 5 6 7 8
from unittest import TestCase

from track import contexts


9
@ddt.ddt
10 11 12
class TestContexts(TestCase):

    COURSE_ID = 'test/course_name/course_run'
13
    SPLIT_COURSE_ID = 'course-v1:test+course_name+course_run'
14 15
    ORG_ID = 'test'

16 17 18 19 20 21 22 23 24 25 26 27
    @ddt.data(
        (COURSE_ID, ''),
        (COURSE_ID, '/more/stuff'),
        (COURSE_ID, '?format=json'),
        (SPLIT_COURSE_ID, ''),
        (SPLIT_COURSE_ID, '/more/stuff'),
        (SPLIT_COURSE_ID, '?format=json')
    )
    @ddt.unpack
    def test_course_id_from_url(self, course_id, postfix):
        url = 'http://foo.bar.com/courses/{}{}'.format(course_id, postfix)
        self.assert_parses_course_id_from_url(url, course_id)
28

29
    def assert_parses_course_id_from_url(self, format_string, course_id):
30
        self.assertEquals(
31
            contexts.course_context_from_url(format_string.format(course_id=course_id)),
32
            {
33
                'course_id': course_id,
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
                'org_id': self.ORG_ID
            }
        )

    def test_no_course_id_in_url(self):
        self.assert_empty_context_for_url('http://foo.bar.com/dashboard')

    def assert_empty_context_for_url(self, url):
        self.assertEquals(
            contexts.course_context_from_url(url),
            {
                'course_id': '',
                'org_id': ''
            }
        )

50 51 52
    @ddt.data('', '/', '/?', '?format=json')
    def test_malformed_course_id(self, postfix):
        self.assert_empty_context_for_url('http://foo.bar.com/courses/test/course_name{}'.format(postfix))
53

54 55 56 57 58 59 60 61 62 63 64 65
    @ddt.data(
        (COURSE_ID, ''),
        (COURSE_ID, '/more/stuff'),
        (COURSE_ID, '?format=json'),
        (SPLIT_COURSE_ID, ''),
        (SPLIT_COURSE_ID, '/more/stuff'),
        (SPLIT_COURSE_ID, '?format=json')
    )
    @ddt.unpack
    def test_course_id_later_in_url(self, course_id, postfix):
        url = 'http://foo.bar.com/x/y/z/courses/{}{}'.format(course_id, postfix)
        self.assert_parses_course_id_from_url(url, course_id)
66 67 68

    def test_no_url(self):
        self.assert_empty_context_for_url(None)