75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
"""Test high level package capabilities.
|
|
|
|
SPDX-FileCopyrightText: © 2023 Brian S. Stephan <bss@incorporeal.org>
|
|
SPDX-License-Identifier: GPL-3.0-or-later
|
|
"""
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
from decorator import decorator
|
|
|
|
from gp2040ce_bintools import get_config_pb2
|
|
|
|
HERE = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
@decorator
|
|
def with_pb2s(test, *args, **kwargs):
|
|
"""Wrap a test with precompiled pb2 files on the path."""
|
|
proto_path = os.path.join(HERE, 'test-files', 'pb2-files')
|
|
sys.path.append(proto_path)
|
|
|
|
test(*args, **kwargs)
|
|
|
|
sys.path.pop()
|
|
del sys.modules['config_pb2']
|
|
del sys.modules['enums_pb2']
|
|
del sys.modules['nanopb_pb2']
|
|
|
|
|
|
@decorator
|
|
def with_protos(test, *args, **kwargs):
|
|
"""Wrap a test with .proto files on the path."""
|
|
proto_path = os.path.join(HERE, 'test-files', 'proto-files')
|
|
sys.path.append(proto_path)
|
|
|
|
test(*args, **kwargs)
|
|
|
|
sys.path.pop()
|
|
del sys.modules['config_pb2']
|
|
del sys.modules['enums_pb2']
|
|
del sys.modules['nanopb_pb2']
|
|
|
|
|
|
@with_pb2s
|
|
def test_get_config_pb2_precompiled():
|
|
"""With precompiled files on the path, test we can read and use them."""
|
|
# get the module from the provided files
|
|
config_pb2 = get_config_pb2()
|
|
_ = config_pb2.Config()
|
|
|
|
|
|
def test_get_config_pb2_exception():
|
|
"""Test that we fail if no config .proto files are available."""
|
|
with pytest.raises(RuntimeError):
|
|
_ = get_config_pb2()
|
|
|
|
|
|
def test_get_config_pb2_shipped_config_files():
|
|
"""Without any precompiled files or proto files on the path, test we DO NOT raise an exception."""
|
|
# use the shipped .proto files to generate the config
|
|
config_pb2 = get_config_pb2(with_fallback=True)
|
|
_ = config_pb2.Config()
|
|
del sys.modules['config_pb2']
|
|
del sys.modules['enums_pb2']
|
|
del sys.modules['nanopb_pb2']
|
|
|
|
|
|
@with_protos
|
|
def test_get_config_pb2_compile():
|
|
"""Without any precompiled files on the path, test we can read proto files and compile them."""
|
|
# let grpc tools compile the proto files on demand and give us the module
|
|
config_pb2 = get_config_pb2()
|
|
_ = config_pb2.Config()
|