skillbox/server/core/tests/test_import.py

19 lines
616 B
Python

from django.test import TestCase, Client
from django.core import management
from users.models import User, Role
class ImportUsersTestCase(TestCase):
def test_import(self):
Role.objects.create_default_roles()
management.call_command('import_users', 'Benutzer.sample.csv')
self.assertEqual(User.objects.count(), 8)
michael = User.objects.get(email='michael@dundermifflin.com')
self.assertEqual(michael.first_name, 'Michael')
self.assertEqual(michael.school_classes.count(), 2)
self.assertTrue(michael.has_perm('users.can_manage_school_class_content'))