conftest.py

import datetime

import pytest
from pytest_mock import MockFixture
from django.contrib.auth.models import Group
from rest_framework.test import APIClient

from common.utils.const import WorkflowStatus, WorkflowType
from sql.models import (
    Instance,
    ResourceGroup,
    SqlWorkflow,
    SqlWorkflowContent,
    QueryPrivilegesApply,
    ArchiveConfig,
    InstanceTag,
    WorkflowAudit,
)
from common.config import SysConfig
from sql.utils.workflow_audit import AuditV2, AuditSetting


@pytest.fixture
def normal_user(django_user_model):
    user = django_user_model.objects.create(
        username="test_user", display="中文显示", is_active=True
    )
    yield user
    user.delete()


@pytest.fixture
def authenticated_api_client(normal_user):
    client = APIClient()
    client.force_authenticate(user=normal_user)
    return client


@pytest.fixture
def workflow_api_data(normal_user, db_instance):
    workflow = SqlWorkflow.objects.create(
        workflow_name="workflow api test",
        group_id=1,
        group_name="group_name",
        instance=db_instance,
        db_name="test_db",
        syntax_type=1,
        is_backup=True,
        engineer=normal_user.username,
        engineer_display=normal_user.display,
        status="workflow_review_pass",
        audit_auth_groups="",
    )
    content = SqlWorkflowContent.objects.create(
        workflow=workflow,
        sql_content="select 1",
        review_content='[{"id": 1, "sql": "select 1", "errlevel": 0}]',
        execute_result="",
    )
    audit = WorkflowAudit.objects.create(
        group_id=workflow.group_id,
        group_name=workflow.group_name,
        workflow_id=workflow.id,
        workflow_type=WorkflowType.SQL_REVIEW,
        workflow_title=workflow.workflow_name,
        workflow_remark="",
        audit_auth_groups="",
        current_audit="",
        next_audit="",
        current_status=WorkflowStatus.PASSED,
        create_user=normal_user.username,
        create_user_display=normal_user.display,
    )
    return workflow, content, audit


@pytest.fixture
def super_user(django_user_model):
    user = django_user_model.objects.create(
        username="super_user", display="超级用户", is_active=True, is_superuser=True
    )
    yield user
    user.delete()


@pytest.fixture
def db_instance(db):
    ins = Instance.objects.create(
        instance_name="some_ins",
        type="slave",
        db_type="mysql",
        host="some_host",
        port=3306,
        user="ins_user",
        password="some_str",
    )
    yield ins
    ins.delete()


@pytest.fixture
def public_instance_permission_data(normal_user, db_instance):
    group = ResourceGroup.objects.create(group_id=101, group_name="public instance")
    can_write = InstanceTag.objects.create(tag_code="can_write", tag_name="支持上线")
    can_read = InstanceTag.objects.create(tag_code="can_read", tag_name="支持查询")
    normal_user.resource_group.add(group)
    db_instance.resource_group.add(group)
    return group, can_write, can_read


@pytest.fixture
def resource_group(db) -> ResourceGroup:
    res_group = ResourceGroup.objects.create(group_id=1, group_name="group_name")
    yield res_group
    res_group.delete()


@pytest.fixture
def sql_workflow(db_instance):
    wf = SqlWorkflow.objects.create(
        workflow_name="some_name",
        group_id=1,
        group_name="g1",
        engineer_display="",
        audit_auth_groups="some_audit_group",
        create_time=datetime.datetime.now(),
        status="workflow_timingtask",
        is_backup=True,
        instance=db_instance,
        db_name="some_db",
        syntax_type=1,
    )
    wf_content = SqlWorkflowContent.objects.create(
        workflow=wf, sql_content="some_sql", execute_result=""
    )
    yield wf, wf_content
    wf.delete()
    wf_content.delete()


@pytest.fixture
def sql_query_apply(db_instance):
    tomorrow = datetime.datetime.today() + datetime.timedelta(days=1)
    query_apply_1 = QueryPrivilegesApply.objects.create(
        group_id=1,
        group_name="some_name",
        title="some_title1",
        user_name="some_user",
        instance=db_instance,
        db_list="some_db,some_db2",
        limit_num=100,
        valid_date=tomorrow,
        priv_type=1,
        status=0,
        audit_auth_groups="1",
    )
    yield query_apply_1
    query_apply_1.delete()


@pytest.fixture
def archive_apply(db_instance, resource_group):
    archive_apply_1 = ArchiveConfig.objects.create(
        title="title",
        resource_group=resource_group,
        audit_auth_groups="",
        src_instance=db_instance,
        src_db_name="src_db_name",
        src_table_name="src_table_name",
        dest_instance=db_instance,
        dest_db_name="src_db_name",
        dest_table_name="src_table_name",
        condition="1=1",
        mode="file",
        no_delete=True,
        sleep=1,
        status=WorkflowStatus.WAITING,
        state=False,
        user_name="some_user",
        user_display="display",
    )
    yield archive_apply_1
    archive_apply_1.delete()


@pytest.fixture
def setup_sys_config(db):
    sys_config = SysConfig()
    yield sys_config
    sys_config.purge()


@pytest.fixture
def create_auth_group(db):
    auth_group = Group.objects.create(name="test_group")
    yield auth_group
    auth_group.delete()


@pytest.fixture
def fake_generate_audit_setting(mocker: MockFixture, super_user, create_auth_group):
    super_user.groups.add(create_auth_group)
    mock_generate_audit_setting = mocker.patch.object(AuditV2, "generate_audit_setting")
    fake_audit_setting = AuditSetting(
        auto_pass=False,
        audit_auth_groups=[create_auth_group.id],
    )
    mock_generate_audit_setting.return_value = fake_audit_setting
    yield mock_generate_audit_setting


@pytest.fixture
def instance_tag(db):
    tag = InstanceTag.objects.create(tag_code="test_tag", tag_name="测试标签")
    yield tag
    tag.delete()


@pytest.fixture
def create_resource_group(db):
    resource_group = ResourceGroup.objects.create(
        group_name="group_name",
        is_deleted=False,
        qywx_webhook="https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=xxx",
        feishu_webhook="https://open.feishu.cn/open-apis/bot/v2/hook/xxx",
        ding_webhook="https://oapi.dingtalk.com/robot/send?access_token=xxx",
    )
    yield resource_group
    resource_group.delete()


@pytest.fixture
def create_audit_workflow(normal_user, create_resource_group):
    audit_wf = WorkflowAudit.objects.create(
        group_id=create_resource_group.group_id,
        group_name=create_resource_group.group_name,
        workflow_id=1,
        workflow_type=2,
        workflow_title="申请标题",
        workflow_remark="申请备注",
        audit_auth_groups="1",
        current_audit="1",
        next_audit="2",
        current_status=0,
        create_user=normal_user.username,
    )
    yield audit_wf
    audit_wf.delete()


@pytest.fixture
def clean_auth_group(db):
    yield
    Group.objects.all().delete()