76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
import datetime
|
|
import re
|
|
import socket
|
|
import unittest
|
|
import mock
|
|
from django.test import SimpleTestCase
|
|
|
|
from ..views import RootView
|
|
|
|
|
|
class RootUnitTestCase(unittest.TestCase):
|
|
def test_get_context_data(self):
|
|
default_color_hex = '47825b'
|
|
|
|
test_data_sets = [
|
|
{},
|
|
{'hostname': 'localhost'},
|
|
{'hostname': 'abcde'},
|
|
{'hostname': 'frodo.abcdef.example.com'},
|
|
{'hostname': 'aBc-dEf', 'expected_color_hex': 'abcdef'},
|
|
{'hostname': 'ab34EF12cd56ZZ', 'expected_color_hex': 'ab34ef'},
|
|
{'hostname': 'abcde.example.com', 'expected_color_hex': 'abcdee'},
|
|
{'hostname': 'abcdef', 'kwargs': {'color_hex': '123456'}, 'expected_color_hex': '123456'},
|
|
]
|
|
|
|
real_hostname = socket.gethostname()
|
|
|
|
for test_data in test_data_sets:
|
|
hostname = test_data.get('hostname', real_hostname)
|
|
expected_color_hex = test_data.get('expected_color_hex', default_color_hex)
|
|
|
|
with mock.patch('socket.gethostname', return_value=hostname):
|
|
view = RootView()
|
|
if 'kwargs' in test_data:
|
|
ctx = view.get_context_data(**test_data['kwargs'])
|
|
else:
|
|
ctx = view.get_context_data()
|
|
|
|
self.assertIsInstance(ctx, dict)
|
|
self.assertIn('hostname', ctx)
|
|
self.assertEqual(ctx['hostname'], hostname)
|
|
self.assertIn('color_hex', ctx)
|
|
self.assertTrue(re.match('[0-9a-f]{6}', ctx['color_hex']))
|
|
self.assertEqual(ctx['color_hex'], expected_color_hex)
|
|
self.assertIn('time', ctx)
|
|
self.assertIsInstance(ctx['time'], datetime.datetime)
|
|
|
|
|
|
class DjangoAdminDjangoTestCase(SimpleTestCase):
|
|
def test_djangoadmin(self):
|
|
response = self.client.get('/djangoadmin', follow=True)
|
|
self.assertContains(response, 'Django administration')
|
|
|
|
|
|
class RootDjangoTestCase(SimpleTestCase):
|
|
def test_root_template(self):
|
|
response = self.client.get('/')
|
|
self.assertTemplateUsed(response, 'base/root.html')
|
|
|
|
def test_root_context(self):
|
|
response = self.client.get('/')
|
|
self.assertIn('hostname', response.context)
|
|
hostname = socket.gethostname()
|
|
self.assertEqual(response.context['hostname'], hostname)
|
|
self.assertIn('color_hex', response.context)
|
|
self.assertTrue(re.match('[0-9a-f]{6}', response.context['color_hex']))
|
|
self.assertIn('time', response.context)
|
|
self.assertIsInstance(response.context['time'], datetime.datetime)
|
|
|
|
def test_root_content(self):
|
|
response = self.client.get('/')
|
|
hostname = socket.gethostname().capitalize()
|
|
self.assertContains(response, hostname)
|