mirror of
https://github.com/github/codeql.git
synced 2025-12-17 09:13:20 +01:00
96897a0cdc
The information that was contained in `schema.yml` is now in `swift/schema.py`, which allows a more integrated IDE experience for writing and navigating it. Another minor change is that `schema.Class` now has a `str` `group` field instead of a `pathlib.Path` `dir` one.
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
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
|