TestRangeParser

Complexity

Total Complexity

15

Size/Duplication

Total Lines	84
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	test_get_page_size()	0	4	1
A	test_range()	0	4	1
A	tearDown()	0	2	1
A	test_content_range()	0	4	1
A	setUp()	0	3	1
A	test_range_header_invallid_header()	0	11	2
A	test_range_header()	0	5	1
A	test_set_response_headers_no_length()	0	5	1
A	test_range_header_max_end()	0	5	1
A	test_range_header_incorrect_end()	0	5	1
A	test_set_response_headers()	0	6	1
A	test_range_header_max_end_2()	0	5	1
A	test_range_page_5()	0	5	1
A	test_range_page()	0	5	1

# -*- coding: utf-8 -*-
import unittest
from pyramid import testing
from oe_utils.range_parser import Range, RangeParseException

try:
    from unittest.mock import Mock, MagicMock
except ImportError:
    from mock import Mock, MagicMock  # pragma: no cover


class TestRangeParser(unittest.TestCase):

    def setUp(self):
        self.config = testing.setUp()
        self.request = testing.DummyRequest()

    def tearDown(self):
        testing.tearDown()

    def test_range(self):
        range = Range.parse(self.request)
        self.assertEqual(0, range.start)
        self.assertEqual(11, range.end)

    def test_range_header(self):
        self.request.headers['Range'] = 'items=0-19'
        range = Range.parse(self.request)
        self.assertEqual(0, range.start)
        self.assertEqual(19, range.end)

    def test_range_header_incorrect_end(self):
        self.request.headers['Range'] = 'items=5-2'
        range = Range.parse(self.request)
        self.assertEqual(5, range.start)
        self.assertEqual(5, range.end)

    def test_range_header_max_end(self):
        self.request.headers['Range'] = 'items=0-75'
        range = Range.parse(self.request)
        self.assertEqual(0, range.start)
        self.assertEqual(50, range.end)

    def test_range_header_max_end_2(self):
        self.request.headers['Range'] = 'items=0-75'
        range = Range.parse(self.request, max_end=None)
        self.assertEqual(0, range.start)
        self.assertEqual(75, range.end)

    def test_range_header_invallid_header(self):
        self.request.headers['Range'] = 'items=0->19'
        error = None
        error_thrown = False
        try:
            Range.parse(self.request)
        except Exception as e:
            error = e
            error_thrown = True
        self.assertTrue(error_thrown)
        self.assertEqual('range header does not match expected format', str(error))

    def test_range_page(self):
        self.request.params['pagina'] = 2
        range = Range.parse(self.request)
        self.assertEqual(12, range.start)
        self.assertEqual(23, range.end)

    def test_range_page_5(self):
        self.request.params['pagina'] = 5
        range = Range.parse(self.request)
        self.assertEqual(48, range.start)
        self.assertEqual(59, range.end)

    def test_content_range(self):
        self.request.params['pagina'] = 2
        range = Range.parse(self.request)
        self.assertEqual('items 12-23/35', range.content_range(35))

    def test_get_page_size(self):
        self.request.params['page'] = 2
        range = Range.parse(self.request)
        self.assertEqual(12, range.get_page_size())

    def test_set_response_headers(self):
        self.request.headers['Range'] = 'items=0-19'
        range = Range.parse(self.request)
        response = Mock()
        range.set_response_headers(response, 50)
        self.assertEqual('items 0-19/50', response.content_range)

    def test_set_response_headers_no_length(self):
        self.request.headers['Range'] = 'items=0-19'
        range = Range.parse(self.request)
        response = Mock()
        self.assertRaises(RangeParseException,range.set_response_headers, response, None)