|
1 | 1 | # -*- coding: utf-8 -*- |
| 2 | +from django.conf import settings |
2 | 3 | from django.test import TestCase |
3 | 4 |
|
4 | | -from filer.models import File as FilerFile |
5 | | -from filer.models import Folder as FilerFolder |
| 5 | +from djangocms_file.models import LINK_TARGET, File, Folder, get_templates |
6 | 6 |
|
7 | | -from djangocms_file.models import File |
| 7 | +from .helpers import get_filer_file, get_filer_folder |
8 | 8 |
|
9 | 9 |
|
10 | | -class FileTestCase(TestCase): |
| 10 | +class CommonModelTestCase(TestCase): |
| 11 | + |
| 12 | + def test_settings(self): |
| 13 | + self.assertEqual(get_templates(), [('default', 'Default')]) |
| 14 | + settings.DJANGOCMS_FILE_TEMPLATES = [('feature', 'Feature')] |
| 15 | + self.assertEqual(get_templates(), [('default', 'Default'), ('feature', 'Feature')]) |
| 16 | + |
| 17 | + |
| 18 | +class FileModelTestCase(TestCase): |
11 | 19 |
|
12 | 20 | def setUp(self): |
13 | | - filer_file = FilerFile.objects.create( |
14 | | - file='test.jpg', |
15 | | - ) |
16 | | - File.objects.create( |
17 | | - file_src=filer_file, |
18 | | - file_name='test', |
19 | | - ) |
| 21 | + self.file = get_filer_file() |
| 22 | + |
| 23 | + def tearDown(self): |
| 24 | + if self.file: |
| 25 | + self.file.delete() |
| 26 | + del self.file |
| 27 | + with self.assertRaises(AttributeError): |
| 28 | + print(self.file) |
| 29 | + File.objects.filter(pk=1).delete() |
| 30 | + self.assertEqual(File.objects.all().count(), 0) |
20 | 31 |
|
21 | 32 | def test_file_instance(self): |
22 | | - """File instance has been created""" |
23 | | - test_file = File.objects.get(file_name='test') |
24 | | - self.assertEqual(test_file.file_name, 'test') |
| 33 | + File.objects.create( |
| 34 | + template="default", |
| 35 | + file_src=self.file, |
| 36 | + file_name="some file", |
| 37 | + link_target=LINK_TARGET[0][0], |
| 38 | + link_title="some link title", |
| 39 | + show_file_size=True, |
| 40 | + attributes="{'data-type', 'file'}", |
| 41 | + ) |
| 42 | + instance = File.objects.all() |
| 43 | + self.assertEqual(len(instance), 1) |
| 44 | + instance = File.objects.get(pk=1) |
| 45 | + self.assertEqual(instance.template, "default") |
| 46 | + self.assertEqual(instance.file_src, self.file) |
| 47 | + self.assertEqual(instance.file_name, "some file") |
| 48 | + self.assertEqual(instance.link_target, "_self") |
| 49 | + self.assertEqual(instance.show_file_size, True) |
| 50 | + self.assertEqual(instance.attributes, "{'data-type', 'file'}") |
| 51 | + # old copy relation and test methods |
| 52 | + instance.copy_relations(instance) |
| 53 | + self.assertEqual(instance.__str__(), "test_file.pdf") |
| 54 | + self.assertEqual(instance.get_short_description(), "some file") |
| 55 | + instance.file_name = None |
| 56 | + self.assertEqual(instance.get_short_description(), "test_file.pdf") |
| 57 | + instance.file_src = None |
| 58 | + self.assertEqual(instance.__str__(), "1") |
| 59 | + self.assertEqual(instance.get_short_description(), "<file is missing>") |
25 | 60 |
|
26 | 61 |
|
27 | | -class FolderTestCase(TestCase): |
| 62 | +class FolderModelTestCase(TestCase): |
28 | 63 |
|
29 | 64 | def setUp(self): |
30 | | - FilerFolder.objects.create( |
31 | | - name='test', |
32 | | - ) |
| 65 | + self.folder = get_filer_folder() |
| 66 | + self.file = get_filer_file() |
| 67 | + |
| 68 | + def tearDown(self): |
| 69 | + if self.file: |
| 70 | + self.file.delete() |
| 71 | + del self.file |
| 72 | + with self.assertRaises(AttributeError): |
| 73 | + print(self.file) |
| 74 | + Folder.objects.filter(pk=1).delete() |
| 75 | + self.assertEqual(File.objects.all().count(), 0) |
33 | 76 |
|
34 | 77 | def test_folder_instance(self): |
35 | | - """Folder instance has been created""" |
36 | | - test_folder = FilerFolder.objects.get(name='test') |
37 | | - self.assertEqual(test_folder.name, 'test') |
| 78 | + Folder.objects.create( |
| 79 | + template="default", |
| 80 | + folder_src=self.folder, |
| 81 | + link_target=LINK_TARGET[0][0], |
| 82 | + show_file_size=True, |
| 83 | + attributes="{'data-type', 'folder'}", |
| 84 | + ) |
| 85 | + instance = Folder.objects.all() |
| 86 | + self.assertEqual(len(instance), 1) |
| 87 | + instance = Folder.objects.get(pk=1) |
| 88 | + self.assertEqual(instance.template, "default") |
| 89 | + self.assertEqual(instance.folder_src, self.folder) |
| 90 | + self.assertEqual(instance.link_target, "_self") |
| 91 | + self.assertEqual(instance.show_file_size, True) |
| 92 | + self.assertEqual(instance.attributes, "{'data-type', 'folder'}") |
| 93 | + # old copy relation and test methods |
| 94 | + instance.copy_relations(instance) |
| 95 | + self.assertEqual(instance.__str__(), "test_folder") |
| 96 | + self.assertEqual(instance.get_short_description(), "test_folder") |
| 97 | + instance.folder_src = None |
| 98 | + self.assertEqual(instance.__str__(), "1") |
| 99 | + self.assertEqual(instance.get_short_description(), "<folder is missing>") |
| 100 | + # get files |
| 101 | + self.assertEqual(instance.get_files(), []) |
| 102 | + self.file = get_filer_file( |
| 103 | + folder=self.folder |
| 104 | + ) |
| 105 | + instance = Folder.objects.create( |
| 106 | + folder_src=self.folder, |
| 107 | + ) |
| 108 | + self.assertEqual(instance.get_files(), [self.file]) |
0 commit comments