def test_signature_change(self): """signature module allows for signature change""" test_user = UserModel.objects.create_user('Bob', '*****@*****.**', 'pass123') signatures.set_user_signature(MockRequest(), test_user, '') self.assertEqual(test_user.signature, '') self.assertEqual(test_user.signature_parsed, '') self.assertEqual(test_user.signature_checksum, '') signatures.set_user_signature(MockRequest(), test_user, 'Hello, world!') self.assertEqual(test_user.signature, 'Hello, world!') self.assertEqual(test_user.signature_parsed, '<p>Hello, world!</p>') self.assertTrue(signatures.is_user_signature_valid(test_user)) test_user.signature_parsed = '<p>Injected evil HTML!</p>' self.assertFalse(signatures.is_user_signature_valid(test_user))
def test_signature_change(self): """signature module allows for signature change""" User = get_user_model() test_user = User.objects.create_user('Bob', '*****@*****.**', 'pass123') signatures.set_user_signature(test_user, '') self.assertEqual(test_user.signature, '') self.assertEqual(test_user.signature_parsed, '') self.assertEqual(test_user.signature_checksum, '') signatures.set_user_signature(test_user, 'Hello, world!') self.assertEqual(test_user.signature, 'Hello, world!') self.assertEqual(test_user.signature_parsed, '<p>Hello, world!</p>') self.assertTrue(signatures.is_user_signature_valid(test_user)) test_user.signature_parsed = '<p>Injected evil HTML!</p>' self.assertFalse(signatures.is_user_signature_valid(test_user))
def get_signature_options(user): options = { 'signature': None, 'limit': settings.signature_length_max, } if user.signature: options['signature'] = { 'plain': user.signature, 'html': user.signature_parsed, } if not is_user_signature_valid(user): options['signature']['html'] = None return Response(options)
def has_valid_signature(self): return is_user_signature_valid(self)