Что еще добавить в юнит тест для функции сортировки?

Question

[nurzhannogerbek]

Здравствуйте! Помогите пожалуйста разобраться.

На странице выводится список названий статей. Пользователь может перетаскивать их и сортировать на свое усмотрение. Использовал JQuery UI. Подобное поведение осуществляет данная вьюха (см. ниже).

Вопрос в следующем. Мог бы кто-нибудь помочь составить юнит тест на данную вьюху. C юнит тестированием мало имел дело. На данный момент сделал лишь проверку status_code. Было бы интерестно узнать ваши мнения. Буду благодарен любой помощи!

views.py:

from braces.views import CsrfExemptMixin, JsonRequestResponseMixin

class ArticleSortingView(CsrfExemptMixin, JsonRequestResponseMixin, FormView):
    def post(self, request, *args, **kwargs):
        for pk, idx in self.request_json.items():
            Slide.objects.filter(pk=pk).update(idx=idx)
        return self.render_json_response({'saved': 'OK'})

tests.py:

class ArticleViewTestCase(TestCase):
    def setUp(self):
        self.client = Client()
        self.credentials = {'username': 'user', 'password': 'password'}
        self.user = User.objects.create_user(**self.credentials)

    def test_sorting_article(self):
        self.assertTrue(self.user)

        logged_in = self.client.login(**self.credentials)
        self.assertTrue(logged_in)

        response = self.client.post(reverse("article:article_sorting"), follow=True)
        self.assertEqual(response.status_code, 200)

Comments

[javedimka]

Советую все таки прочитать книжку на которую я давал ссылку в прошлом вопросе, она на два дня максимум, но таких вопросов не будет.

[nurzhannogerbek]

javedimka, как раз в процессе чтения. Cпасибо! =)

Answer 1

[Anton Kuzmichev]

• в setUp создайте несколько объектов Slide
  
• добавьте в вызов

  self.client.post(reverse("article:article_sorting"), follow=True)

  json, который сменит сортировку этьм объектам
  
• добавьте ещё одну проверку, помимо статуса (и, возможно, json ответа), что объекты Slide сменили свою сортировку
  

Comments

[nurzhannogerbek]

Спасибо за совет! Могли бы вы второй пункт более детальнее описать? Возможно на примере было бы понятнее. Не получается все в голове уложить.

[Anton Kuzmichev]

NogerbekNurzhan, вот документация к Client
Например:

import json
...
test_data = {1: idx1, 2: idx2, ...}  # данные в нужном формате для передаче вьюхе
response = self.client.post(
    reverse("article:article_sorting"),
    json.dumps(test_data),
    content_type='application/json',
    follow=True
)

[nurzhannogerbek]

Anton Kuzmichev, Сделал вроде все по вашей схеме, но значение idx (поле для сортировки) не поменялось после post запроса. Что делаю не так?

ERROR:

Traceback (most recent call last):
  File "/home/nurzhan/CA/static_pages/tests.py", line 120, in test_article_sorting
    self.assertEquals(first_article.idx, 2)
AssertionError: 0 != 2

print response.content возращает {"saved": "OK"} и статус 200. Вроде правильно все, но idx не поменялся.

tests.py:

def test_article_sorting(self):
        first_article = Article.objects.create(pk=150, idx=0, head='First')
        second_article = Article.objects.create(pk=160, idx=1, head='Second')
        third_article = Article.objects.create(pk=170, idx=2, head='Third')
        data = {150: 2, 160: 0, 170: 1}
        response = self.client.post(
            reverse("static_page:static_page_sorting"),
            data=json.dumps(data),
            content_type='application/json',
            follow=True
        )
        self.assertEqual(response.content, '{"saved": "OK"}')
        self.assertEqual(response.status_code, 200)
        self.assertEquals(first_article.idx, 2)  <-- ОШИБКА
        self.assertEquals(second_article.idx, 0)
        self.assertEquals(third_article.idx, 1)

[Anton Kuzmichev]

