Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 21 additions & 2 deletions src/libkernelbot/background_submission_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,9 @@
import logging
from dataclasses import dataclass

from kernelbot.api.api_utils import MultiProgressReporterAPI
from libkernelbot.backend import KernelBackend
from libkernelbot.consts import SubmissionMode
from libkernelbot.report import MultiProgressReporter, RunProgressReporter, RunResultReport
from libkernelbot.submission import ProcessedSubmissionRequest
from libkernelbot.utils import setup_logging

Expand All @@ -28,6 +28,25 @@ class JobItem:
HARD_TIMEOUT_SEC = 60 * 30 # hard timeout 30 mins


# Reporter used in background submission manager, it is a dummy reporter that does nothing
class BackgroundSubmissionManagerReporter(MultiProgressReporter):
def __init__(self):
self.runs = []
async def show(self, title: str):
return
def add_run(self, title: str) ->"BackgroundSubmissionManagerReporterRunProgressReporter":
rep = BackgroundSubmissionManagerReporterRunProgressReporter(title)
self.runs.append(rep)
return rep
def make_message(self):
return
class BackgroundSubmissionManagerReporterRunProgressReporter(RunProgressReporter):
async def _update_message(self):
pass
async def display_report(self, title: str, report: RunResultReport):
pass


class BackgroundSubmissionManager:
"""
This class manages submission in the backeground. It is responsible for
Expand Down Expand Up @@ -211,7 +230,7 @@ async def heartbeat():

hb_task = asyncio.create_task(heartbeat(), name=f"hb-{sub_id}")
try:
reporter = MultiProgressReporterAPI()
reporter = BackgroundSubmissionManagerReporter()
await asyncio.wait_for(
self.backend.submit_full(
item.req, item.mode, reporter, sub_id
Expand Down
22 changes: 0 additions & 22 deletions tests/conftest.py
Original file line number Diff line number Diff line change
@@ -1,31 +1,9 @@
import os
import subprocess
import time
from pathlib import Path

import pytest

REQUIRED = {
"DISCORD_TOKEN": "dummy",
"GITHUB_TOKEN": "dummy",
"GITHUB_REPO": "dummy",
}

for k, v in REQUIRED.items():
os.environ.setdefault(k, v)

@pytest.fixture(scope="session", autouse=True)
def _restore_env():
old = {k: os.environ.get(k) for k in REQUIRED}
try:
yield
finally:
for k, v in old.items():
if v is None:
os.environ.pop(k, None)
else:
os.environ[k] = v

DATABASE_URL = "postgresql://postgres:postgres@localhost:5433/clusterdev"


Expand Down
70 changes: 0 additions & 70 deletions tests/test_validate_user_header.py

This file was deleted.

Loading