lostplaces-backend/lostplaces/lostplaces_app/tests/models/test_place_image_model.py

import datetime
import os
import shutil
from unittest import mock

from django.test import TestCase
from django.db import models
from django.core.files import File
from django.conf import settings
from django.contrib.auth.models import User

from lostplaces_app.models import PlaceImage, Place
from lostplaces_app.tests.models import SubmittableTestCase

from easy_thumbnails.fields import ThumbnailerImageField

class TestPlaceImage(SubmittableTestCase, TestCase):
    model = PlaceImage

    @classmethod
    def setUpTestData(cls):
        user = User.objects.create_user(
            username='testpeter',
            password='Develop123'
        )
        
        place = Place.objects.create(
            name='Im a place',
            submitted_when=datetime.datetime.now(),
            submitted_by=User.objects.get(username='testpeter').explorer,
            location='Testtown',
            latitude=50.5,
            longitude=7.0,
            description='This is just a test, do not worry'
        )
        place.tags.add('I a tag', 'testlocation')
        place.save()

        if not os.path.isdir(settings.MEDIA_ROOT):
            os.mkdir(settings.MEDIA_ROOT)

        current_dir = os.path.dirname(os.path.abspath(__file__))
        if not os.path.isfile(os.path.join(settings.MEDIA_ROOT, 'im_a_image_copy.jpeg')):
            shutil.copyfile(
                os.path.join(current_dir, 'im_a_image.jpeg'),
                os.path.join(settings.MEDIA_ROOT, 'im_a_image_copy.jpeg')
            )
            
            shutil.copyfile(
                os.path.join(current_dir, 'im_a_image.jpeg'),
                os.path.join(settings.MEDIA_ROOT, 'im_a_image_changed.jpeg')
            )
            
        PlaceImage.objects.create(
            description='Im a description',
            filename=os.path.join(settings.MEDIA_ROOT, 'im_a_image_copy.jpeg'),
            place=place,
            submitted_when=datetime.datetime.now(),
            submitted_by=user.explorer
        )

    def test_description(self):
        self._test_field('description', models.TextField)

    def test_filename(self):
        self._test_field('filename',ThumbnailerImageField)

    def test_place(self):
        field = self._test_field('place', models.ForeignKey)
        self.assertEqual(field.remote_field.on_delete, models.CASCADE,
            msg='%s.%s deleting of %s should be cascadinf' % (
                self.model_name,
                'place',
                self.model_name
            )
        )
        self.assertEqual(field.remote_field.related_name, 'images',
            msg='%s.%s related name should be images' % (
                self.model_name, 
                'place'                
            )
        )

    def test_str(self):
        place_image = self.object
        self.assertTrue(place_image.place.name.lower() in str(place_image).lower(),
            msg='Expecting %s.__str__ to contain  the name of the place' % (
                self.model_name
            )
        )
        
    def test_change_filename(self):
        path = self.object.filename.path
        self.object.filename = os.path.join(settings.MEDIA_ROOT, 'im_a_image_changed.jpeg')
        self.object.save()
        self.assertFalse(
            os.path.isfile(path),
            msg='Expecting the old file of an place_image to be deleteed when an place_image file is changed'
        )
        
    def test_deletion(self):
        path = self.object.filename.path
        self.object.delete()
        self.assertFalse(
            os.path.isfile(path),
            msg='Expecting the file of an place_image to be deleteed when an place_image is deleted'
        )