# -*- coding: utf-8 -*-
"""
 * TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-蓝鲸 PaaS 平台(BlueKing-PaaS) available.
 * Copyright (C) 2017-2021 THL A29 Limited, a Tencent company. All rights reserved.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at http://opensource.org/licenses/MIT
 * 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.
"""
import tempfile as _tempfile_
from pathlib import Path
from typing import Callable, Generator, Optional

import pytest


@pytest.fixture
def mktemp() -> Generator[Callable[..., Path], None, None]:
    files = []

    def core(content: Optional[str] = None):
        filepath = Path(_tempfile_.mktemp())
        files.append(filepath)
        if content:
            with open(filepath, mode="w") as fh:
                fh.write(content)
        return filepath

    yield core
    for item in files:
        if item.exists:
            item.unlink()
