trustchain-oc1-orchestral
/
authentik
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

stages/prompt: fix tests for file field 

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2022-07-27 09:48:11 +02:00
parent de26c65fa0
commit 277df4f04f
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
authentik/stages/prompt/models.py
View File

@ -1,6 +1,6 @@
"""prompt models""" """prompt models"""
from typing import Any, Optional from typing import Any, Optional
from urllib.parse import urlparse from urllib.parse import urlparse, urlunparse
from uuid import uuid4 from uuid import uuid4
from django.db import models from django.db import models
@ -89,7 +89,7 @@ class InlineFileField(CharField):
_mime, _, enc = header.partition(";") _mime, _, enc = header.partition(";")
if enc != "base64": if enc != "base64":
raise ValidationError("Invalid encoding") raise ValidationError("Invalid encoding")
return super().to_internal_value(uri) return super().to_internal_value(urlunparse(uri))
class Prompt(SerializerModel): class Prompt(SerializerModel):

4
authentik/stages/prompt/tests.py
View File

@ -118,11 +118,11 @@ class TestPromptStage(FlowTestCase):
InlineFileField().to_internal_value("data:foo/bar;foo,qwer") InlineFileField().to_internal_value("data:foo/bar;foo,qwer")
self.assertEqual( self.assertEqual(
InlineFileField().to_internal_value("data:mine/type;base64,Zm9v"), InlineFileField().to_internal_value("data:mine/type;base64,Zm9v"),
"foo", "data:mine/type;base64,Zm9v",
) )
self.assertEqual( self.assertEqual(
InlineFileField().to_internal_value("data:mine/type;base64,Zm9vqwer"), InlineFileField().to_internal_value("data:mine/type;base64,Zm9vqwer"),
"Zm9vqwer", "data:mine/type;base64,Zm9vqwer",
) )
def test_render(self): def test_render(self):