codeql/swift/codegen/test/utils.py at codeql-cli-2.11.3 · github/codeql · GitHub

History

65 lines (47 loc) · 1.61 KB

Raw

import pathlib

from unittest import mock

import pytest

from swift.codegen.lib import render, schema, paths

schema_dir = pathlib.Path("a", "dir")

schema_file = schema_dir / "schema.yml"

dbscheme_file = pathlib.Path("another", "dir", "test.dbscheme")

def write(out, contents=""):

out.parent.mkdir(parents=True, exist_ok=True)

with open(out, "w") as out:

out.write(contents)

@pytest.fixture

def renderer():

return mock.Mock(spec=render.Renderer(""))

@pytest.fixture

def opts():

ret = mock.MagicMock()

ret.swift_dir = paths.swift_dir

return ret

@pytest.fixture(autouse=True)

def override_paths(tmp_path):

with mock.patch("swift.codegen.lib.paths.swift_dir", tmp_path):

yield

@pytest.fixture

def input(opts, tmp_path):

opts.schema = tmp_path / schema_file

with mock.patch("swift.codegen.lib.schema.load_file") as load_mock:

load_mock.return_value = schema.Schema([])

yield load_mock.return_value

assert load_mock.mock_calls == [

mock.call(opts.schema),

], load_mock.mock_calls

@pytest.fixture

def dbscheme_input(opts, tmp_path):

opts.dbscheme = tmp_path / dbscheme_file

with mock.patch("swift.codegen.lib.dbscheme.iterload") as load_mock:

load_mock.entities = []

load_mock.side_effect = lambda _: load_mock.entities

yield load_mock

assert load_mock.mock_calls == [

mock.call(opts.dbscheme),

], load_mock.mock_calls

def run_generation(generate, opts, renderer):

output = {}

renderer.render.side_effect = lambda data, out: output.__setitem__(out, data)

generate(opts, renderer)

return output

codeql/swift/codegen/test/utils.py at codeql-cli-2.11.3 · github/codeql · GitHub

Expand file tree

Breadcrumbs

utils.py

Latest commit

History

Breadcrumbs

utils.py

File metadata and controls