OpenSlides/tests/motion/test_csv_import.py

102 lines
5.1 KiB
Python
Raw Normal View History

2013-05-08 18:07:09 +02:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Tests for openslides.motion.models
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
:copyright: 20112013 by OpenSlides team, see AUTHORS.
:license: GNU GPL, see LICENSE for more details.
"""
2013-05-12 00:47:49 +02:00
import StringIO
2013-05-08 18:07:09 +02:00
import os
from django.test.client import Client
from openslides.config.api import config
2013-05-12 00:47:49 +02:00
from openslides.motion.models import Motion, Category
2013-05-08 18:07:09 +02:00
from openslides.motion.csv_import import import_motions
2013-05-12 00:47:49 +02:00
from openslides.participant.models import User
from openslides.utils.test import TestCase
2013-05-08 18:07:09 +02:00
class CSVImport(TestCase):
def setUp(self):
# Admin
2013-05-12 00:47:49 +02:00
self.admin = User.objects.create_superuser('Admin_ieY0Eereimeimeizuosh', 'admin@admin.admin', 'eHiK1aiRahxaix0Iequ2')
2013-05-08 18:07:09 +02:00
self.admin_client = Client()
2013-05-12 00:47:49 +02:00
self.admin_client.login(username='Admin_ieY0Eereimeimeizuosh', password='eHiK1aiRahxaix0Iequ2')
# Normal user
self.normal_user = User.objects.create_user('User_CiuNgo4giqueeChie5oi', 'user@user.user', 'eihi1Eequaek4eagaiKu')
self.normal_client = Client()
self.normal_client.login(username='User_CiuNgo4giqueeChie5oi', password='eihi1Eequaek4eagaiKu')
# Category
self.category1 = Category.objects.create(name='Satzung', prefix='S')
self.category2 = Category.objects.create(name='Resolution', prefix='R1')
self.category3 = Category.objects.create(name='Resolution', prefix='R2')
def test_example_file_de(self):
special_user = User.objects.create_user(username='Fritz_Fleiner',
email='fritz@fritz.fritz',
password='iegheeChaje7guthie4a',
first_name='Fritz',
last_name='Fleiner')
for i in range(2):
username = 'Hilde_Müller_%d' % i
User.objects.create_user(username=username,
email='hilde@hilde.hilde',
password='default',
first_name='Hilde',
last_name='Müller',
title='Dr.')
2013-05-08 18:07:09 +02:00
csv_dir = os.path.join(os.path.dirname(__file__), '..', '..', 'extras', 'csv-examples')
2013-05-12 00:47:49 +02:00
self.assertEqual(Motion.objects.count(), 0)
2013-05-08 18:07:09 +02:00
with open(csv_dir + '/motions-demo_de.csv') as f:
2013-05-12 00:47:49 +02:00
count_success, error_messages, warning_messages = import_motions(csv_file=f, default_submitter=self.normal_user.person_id)
2013-05-08 18:07:09 +02:00
self.assertEqual(Motion.objects.count(), 4)
2013-05-12 00:47:49 +02:00
self.assertEqual(count_success, 4)
motion1 = Motion.objects.get(pk=1)
self.assertEqual(motion1.identifier, '1')
self.assertEqual(motion1.title, u'Entlastung des Vorstandes')
self.assertEqual(motion1.text, u'Die Versammlung möge beschließen, den Vorstand für seine letzte Legislaturperiode zu entlasten.')
self.assertEqual(motion1.reason, u'Bericht erfolgt mündlich.')
self.assertEqual(len(motion1.submitter.all()), 1)
self.assertEqual(motion1.submitter.all()[0].person, self.normal_user)
self.assertTrue(motion1.category is None)
self.assertTrue('Line 2 in import file: Default submitter is used.' in warning_messages)
self.assertTrue('Line 2 in import file: Category not found.' in error_messages)
motion2 = Motion.objects.get(pk=2)
self.assertEqual(motion2.identifier, 'S 2')
self.assertEqual(motion2.title, u'Satzungsänderung § 2 Abs. 3')
self.assertEqual(motion2.text, u'Die Versammlung möge beschließen, die Satzung in § 2 Abs. 3 wie folgt zu ändern:\n\n'
u'Es wird nach dem Wort "Zweck" der Satz "..." eingefügt.')
self.assertEqual(motion2.reason, u'Die Änderung der Satzung ist aufgrund der letzten Erfahrungen eine sinnvolle Maßnahme, weil ...')
self.assertEqual(len(motion2.submitter.all()), 1)
self.assertEqual(motion2.submitter.all()[0].person, special_user)
self.assertEqual(motion2.category, self.category1)
self.assertTrue('Line 5 in import file: Multiple persons found.' in error_messages)
self.assertTrue('Line 5 in import file: Multiple categories found.' in error_messages)
2013-05-08 18:07:09 +02:00
2013-05-12 00:47:49 +02:00
def test_malformed_file(self):
csv_file = StringIO.StringIO()
csv_file.write('Header\nMalformed data,\n,Title,Text,,,\n')
count_success, error_messages, warning_messages = import_motions(csv_file=csv_file, default_submitter=self.normal_user.person_id)
self.assertEqual(count_success, 1)
self.assertTrue('Ignoring malformed line 2 in import file.' in error_messages)
2013-05-08 18:07:09 +02:00
2013-05-12 00:47:49 +02:00
def test_wrong_encoding(self):
csv_file = StringIO.StringIO()
text = u'Müller'.encode('iso-8859-15')
csv_file.write(text)
csv_file.seek(0)
count_success, error_messages, warning_messages = import_motions(csv_file=csv_file, default_submitter=self.normal_user.person_id)
self.assertEqual(count_success, 0)
self.assertTrue('Encoding error in import file. Ensure using UTF-8.' in error_messages)