generated from jCloud/repository-template
Move make_request function to a separate utils module
This commit is contained in:
+4
-30
@@ -13,38 +13,10 @@
|
||||
# limitations under the License.
|
||||
|
||||
from src.jcloud_deployment_server.integrations.gitea.middlewares.signature import GiteaSignatureMiddleware
|
||||
from starlette.requests import Request
|
||||
from fastapi import HTTPException
|
||||
from tests.utils.make_request import make_request, call_next
|
||||
import pytest
|
||||
import hashlib
|
||||
import hmac
|
||||
import os
|
||||
|
||||
class DummyResponse:
|
||||
def __init__(self):
|
||||
self.called = True
|
||||
|
||||
async def call_next(request):
|
||||
return DummyResponse()
|
||||
|
||||
def make_request(body: bytes, headers = None) -> Request:
|
||||
scope = {
|
||||
'type': 'http',
|
||||
'method': 'POST',
|
||||
'path': '/gitea/webhook',
|
||||
'headers': [
|
||||
(k.lower().encode(), v.encode())
|
||||
for k, v in (headers or {}).items()
|
||||
]
|
||||
}
|
||||
|
||||
async def receive():
|
||||
return {
|
||||
'type': 'http.request',
|
||||
'body': body
|
||||
}
|
||||
|
||||
return Request(scope, receive)
|
||||
|
||||
def make_signature(body: bytes, secret: bytes) -> str:
|
||||
signature = hmac.new(
|
||||
@@ -98,6 +70,7 @@ async def test_GiteaSignatureMiddleware_valid_signature(body, secret):
|
||||
|
||||
req = make_request(
|
||||
body,
|
||||
'/gitea/webhook',
|
||||
headers
|
||||
)
|
||||
|
||||
@@ -136,7 +109,8 @@ async def test_GiteaSignatureMiddleware_invalid_signature(signature, body, secre
|
||||
|
||||
req = make_request(
|
||||
body,
|
||||
headers
|
||||
'/gitea/webhook',
|
||||
headers,
|
||||
)
|
||||
|
||||
res = await middleware.dispatch(req, call_next)
|
||||
|
||||
@@ -0,0 +1,60 @@
|
||||
# Copyright 2026 jCloud
|
||||
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from starlette.requests import Request
|
||||
|
||||
__all__ = [
|
||||
'call_next',
|
||||
'make_request'
|
||||
]
|
||||
|
||||
class _DummyResponse:
|
||||
def __init__(self):
|
||||
self.called = True
|
||||
|
||||
async def call_next(request):
|
||||
return _DummyResponse()
|
||||
|
||||
def make_request(body: bytes, path: str, headers: dict = None) -> Request:
|
||||
'''
|
||||
Makes a request.
|
||||
|
||||
:param body: The request body.
|
||||
:type body: bytes
|
||||
:param path: The path.
|
||||
:type path: str
|
||||
:param headers: The headers.
|
||||
:type headers: dict
|
||||
|
||||
:return: The request.
|
||||
:rtype: Request
|
||||
'''
|
||||
|
||||
scope = {
|
||||
'type': 'http',
|
||||
'method': 'POST',
|
||||
'path': path,
|
||||
'headers': [
|
||||
(k.lower().encode(), v.encode())
|
||||
for k, v in (headers or {}).items()
|
||||
]
|
||||
}
|
||||
|
||||
async def receive():
|
||||
return {
|
||||
'type': 'http.request',
|
||||
'body': body
|
||||
}
|
||||
|
||||
return Request(scope, receive)
|
||||
Reference in New Issue
Block a user