request_test.py

# coding=utf-8
from datetime import datetime
import unittest
import server
import mock

request_ok_json = """{"name": "https://mail.ru", "mimetype": "webpage", "uri": "https://mail.ru", "is_active": false,
                "start_date": "2016-07-19T12:42:00.000Z", "end_date": "2016-07-26T12:42:00.000Z", "duration": "30",
                "is_enabled": 0, "nocache": 0, "play_order": 0, "skip_asset_check": 0}"""

request_json_no_name = """{"name": null, "mimetype": "webpage", "uri": "https://mail.ru", "is_active": false,
                "start_date": "2016-07-19T12:42:00.000Z", "end_date": "2016-07-26T12:42:00.000Z", "duration": "30",
                "is_enabled": 0, "nocache": 0, "play_order": 0, "skip_asset_check": 0}"""

request_json_no_mime = """{"name": "https://mail.ru", "mimetype": null, "uri": "https://mail.ru", "is_active": false,
                "start_date": "2016-07-19T12:42:00.000Z", "end_date": "2016-07-26T12:42:00.000Z", "duration": "30",
                "is_enabled": 0, "nocache": 0, "play_order": 0, "skip_asset_check": 0}"""

request_json2 = """{"name": null, "mimetype": null, "uri": null, "is_active": false,
                "start_date": "2016-07-19T12:42:00.000Z", "end_date": "2016-07-26T12:42:00.000Z", "duration": "30",
                "is_enabled": 0, "nocache": 0, "play_order": 0, "skip_asset_check": 0}"""


class RequestParseTest(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_asset_should_be_correct_V1_0(self):
        server.Request = lambda a: mock.Mock(form={'model': request_ok_json}, files=mock.Mock(get=lambda a: None))
        asset = server.prepare_asset(mock.Mock(environ={}))
        self.assertEqual(asset['duration'], 30)
        self.assertEqual(asset['is_enabled'], 0)
        self.assertEqual(asset['mimetype'], u'webpage')
        self.assertEqual(asset['name'], u'https://mail.ru')
        self.assertEqual(asset['end_date'], datetime(2016, 7, 26, 12, 42))
        self.assertEqual(asset['start_date'], datetime(2016, 7, 19, 12, 42))

    def test_exception_should_rise_if_no_name_presented_V1_0(self):
        server.Request = lambda a: mock.Mock(form={'model': request_json_no_name}, files=mock.Mock(get=lambda a: None))
        with self.assertRaises(Exception):
            server.prepare_asset(mock.Mock(environ={}))

    def test_exception_should_rise_if_no_mime_presented_V1_0(self):
        server.Request = lambda a: mock.Mock(form={'model': request_json_no_mime}, files=mock.Mock(get=lambda a: None))
        with self.assertRaises(Exception):
            server.prepare_asset(mock.Mock(environ={}))

    def test_asset_should_be_correct_V1_1(self):
        server.Request = lambda a: mock.Mock(data=request_ok_json, files=mock.Mock(get=lambda a: None))
        asset = server.prepare_asset(mock.Mock(environ={}))
        self.assertEqual(asset['duration'], 30)
        self.assertEqual(asset['is_enabled'], 0)
        self.assertEqual(asset['mimetype'], u'webpage')
        self.assertEqual(asset['name'], u'https://mail.ru')
        self.assertEqual(asset['end_date'], datetime(2016, 7, 26, 12, 42))
        self.assertEqual(asset['start_date'], datetime(2016, 7, 19, 12, 42))

    def test_exception_should_rise_if_no_name_presented_V1_1(self):
        server.Request = lambda a: mock.Mock(data=request_json_no_name, files=mock.Mock(get=lambda a: None))
        with self.assertRaises(Exception):
            server.prepare_asset(mock.Mock(environ={}))

    def test_exception_should_rise_if_no_mime_presented_V1_1(self):
        server.Request = lambda a: mock.Mock(data=request_json_no_mime, files=mock.Mock(get=lambda a: None))
        with self.assertRaises(Exception):
            server.prepare_asset(mock.Mock(environ={}))