esphome/tests/component_tests/conftest.py

32 lines
727 B
Python
Raw Permalink Normal View History

"""Fixtures for component tests."""
2020-07-15 14:04:00 +02:00
2021-03-12 23:58:43 +01:00
import sys
from pathlib import Path
# Add package root to python path
here = Path(__file__).parent
package_root = here.parent.parent
sys.path.insert(0, package_root.as_posix())
2020-07-15 14:04:00 +02:00
import pytest
from esphome.core import CORE
from esphome.config import read_config
from esphome.__main__ import generate_cpp_contents
@pytest.fixture
def generate_main():
"""Generates the C++ main.cpp file and returns it in string form."""
2020-07-15 14:04:00 +02:00
def generator(path: str) -> str:
CORE.config_path = path
CORE.config = read_config({})
generate_cpp_contents(CORE.config)
print(CORE.cpp_main_section)
return CORE.cpp_main_section
yield generator
CORE.reset()