mirror of https://github.com/snachodog/mybuddy.git
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from django.test import TestCase
|
||
|
|
from django.utils import timezone
|
||
|
|
|
||
|
|
from rest_framework.views import APIView
|
||
|
|
from rest_framework.test import APIRequestFactory
|
||
|
|
|
||
|
|
from api.utils import filter_by_params
|
||
|
|
from core.models import Child
|
||
|
|
|
||
|
|
|
||
|
|
class UtilsTestCase(TestCase):
|
||
|
|
def test_filter_by_params(self):
|
||
|
|
factory = APIRequestFactory()
|
||
|
|
|
||
|
|
Child.objects.create(
|
||
|
|
first_name='First',
|
||
|
|
last_name='Child',
|
||
|
|
birth_date=timezone.localdate())
|
||
|
|
Child.objects.create(
|
||
|
|
first_name='Second',
|
||
|
|
last_name='Child',
|
||
|
|
birth_date=timezone.localdate())
|
||
|
|
|
||
|
|
request = factory.get('/children/')
|
||
|
|
request = APIView().initialize_request(request)
|
||
|
|
response = filter_by_params(request, Child, [])
|
||
|
|
self.assertTrue(response, Child.objects.all())
|
||
|
|
|
||
|
|
request = factory.get('/children/', {'first_name': 'First'})
|
||
|
|
request = APIView().initialize_request(request)
|
||
|
|
response = filter_by_params(request, Child, ['first_name'])
|
||
|
|
self.assertTrue(response, Child.objects.filter(first_name='First'))
|