|
1
|
|
|
# coding: utf-8 |
|
2
|
|
|
import arrow |
|
3
|
|
|
|
|
4
|
|
|
from django.test import TestCase |
|
5
|
|
|
from django.conf import settings |
|
6
|
|
|
from django.contrib.auth.models import User |
|
7
|
|
|
from th_holidays.models import Holidays |
|
8
|
|
|
|
|
9
|
|
|
from django_th.models import ServicesActivated, UserService, TriggerService |
|
10
|
|
|
|
|
11
|
|
|
|
|
12
|
|
|
class HolidaysTest(TestCase): |
|
13
|
|
|
|
|
14
|
|
|
""" |
|
15
|
|
|
Holidays Model |
|
16
|
|
|
""" |
|
17
|
|
|
def setUp(self): |
|
18
|
|
|
try: |
|
19
|
|
|
self.user = User.objects.get(username='john') |
|
20
|
|
|
except User.DoesNotExist: |
|
21
|
|
|
self.user = User.objects.create_user( |
|
22
|
|
|
username='john', email='[email protected]', password='doe') |
|
23
|
|
|
|
|
24
|
|
|
def create_holidays(self, status=True, trigger="My first Service"): |
|
25
|
|
|
""" |
|
26
|
|
|
dummy data |
|
27
|
|
|
""" |
|
28
|
|
|
now = arrow.utcnow().to( |
|
29
|
|
|
settings.TIME_ZONE).format('YYYY-MM-DD HH:mm:ss') |
|
30
|
|
|
date_created = now |
|
31
|
|
|
user = self.user |
|
32
|
|
|
status = True |
|
33
|
|
|
service_provider = ServicesActivated.objects.create( |
|
34
|
|
|
name='ServiceRss9', status=True, |
|
35
|
|
|
auth_required=False, description='Service RSS2') |
|
36
|
|
|
service_consumer = ServicesActivated.objects.create( |
|
37
|
|
|
name='ServiceEvernote9', status=True, |
|
38
|
|
|
auth_required=True, description='Service Evernote2') |
|
39
|
|
|
provider = UserService.objects.create(user=user, |
|
40
|
|
|
token="", |
|
41
|
|
|
name=service_provider) |
|
42
|
|
|
consumer = UserService.objects.create(user=user, |
|
43
|
|
|
token="AZERTY1234", |
|
44
|
|
|
name=service_consumer) |
|
45
|
|
|
trigger = TriggerService.objects.create(provider=provider, |
|
46
|
|
|
consumer=consumer, |
|
47
|
|
|
user=user, |
|
48
|
|
|
date_created=date_created, |
|
49
|
|
|
description="My first Service", |
|
50
|
|
|
status=status) |
|
51
|
|
|
|
|
52
|
|
|
return Holidays.objects.create(user=user, |
|
53
|
|
|
trigger=trigger, |
|
54
|
|
|
status=status) |
|
55
|
|
|
|
|
56
|
|
|
def test_holidays(self): |
|
57
|
|
|
h = self.create_holidays() |
|
58
|
|
|
self.assertTrue(isinstance(h, Holidays)) |
|
59
|
|
|
self.assertEqual(h.show(), "Holidays for service {}".format(h.trigger)) |
|
60
|
|
|
self.assertEqual(h.__str__(), h.trigger) |
|
61
|
|
|
|
foxmask /
django-th
