hmz007
36ed224bac
Signed-off-by: hmz007 <hmz007@gmail.com> Change-Id: Ib87fdbe7a0be7dc961af3500fe9ea4589a127f9f
95 lines
3.2 KiB
Python
Executable File
95 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# Copyright 2019 The ChromiumOS Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Tests for post_process_orderfile.py."""
|
|
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
import post_process_orderfile
|
|
|
|
|
|
def _write_nm_file(name):
|
|
with open(name, "w") as out:
|
|
out.write("000001 s NotAValidSymbol1\n")
|
|
out.write("000002 S NotAValidSymbol2\n")
|
|
out.write("000010 t FirstValidSymbol\n")
|
|
out.write("000012 t \n")
|
|
out.write("000020 T Builtins_SecondValidSymbol\n")
|
|
out.write("000030 T $SymbolToIgnore\n")
|
|
out.write("000036 T Builtins_LastValidSymbol\n")
|
|
|
|
|
|
def _write_orderfile(name):
|
|
with open(name, "w") as out:
|
|
out.write("SymbolOrdered1\n")
|
|
out.write("SymbolOrdered2\n")
|
|
|
|
|
|
def _cleanup(files):
|
|
for f in files:
|
|
shutil.rmtree(f, ignore_errors=True)
|
|
|
|
|
|
class Tests(unittest.TestCase):
|
|
"""All of our tests for post_process_orderfile."""
|
|
|
|
# pylint: disable=protected-access
|
|
def test__parse_nm_output(self):
|
|
temp_dir = tempfile.mkdtemp()
|
|
self.addCleanup(_cleanup, [temp_dir])
|
|
chrome_nm_file = os.path.join(temp_dir, "chrome_nm.txt")
|
|
_write_nm_file(chrome_nm_file)
|
|
with open(chrome_nm_file) as f:
|
|
results = list(post_process_orderfile._parse_nm_output(f))
|
|
self.assertEqual(len(results), 3)
|
|
self.assertIn("FirstValidSymbol", results)
|
|
self.assertIn("Builtins_SecondValidSymbol", results)
|
|
self.assertIn("Builtins_LastValidSymbol", results)
|
|
|
|
def test__remove_duplicates(self):
|
|
duplicates = ["marker1", "marker2", "marker3", "marker2", "marker1"]
|
|
results = list(post_process_orderfile._remove_duplicates(duplicates))
|
|
self.assertEqual(results, ["marker1", "marker2", "marker3"])
|
|
|
|
def test_run(self):
|
|
temp_dir = tempfile.mkdtemp()
|
|
self.addCleanup(_cleanup, [temp_dir])
|
|
orderfile_input = os.path.join(temp_dir, "orderfile.in.txt")
|
|
orderfile_output = os.path.join(temp_dir, "orderfile.out.txt")
|
|
chrome_nm_file = os.path.join(temp_dir, "chrome_nm.txt")
|
|
_write_nm_file(chrome_nm_file)
|
|
_write_orderfile(orderfile_input)
|
|
with open(orderfile_input) as in_stream, open(
|
|
orderfile_output, "w"
|
|
) as out_stream, open(chrome_nm_file) as chrome_nm_stream:
|
|
post_process_orderfile.run(in_stream, chrome_nm_stream, out_stream)
|
|
|
|
with open(orderfile_output) as check:
|
|
results = [x.strip() for x in check.readlines()]
|
|
self.assertEqual(
|
|
results,
|
|
[
|
|
# Start marker should be put first.
|
|
"chrome_begin_ordered_code",
|
|
# Symbols in orderfile come next.
|
|
"SymbolOrdered1",
|
|
"SymbolOrdered2",
|
|
# Builtin functions in chrome_nm come next, and sorted.
|
|
"Builtins_LastValidSymbol",
|
|
"Builtins_SecondValidSymbol",
|
|
# Last symbol should be end marker.
|
|
"chrome_end_ordered_code",
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|