got session id working.
This commit is contained in:
@ -1,12 +1,16 @@
|
||||
"""Base for MoreThanTest test cases."""
|
||||
|
||||
from aiohttp import ClientSession, CookieJar
|
||||
from asyncio import create_subprocess_exec
|
||||
from asyncio import create_subprocess_exec, sleep
|
||||
from pathlib import Path
|
||||
from socket import socket
|
||||
from unittest import IsolatedAsyncioTestCase
|
||||
from aiohttp import ClientSession
|
||||
|
||||
|
||||
LOCALHOST = "127.0.0.1"
|
||||
SESSION_KEY = "sessionid"
|
||||
HOST = "example.com"
|
||||
|
||||
|
||||
class Server:
|
||||
"""Setup and run servers."""
|
||||
@ -28,9 +32,9 @@ class Server:
|
||||
if get_addr:
|
||||
addr = item
|
||||
get_addr = False
|
||||
if item == "-a" or item == "--address":
|
||||
if item in ("-a", "--address"):
|
||||
get_addr = True
|
||||
if item == "-p" or item == "--port":
|
||||
if item in ("-p", "--port"):
|
||||
get_port = True
|
||||
else:
|
||||
self.cmd = [app]
|
||||
@ -40,6 +44,7 @@ class Server:
|
||||
async def create(self):
|
||||
"""Cerate the server"""
|
||||
self.server = await create_subprocess_exec(*self.cmd)
|
||||
await sleep(1)
|
||||
|
||||
async def destroy(self):
|
||||
"""destroy servers"""
|
||||
@ -53,8 +58,8 @@ class MTTClusterTC(IsolatedAsyncioTestCase):
|
||||
async def asyncSetUp(self):
|
||||
"""Test setup"""
|
||||
self.servers = []
|
||||
self.jar = CookieJar(unsafe=True)
|
||||
self.session = ClientSession(cookie_jar=self.jar)
|
||||
self.cookies = {}
|
||||
self.session = ClientSession()
|
||||
|
||||
async def asyncTearDown(self):
|
||||
"""Test tear down."""
|
||||
@ -78,11 +83,16 @@ class MTTClusterTC(IsolatedAsyncioTestCase):
|
||||
self.servers.append(server)
|
||||
|
||||
async def create_server(self):
|
||||
"""Create a server on a random port."""
|
||||
port = await self.get_port()
|
||||
await self.create_server_with_flags("-p", str(port))
|
||||
|
||||
async def run_tests(self, uri, func):
|
||||
"""Run the tests on each server."""
|
||||
for server in self.servers:
|
||||
async with self.session.get(f"{server.host}{uri}") as response:
|
||||
async with self.session.get(
|
||||
f"{server.host}{uri}", cookies=self.cookies
|
||||
) as response:
|
||||
if SESSION_KEY in response.cookies:
|
||||
self.cookies[SESSION_KEY] = response.cookies[SESSION_KEY].value
|
||||
func(response)
|
||||
|
Reference in New Issue
Block a user