Merge pull request #45 from rpkilby/test-restructure

Restructure tests

Author: philipn

manage.py

@@ -3,7 +3,7 @@
 import sys
 
 if __name__ == "__main__":
-    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "testsettings")
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tests.settings")
 
     from django.core.management import execute_from_command_line
 

tests/__init__.py

@@ -0,0 +1,178 @@
+
+from rest_framework_filters import filters
+from rest_framework_filters.filters import RelatedFilter, AllLookupsFilter
+from rest_framework_filters.filterset import FilterSet
+
+
+from .models import (
+    User, Note, Post, Cover, Page, A, B, C, Person, Tag, BlogPost,
+)
+
+
+class NoteFilterWithAll(FilterSet):
+    title = AllLookupsFilter(name='title')
+
+    class Meta:
+        model = Note
+
+
+class UserFilter(FilterSet):
+    username = filters.CharFilter(name='username')
+    email = filters.CharFilter(name='email')
+
+    class Meta:
+        model = User
+
+
+class UserFilterWithAll(FilterSet):
+    username = AllLookupsFilter(name='username')
+    email = filters.CharFilter(name='email')
+
+    class Meta:
+        model = User
+
+
+class NoteFilterWithRelated(FilterSet):
+    title = filters.CharFilter(name='title')
+    author = RelatedFilter(UserFilter, name='author')
+
+    class Meta:
+        model = Note
+
+
+class NoteFilterWithRelatedAll(FilterSet):
+    title = filters.CharFilter(name='title')
+    author = RelatedFilter(UserFilterWithAll, name='author')
+
+    class Meta:
+        model = Note
+
+
+class NoteFilterWithRelatedAllDifferentFilterName(FilterSet):
+    title = filters.CharFilter(name='title')
+    writer = RelatedFilter(UserFilterWithAll, name='author')
+
+    class Meta:
+        model = Note
+
+
+class PostFilterWithRelated(FilterSet):
+    note = RelatedFilter(NoteFilterWithRelatedAll, name='note')
+
+    class Meta:
+        model = Post
+
+
+class CoverFilterWithRelated(FilterSet):
+    comment = filters.CharFilter(name='comment')
+    post = RelatedFilter(PostFilterWithRelated, name='post')
+
+    class Meta:
+        model = Cover
+
+
+class PageFilterWithRelated(FilterSet):
+    title = filters.CharFilter(name='title')
+    previous_page = RelatedFilter(PostFilterWithRelated, name='previous_page')
+
+    class Meta:
+        model = Page
+
+
+class TagFilter(FilterSet):
+    name = AllLookupsFilter(name='name')
+
+    class Meta:
+        model = Tag
+
+
+class BlogPostFilter(FilterSet):
+    title = filters.CharFilter(name='title')
+    tags = RelatedFilter(TagFilter, name='tags')
+
+    class Meta:
+        model = BlogPost
+
+
+class UserFilterWithDifferentName(FilterSet):
+    name = filters.CharFilter(name='username')
+
+    class Meta:
+        model = User
+
+
+class NoteFilterWithRelatedDifferentName(FilterSet):
+    author = RelatedFilter(UserFilterWithDifferentName, name='author')
+
+    class Meta:
+        model = Note
+
+
+#############################################################
+# Recursive filtersets
+#############################################################
+class AFilter(FilterSet):
+    title = filters.CharFilter(name='title')
+    b = RelatedFilter('tests.filters.BFilter', name='b')
+
+    class Meta:
+        model = A
+
+
+class CFilter(FilterSet):
+    title = filters.CharFilter(name='title')
+    a = RelatedFilter(AFilter, name='a')
+
+    class Meta:
+        model = C
+
+
+class BFilter(FilterSet):
+    name = AllLookupsFilter(name='name')
+    c = RelatedFilter(CFilter, name='c')
+
+    class Meta:
+        model = B
+
+
+class PersonFilter(FilterSet):
+    name = AllLookupsFilter(name='name')
+    best_friend = RelatedFilter('tests.filters.PersonFilter', name='best_friend')
+
+    class Meta:
+        model = Person
+
+
+#############################################################
+# Extensions to django_filter fields for DRF.
+#############################################################
+class AllLookupsPersonDateFilter(FilterSet):
+    date_joined = AllLookupsFilter(name='date_joined')
+    time_joined = AllLookupsFilter(name='time_joined')
+    datetime_joined = AllLookupsFilter(name='datetime_joined')
+
+    class Meta:
+        model = Person
+
+
+class ExplicitLookupsPersonDateFilter(FilterSet):
+    date_joined = AllLookupsFilter(name='date_joined')
+    time_joined = AllLookupsFilter(name='time_joined')
+    datetime_joined = AllLookupsFilter(name='datetime_joined')
+
+    class Meta:
+        model = Person
+
+
+class InSetLookupPersonIDFilter(FilterSet):
+    pk = AllLookupsFilter('id')
+
+    class Meta:
+        model = Person
+
+
+class InSetLookupPersonNameFilter(FilterSet):
+    name = AllLookupsFilter('name')
+
+    class Meta:
+        model = Person

tests/filters.py

@@ -0,0 +1,59 @@
+
+from django.db import models
+from django.contrib.auth.models import User
+
+
+class Note(models.Model):
+    title = models.CharField(max_length=100)
+    content = models.TextField()
+    author = models.ForeignKey(User)
+
+
+class Post(models.Model):
+    note = models.ForeignKey(Note)
+    content = models.TextField()
+
+
+class Cover(models.Model):
+    comment = models.CharField(max_length=100)
+    post = models.ForeignKey(Post)
+
+
+class Page(models.Model):
+    title = models.CharField(max_length=100)
+    content = models.TextField()
+    previous_page = models.ForeignKey('self', null=True)
+
+
+class A(models.Model):
+    title = models.CharField(max_length=100)
+    b = models.ForeignKey('B', null=True)
+
+
+class C(models.Model):
+    title = models.CharField(max_length=100)
+    a = models.ForeignKey(A, null=True)
+
+
+class B(models.Model):
+    name = models.CharField(max_length=100)
+    c = models.ForeignKey(C, null=True)
+
+
+class Person(models.Model):
+    name = models.CharField(max_length=100)
+    best_friend = models.ForeignKey('self', null=True)
+
+    date_joined = models.DateField(auto_now_add=True)
+    time_joined = models.TimeField(auto_now_add=True)
+    datetime_joined = models.DateTimeField(auto_now_add=True)
+
+
+class Tag(models.Model):
+    name = models.CharField(max_length=100)
+
+
+class BlogPost(models.Model):
+    title = models.CharField(max_length=100)
+    content = models.TextField()
+    tags = models.ManyToManyField(Tag, null=True)

tests/models.py

@@ -9,7 +9,7 @@
     'django.contrib.auth',
     'django.contrib.contenttypes',
     'rest_framework',
-    'rest_framework_filters',
+    'tests',
 )
 
 SECRET_KEY = 'testsecretkey'