NogerbekNurzhan, вы запросили объекты из БД, а после вашего post-запроса данные же поменялись. Вам нужно перезапросить данные из БД, как-то вроде:

first_article.refresh_from_db()
second_article.refresh_from_db()
third_article.refresh_from_db()
self.assertEquals(first_article.idx, 2)  <-- вот теперь норм должно быть
self.assertEquals(second_article.idx, 0)
self.assertEquals(third_article.idx, 1)

[nurzhannogerbek]

Anton Kuzmichev, Cпасибо большое вам за помощь и подробные объяснения! =)

[nurzhannogerbek]

Anton Kuzmichev, Можно ли вам задать еще один вопрос? По этой же теме. К примеру у модели данных Article есть поле image c типом данных ImageField.

models.py:

class Article(models.Model):
    image = models.ImageField(
        verbose_name='Изображение',
        upload_to='article/images/%Y/%m/%d/',
        blank=False,
    )

    def save(self, *args, **kwargs):
        try:
            article = Article.objects.get(id=self.id)
            if article.image != self.image:
                article.image.delete(save=False)
        except BaseException:
            pass
        super(Article, self).save(*args, **kwargs)

Почему-то после добавления поля ImageField кажется данная строка (first_article.refresh_from_db()) перестала работать. Что можете сказать по этому поводу? Ошибку приложил внизу. Cтранно как-то все это вы так не думаете?

tests.py:

from django.core.files.uploadedfile import SimpleUploadedFile
        ***
        # Создаем изображение
        image = Image.new('RGB', (100, 100))
        file = tempfile.NamedTemporaryFile(suffix='.jpg')
        image.save(file)

        first_article = Article.objects.create(
                pk=150,
                idx=0,
                head='First',
                image=SimpleUploadedFile(
                        name='test.jpg',
                        content=open(file.name, 'rb').read(),
                        content_type='image/jpeg'
                )
        )

        second_article = Article.objects.create(
            pk=160,
            idx=1,
            head='Second',
            image=SimpleUploadedFile(
                name='test.jpg',
                content=open(file.name, 'rb').read(),
                content_type='image/jpeg'
            )
        )

        third_article = Article.objects.create(
            pk=170,
            idx=2,
            head='Third',        
            image=SimpleUploadedFile(
                name='test.jpg',
                content=open(file.name, 'rb').read(),
                content_type='image/jpeg'
            )
        )

        data = {150: 2, 160: 0, 170: 1}
        response = self.client.post(
            reverse("article:article_sorting"),
            data=json.dumps(data),
            content_type='application/json; charset= utf-8',
            follow=True
        )
        self.assertEqual(response.content, '{"saved": "OK"}')
        self.assertEqual(response.status_code, 200)

        first_article.refresh_from_db()
        print str(first_slide.idx)  <-- Возвращает 0, не обновилось. Должно быть 2 после обновления
        self.assertEquals(first_article.idx, 2)
        second_article.refresh_from_db()
        self.assertEquals(second_article.idx, 0)
        third_article.refresh_from_db()
        self.assertEquals(third_article.idx, 1)

ERROR:

Traceback (most recent call last):
  File "/home/nurzhan/CA/article/tests.py", line 176, in test_article_sorting
    self.assertEquals(first_article.idx, 2)
AssertionError: 0 != 2

[Anton Kuzmichev]

NogerbekNurzhan, честно говоря, не могу понять. А данные после выполнения self.client.post в тестовой БД точно меняются?
content_type='application/json; charset= utf-8',
Кодировку, в принципе, можно и не указывать.

[nurzhannogerbek]

Anton Kuzmichev, Сам запутался. post запрос прошел успешно. Вернул статус 200 и вернул правельный response.content. print str(first_slide.idx) возвращает 0, думаю не обновилось либо не прошел refresh_from_db. Должно быть 2 после обновления. Как вы думаете как следует правильно создавать записи с изображениями в юнит тестах? Можно ли как-то загрузить из сервера из папки static?