import datetime import json import mock from django.test import TestCase from django.utils.encoding import smart_text from data_explorer_example.models import Author, Book, City from data_explorer_example.data_explorer import Endpoint, BookFilter, get_age import django_data_explorer.filterform def _get_date(date_str): return datetime.datetime.strptime(date_str, "%Y-%m-%d") def _client_params_ordering(order_column, order_dir): return json.dumps({ 'datatables_params': {"iDisplayStart": 0, "iDisplayLength": 10, "sEcho": 1, "iSortingCols": 1, "iSortCol_0": order_column, "sSortDir_0": order_dir} }) class TableFiltersTestCase(TestCase): def setUp(self): self.fakeauthor = Author.objects.create(name="FakeAuthor", birth_date="2015-01-01") self.fakeauthor2 = Author.objects.create(name="FakeAuthor2", birth_date="2014-01-01") self.testbook1 = Book.objects.create( title="TestBook1", pages=20, author=self.fakeauthor) self.testbook2 = Book.objects.create( title="TestBook2", author=self.fakeauthor, pages=10) self.nontestbook = Book.objects.create( title="nonTestBook", author=self.fakeauthor2) self.objects_list = [self.testbook1, self.testbook2, self.nontestbook] def test_no_filters(self): records = Endpoint.get_widget("book-table", ()).get_data("{}")['aaData'] self.assertEqual(len(records), 3) def test_single_text_filter(self): client_params = '{"filter_query":"pages__gt=19"}' records = Endpoint.get_widget("book-table", ()).get_data(client_params)['aaData'] self.assertEqual(len(records), 1) def test_second_page(self): PAGE_SIZE = 10 for _ in range(PAGE_SIZE): Book.objects.create(title="Book", author=self.fakeauthor) client_params = json.dumps({ 'datatables_params': {"iDisplayStart": PAGE_SIZE, "iDisplayLength": PAGE_SIZE, "sEcho": 1}}) records = Endpoint.get_widget("book-table", ()).get_data(client_params)['aaData'] self.assertEqual(len(records), 3) def test_sort_by_lambda_column(self): column = Endpoint.get_widget("author-table", ()).columns.index(('Age', get_age)) client_params = _client_params_ordering(column, "desc") records = Endpoint.get_widget("author-table", ()).get_data(client_params)['aaData'] self.assertLess(_get_date(records[0]['1']), _get_date(records[1]['1'])) client_params = _client_params_ordering(column, "asc") records = Endpoint.get_widget("author-table", ()).get_data(client_params)['aaData'] self.assertGreater(_get_date(records[0]['1']), _get_date(records[1]['1'])) class TableStrRepresentation(TestCase): def test_str(self): text = smart_text(Endpoint.get_widget("book-table", ())) self.assertIn("Title", text) class FilterFormTestCase(TestCase): def setUp(self): self.author1 = Author.objects.create(name="Author_1") self.author2 = Author.objects.create(name="Author_2") self.testbook1 = Book.objects.create( title="Book_1", pages=20, author=self.author1) self.testbook2 = Book.objects.create( title="Book_2", author=self.author2, pages=10) self.testbook3 = Book.objects.create( title="Book_3", author=self.author2, publication_date=datetime.date(2008, 6, 24)) def test_filter_lt_condition(self): filterform = BookFilter('filter-name') filter_data = {'pages__lt': '15'} filtered = filterform.filter_queryset(filter_data, Book.objects.all()) self.assertQuerysetEqual( filtered, ['Book_2'], lambda b: b.title, False) def test_filter_no_results(self): filterform = BookFilter('filter-name') filter_data = {'pages__gt': '25'} filtered = filterform.filter_queryset(filter_data, Book.objects.all()) self.assertQuerysetEqual(filtered, [], lambda b: b.title, False) def test_empty_filter(self): filterform = BookFilter('filter-name') filter_data = {} filtered = filterform.filter_queryset(filter_data, Book.objects.all()) self.assertQuerysetEqual( filtered, ['Book_1', 'Book_2', 'Book_3'], lambda b: b.title, False) def test_excluding_filter(self): filterform = BookFilter('filter-name') filter_data = { 'pages__gt': 15, 'pages__lt': 14, } filtered = filterform.filter_queryset(filter_data, Book.objects.all()) self.assertQuerysetEqual(filtered, [], lambda b: b.title, False) def test_filter_exact(self): filterform = BookFilter('filter-name') filter_data = { 'publication_date__year': 2008, } filtered = filterform.filter_queryset(filter_data, Book.objects.all()) self.assertQuerysetEqual( filtered, ['Book_3'], lambda b: b.title, False) @mock.patch("django_data_explorer.FilterForm.filter_queryset") def test_filter_queryset_query_string(self, filter_queryset_mock): filterform = BookFilter('filter-name') filterform.filter_queryset_query_string("a=b&c=d", "queryset here") parsed = filter_queryset_mock.call_args[0][0] self.assertEqual(parsed, {"a": ["b"], "c": ["d"]}) class FilterFormParseToLabelTests(TestCase): def test_filter_label_string(self): filterform = BookFilter('filter-name') self.assertEqual(filterform.form.fields['pages__lt'].label, 'Pages less than') self.assertEqual(filterform.form.fields['pages__gt'].label, 'Pages greater than') self.assertEqual(filterform.form.fields['publication_date__gte'].label, 'Publication date greater than or equal to') self.assertEqual(filterform.form.fields['publication_date__lte'].label, 'Publication date less than or equal to') self.assertEqual(filterform.form.fields['publication_date__year__gt'].label, 'Publication date year greater than') self.assertEqual(filterform.form.fields['publication_date__year'].label, 'Publication date year') def test_parse_to_label(self): label = django_data_explorer.filterform.parse_to_label("lat__gte") self.assertEqual(label, "Lat greater than or equal to") def test_parse_to_label_empty(self): """Tests if the function doesn't crash on empty label.""" label = django_data_explorer.filterform.parse_to_label("") self.assertEqual(label, "") def test_parse_to_label_underscore(self): label = django_data_explorer.filterform.parse_to_label("a_b") self.assertEqual(label, "A b") def test_parse_to_label_trailing_underscore(self): label = django_data_explorer.filterform.parse_to_label("a___b") self.assertEqual(label, "A b") class MapGetDataTestCase(TestCase): def setUp(self): self.city1 = City.objects.create(name="City_1", citizens_number=3, latitude=30.0, longitude=20.0) self.city2 = City.objects.create(name="City_2", citizens_number=33, latitude=33.0, longitude=21.0) self.city3 = City.objects.create(name="City_3", citizens_number=333, latitude=35.0, longitude=22.0) def test_get_all_data(self): city_map = Endpoint.get_widget("city-map", ()) client_params = {} data = city_map.get_data(client_params) self.assertEqual(data, [ {'description': "City City_1 with latitude 30 " + "and longitude 20", 'latitude': 30.0, 'longitude': 20.0}, {'description': "City City_2 with latitude 33 " + "and longitude 21", 'latitude': 33.0, 'longitude': 21.0}, {'description': "City City_3 with latitude 35 " + "and longitude 22", 'latitude': 35.0, 'longitude': 22.0}]) def test_get_some_data(self): city_map = Endpoint.get_widget("city-map", ()) client_params = 'citizens_number__gt=30&latitude__lt=34.0' data = city_map.get_data(client_params) self.assertEqual(data[0]['latitude'], 33.0) def test_get_no_data(self): city_map = Endpoint.get_widget("city-map", ()) client_params = 'citizens_number__gt=30&latitude__lt=20.0' data = city_map.get_data(client_params) self.assertEqual(data, [])