Revision 47abaf1f cloudcms/tests/__init__.py
| b/cloudcms/tests/__init__.py | ||
|---|---|---|
| 35 | 35 |
import os |
| 36 | 36 |
import tempfile |
| 37 | 37 |
|
| 38 |
from django.test import TestCase, TransactionTestCase
|
|
| 38 |
from django.test import TransactionTestCase |
|
| 39 | 39 |
from django.contrib.auth.models import User |
| 40 | 40 |
from django.core.files.uploadedfile import SimpleUploadedFile |
| 41 | 41 |
|
| 42 |
from feincms.module.medialibrary.models import Category as MediaCategory, \ |
|
| 43 |
MediaFile |
|
| 42 |
from feincms.module.medialibrary.models import MediaFile |
|
| 44 | 43 |
|
| 45 | 44 |
from cloudcmsguide.models import UserGuideEntry |
| 46 |
from cloudcmsfaq.models import Category, Question |
|
| 47 |
from cloudcms.rstutils import generate_rst_contents_from_dir |
|
| 48 |
from cloudcms.models import Service, ServiceTranslation |
|
| 45 |
from cloudcmsfaq.models import Question |
|
| 49 | 46 |
from cloudcms.forms import RstZipImportForm |
| 50 | 47 |
|
| 51 |
from synnefo.lib.log import initialize_logging |
|
| 52 |
|
|
| 53 | 48 |
from django.conf import settings |
| 54 | 49 |
|
| 55 |
initialize_logging() |
|
| 56 | 50 |
|
| 57 | 51 |
TESTS_BASE = os.path.abspath(os.path.dirname(__file__)) |
| 58 | 52 |
|
| ... | ... | |
| 65 | 59 |
|
| 66 | 60 |
settings.MEDIA_ROOT = tempfile.mkdtemp("cms-tests")
|
| 67 | 61 |
|
| 62 |
|
|
| 68 | 63 |
class TestSphinxImport(TransactionTestCase): |
| 69 | 64 |
|
| 70 | 65 |
fixtures = ['cloudcms_media_categories', 'cloudcms_default_services'] |
| 71 | 66 |
|
| 72 | 67 |
def setUp(self): |
| 73 |
c = Category.objects.create()
|
|
| 74 |
self.user = User.objects.create(username="admin", email="admin@admin.com")
|
|
| 68 |
self.user = User.objects.create(username="admin",
|
|
| 69 |
email="admin@admin.com")
|
|
| 75 | 70 |
self.user.is_superuser = True |
| 76 | 71 |
self.user.set_password("test")
|
| 77 | 72 |
self.user.save() |
| 78 | 73 |
|
| 79 | 74 |
def test_from_dir1(self): |
| 80 | 75 |
zipfile = SimpleUploadedFile('file.zip', file(TEST_VALID_ZIP).read())
|
| 81 |
form = RstZipImportForm({'import_categories': ['faqs', 'guide']}, files={'import_file': zipfile})
|
|
| 76 |
form = RstZipImportForm({
|
|
| 77 |
'import_categories': ['faqs', 'guide'] |
|
| 78 |
}, files={'import_file': zipfile})
|
|
| 82 | 79 |
form.is_valid() |
| 83 | 80 |
success, log = form.save(self.user, TEST_DIR1) |
| 84 | 81 |
|
| ... | ... | |
| 89 | 86 |
|
| 90 | 87 |
def test_form(self): |
| 91 | 88 |
zipfile = SimpleUploadedFile('file.zip', file(TEST_INVALID_ZIP).read())
|
| 92 |
form = RstZipImportForm({'import_categories': ['faqs', 'guide']}, files={'import_file': zipfile})
|
|
| 89 |
form = RstZipImportForm({
|
|
| 90 |
'import_categories': ['faqs', 'guide']}, |
|
| 91 |
files={'import_file': zipfile})
|
|
| 93 | 92 |
self.assertEqual(form.is_valid(), False) |
| 94 | 93 |
|
| 95 | 94 |
zipfile = SimpleUploadedFile('file.zip', file(TEST_VALID_ZIP).read())
|
| 96 |
form = RstZipImportForm({'import_categories': ['faqs', 'guide']}, files={'import_file': zipfile})
|
|
| 95 |
form = RstZipImportForm({
|
|
| 96 |
'import_categories': ['faqs', 'guide']}, |
|
| 97 |
files={'import_file': zipfile})
|
|
| 97 | 98 |
self.assertEqual(form.is_valid(), True) |
| 98 | 99 |
|
| 99 | 100 |
form.save(self.user) |
| 100 |
self.assertEqual(MediaFile.objects.all().count(), 5)
|
|
| 101 |
self.assertEqual(MediaFile.objects.all().count(), 3)
|
|
| 101 | 102 |
self.assertEqual(Question.objects.all().count(), 4) |
| 102 | 103 |
self.assertEqual(UserGuideEntry.objects.all().count(), 3) |
| 103 |
|
|
Also available in: Unified diff