def handle(self, *args, **options):
source = f.SourceFactory.create()
attributes = f.AttributeFactory.create_batch(10)
roles = [
Role.objects.get_or_create(name='teacher'),
Role.objects.get_or_create(name='student')
]
for m in xrange(10):
muni = f.MunicipalityFactory.create(data_source=source)
schools = f.SchoolFactory.create_batch(10,
municipality=muni,
data_source=source)
for u in xrange(1000):
user = f.UserFactory()
role = random.choice(roles)
for attr in attributes:
f.UserAttributeFactory.create(user=user,
attribute=attr,
data_source=source)
for g in xrange(random.randint(1, 10)):
f.AttendanceFactory(user=user,
school=random.choice(schools),
role=role,
data_source=source)
sys.stdout.write('.')
sys.stdout.flush()
sys.stdout.write('\n')
def test_user(self):
o = f.UserFactory()
self.assertTrue(o)
self.assertTrue(unicode(o) == o.username)
u = models.User.objects.create(username='******')
u.email = '*****@*****.**'
u.save()
self.assertTrue(u.email)
def test_timestamp_filter(self):
obj = authdata.views.UserFilter()
date = 0
for _ in xrange(10):
f.UserFactory()
qs = authdata.models.User.objects.all()
qs_out = obj.timestamp_filter(qs, date)
self.assertTrue(qs_out is not None)
def test_attribute_data_no_request(self):
obj = serializers.UserSerializer()
self.assertTrue(obj)
user_obj = f.UserFactory()
user_attribute_obj = f.UserAttributeFactory(user=user_obj)
data = obj.attribute_data(user_obj)
self.assertEqual(len(data), 1)
d = {}
d['name'] = user_attribute_obj.attribute.name
d['value'] = user_attribute_obj.value
self.assertEqual(data[0], d)
def test_save(self):
user_obj = f.UserFactory(username='******')
data = {'name': 'municipality',
'municipality_id': 'foo'}
obj = serializers.MunicipalitySerializer(data=data)
self.assertTrue(obj.is_valid(), 'Not Valid')
obj.context['request'] = Mock()
obj.context['request'].user = Mock()
obj.context['request'].user.username = user_obj.username
municipality = obj.save()
self.assertEqual(municipality.data_source.name, user_obj.username)
def test_save(self):
user_obj = f.UserFactory(username='******')
muni_obj = f.MunicipalityFactory()
data = {'name': 'school',
'school_id': 'foo',
'municipality': muni_obj.pk}
obj = serializers.SchoolSerializer(data=data)
self.assertTrue(obj.is_valid(), 'Not Valid')
obj.context['request'] = Mock()
obj.context['request'].user = Mock()
obj.context['request'].user.username = user_obj.username
school = obj.save()
self.assertEqual(school.data_source.name, user_obj.username)
def test_role_data(self):
obj = serializers.QuerySerializer()
self.assertTrue(obj)
user_obj = f.UserFactory()
attendance_obj = f.AttendanceFactory(user=user_obj)
data = obj.role_data(user_obj)
self.assertEqual(len(data), 1)
d = {}
d['school'] = attendance_obj.school.school_id
d['group'] = attendance_obj.group
d['role'] = attendance_obj.role.name
d['municipality'] = attendance_obj.school.municipality.municipality_id
self.assertEqual(data[0], d)
def test_save(self):
user_obj = f.UserFactory(username='******')
attribute_obj = f.AttributeFactory()
data = {'attribute': attribute_obj.name,
'user': user_obj.username}
obj = serializers.UserAttributeSerializer(data=data)
self.assertTrue(obj.is_valid(), 'Not Valid')
obj.context['request'] = Mock()
obj.context['request'].user = Mock()
obj.context['request'].user.username = user_obj.username
user_attribute_obj = obj.save()
self.assertEqual(user_attribute_obj.attribute, attribute_obj)
self.assertEqual(user_attribute_obj.user, user_obj)
self.assertEqual(user_attribute_obj.data_source.name, user_obj.username)
def test_attribute_data_with_request(self):
obj = serializers.UserSerializer()
self.assertTrue(obj)
obj.context['request'] = Mock()
obj.context['request'].user = Mock()
obj.context['request'].user.username = u'foo'
user_obj = f.UserFactory()
user_attribute_obj = f.UserAttributeFactory(user=user_obj, data_source__name=u'foo')
data = obj.attribute_data(user_obj)
self.assertEqual(len(data), 1)
d = {}
d['name'] = user_attribute_obj.attribute.name
d['value'] = user_attribute_obj.value
self.assertEqual(data[0], d)
def test_save(self):
user_obj = f.UserFactory(username='******')
school_obj = f.SchoolFactory()
role_obj = f.RoleFactory()
data = {'user': user_obj.pk,
'school': school_obj.pk,
'role': role_obj.pk,
'group': 'Group1',
}
obj = serializers.AttendanceSerializer(data=data)
self.assertTrue(obj.is_valid(), 'Not Valid')
obj.context['request'] = Mock()
obj.context['request'].user = Mock()
obj.context['request'].user.username = user_obj.username
attendance_obj = obj.save()
self.assertEqual(attendance_obj.data_source.name, user_obj.username)
def test_clean_password(self):
user_obj = f.UserFactory(username='******', password='******')
obj = authdata.forms.UserChangeForm({'password': '******'}, instance=user_obj)
password = obj.clean_password()
self.assertEqual(password, 'originalpass1')
def test_clean_username_error_duplicate(self):
f.UserFactory(username='******')
obj = authdata.forms.UserCreationForm({'username': '******'})
self.assertFalse(obj.is_valid())
self.assertEqual(obj.errors, {'username': [u'A user with that username already exists.']})
def test_user_external_fields(self):
u = f.UserFactory()
self.assertFalse(u.external_source)
self.assertFalse(u.external_id)
def test_user_last_login(self):
u = f.UserFactory()
self.assertTrue(u.last_login)
def test_user_timestamps(self):
u = f.UserFactory()
self.assertTrue(u.created)
self.assertTrue(u.modified)
