Try fixing regex lookup

Adds tests
This commit is contained in:
Nikolai R Kristiansen committed 2017-11-01 21:44:15 +01:00
1 parent 4e41af0bd8
commit 2a037fbe8f
2 files changed
+37 -5

No files matched your search

+29 -2
View File
@@ -1,3 +1,30 @@
from django.test import TestCase
from django.contrib.auth.models import User
from django.urls import reverse
from rest_framework.test import APITestCase
# Create your tests here.
from dpdapi.models import Domain, Alias
class RegexTestCase(APITestCase):
aliases = ['ops-alert@example.com', 'yolo-announce@example.com']
def setUp(self):
user = User.objects.create(username='yolo')
d = Domain.objects.create(name='example.com')
for a in self.aliases:
Alias.objects.create(source=a, destination='@example.com', domain=d)
self.client.force_login(user)
def test_get_all(self):
res = self.client.get(reverse('alias-list'))
self.assertEqual(len(res.data), len(self.aliases))
def test_domain_filter(self):
params = {'domain__name': 'example.com'}
res = self.client.get(reverse('alias-list'), data=params)
self.assertEqual(len(res.data), len(self.aliases))
def test_regex_filter(self):
params = {'source_regex': '^ops-|^ops@'}
res = self.client.get(reverse('alias-list'), data=params)
self.assertEqual(len(res.data), 1)