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)