from datetime import datetime, timezone from typing import Dict, List import pytest from kubernetes import client from sqlalchemy.orm import Session import mlrun.api.crud as crud from mlrun.api.constants import LogSources from mlrun.api.utils.singletons.db import get_db from mlrun.api.utils.singletons.k8s import get_k8s from mlrun.runtimes import get_runtime_handler from mlrun.runtimes.constants import PodPhases, RunStates from mlrun.utils import create_logger, now_date logger = create_logger(level="debug", name="test-runtime-handlers") class TestRuntimeHandlerBase: def setup_method(self, method): self._logger = logger self._logger.info( f"Setting up test {self.__class__.__name__}::{method.__name__}") self.project = "test-project" self.run_uid = "test_run_uid" self.custom_setup() self._logger.info( f"Finished setting up test {self.__class__.__name__}::{method.__name__}"
def __init__(self): self.state = None self.logger = create_logger('debug', 'human', "flow", sys.stdout) self.worker_id = 0 self.Response = Response self.root = None
import os import pathlib import pytest import sys import yaml from mlrun import get_run_db, mlconf, set_environment from mlrun.utils import create_logger logger = create_logger(level="debug", name="test") class TestMLRunSystem: project_name = "system-test-project" root_path = pathlib.Path(__file__).absolute().parent.parent.parent env_file_path = root_path / "tests" / "system" / "env.yml" results_path = root_path / "tests" / "test_results" / "system" mandatory_env_vars = [ "MLRUN_DBPATH", "V3IO_API", "V3IO_USERNAME", "V3IO_ACCESS_KEY", ] def setup_method(self, method): self._logger = logger self._logger.info( f"Setting up test {self.__class__.__name__}::{method.__name__}") self._test_env = {} self._old_env = {}
import os import pathlib import sys import typing import pytest import yaml import mlrun.api.schemas from mlrun import get_run_db, mlconf, set_environment from mlrun.utils import create_logger logger = create_logger(level="debug", name="test-system") class TestMLRunSystem: project_name = "system-test-project" root_path = pathlib.Path(__file__).absolute().parent.parent.parent env_file_path = root_path / "tests" / "system" / "env.yml" results_path = root_path / "tests" / "test_results" / "system" enterprise_marker_name = "enterprise" mandatory_env_vars = [ "MLRUN_DBPATH", ] mandatory_enterprise_env_vars = mandatory_env_vars + [ "V3IO_API", "V3IO_FRAMESD", "V3IO_USERNAME", "V3IO_ACCESS_KEY", ]
import fastapi.testclient import pytest import sqlalchemy.orm from kubernetes import client from kubernetes import client as k8s_client from kubernetes.client import V1EnvVar from mlrun.api.utils.singletons.k8s import get_k8s from mlrun.config import config as mlconf from mlrun.model import new_task from mlrun.runtimes.constants import PodPhases from mlrun.utils import create_logger from mlrun.utils.azure_vault import AzureVaultStore from mlrun.utils.vault import VaultStore logger = create_logger(level="debug", name="test-runtime") class TestRuntimeBase: def setup_method(self, method): self.namespace = mlconf.namespace = "test-namespace" get_k8s().namespace = self.namespace # set auto-mount to work as if this is an Iguazio system (otherwise it may try to mount PVC) mlconf.igz_version = "1.1.1" mlconf.storage.auto_mount_type = "auto" mlconf.storage.auto_mount_params = "" self._logger = logger self.project = "test-project" self.name = "test-function"
import copy import os import pathlib import subprocess import sys import mlrun import mlrun.api.schemas import tests.conftest from mlrun.utils import create_logger logger = create_logger(level="debug", name="test-integration") class TestMLRunIntegration: project_name = "system-test-project" root_path = pathlib.Path(__file__).absolute().parent.parent.parent.parent results_path = root_path / "tests" / "test_results" / "integration" def setup_method(self, method): self._logger = logger self._logger.info( f"Setting up test {self.__class__.__name__}::{method.__name__}") self.container_id, api_url = self._run_api() self._test_env = {} self._old_env = {} self._setup_env({"MLRUN_DBPATH": api_url}) self.custom_setup()