tests/lib/test_elfparser.py

############################################################################
# SPDX-License-Identifier: Apache-2.0
#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.  The
# ASF licenses this file to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance with the
# License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# 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 re

import pytest

from ntfc.lib.elf.elf_parser import ElfParser


def test_lib_elf_parser():
    with pytest.raises(AttributeError):
        _ = ElfParser("xxx")

    with pytest.raises(AttributeError):
        _ = ElfParser("./tests/resources/nuttx/sim/kv_config")

    a = ElfParser("./tests/resources/nuttx/sim/nuttx")

    assert a.has_symbol("hello_main") is True
    assert a.has_symbol("dummy_main") is False
    assert a.has_symbol(re.compile("hello")) is True
    assert a.has_symbol(re.compile("dummy_main")) is False

    assert a.get_symbols_by_pattern("hello", "main")[0].name == "hello_main"
    assert len(a.get_symbols_by_pattern("", "main")) > 1
    assert a.get_symbols_by_pattern("dummy", "main") == []

    with pytest.raises(ValueError):
        _ = a.has_symbol(b"xx")