mirror of
https://github.com/espressif/mbedtls.git
synced 2025-07-13 10:15:49 +08:00
f80458a160
For easier maintenance the framework repository is flattened here and added to the forked branch in source format.
1916 lines
58 KiB
Python
Executable File
1916 lines
58 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Unit test for generate_test_code.py
|
|
#
|
|
# Copyright The Mbed TLS Contributors
|
|
# SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
|
|
|
|
"""
|
|
Unit tests for generate_test_code.py
|
|
"""
|
|
|
|
from io import StringIO
|
|
from unittest import TestCase, main as unittest_main
|
|
from unittest.mock import patch
|
|
|
|
from generate_test_code import gen_dependencies, gen_dependencies_one_line
|
|
from generate_test_code import gen_function_wrapper, gen_dispatch
|
|
from generate_test_code import parse_until_pattern, GeneratorInputError
|
|
from generate_test_code import parse_suite_dependencies
|
|
from generate_test_code import parse_function_dependencies
|
|
from generate_test_code import parse_function_arguments, parse_function_code
|
|
from generate_test_code import parse_functions, END_HEADER_REGEX
|
|
from generate_test_code import END_SUITE_HELPERS_REGEX, escaped_split
|
|
from generate_test_code import parse_test_data, gen_dep_check
|
|
from generate_test_code import gen_expression_check, write_dependencies
|
|
from generate_test_code import write_parameters, gen_suite_dep_checks
|
|
from generate_test_code import gen_from_test_data
|
|
|
|
|
|
class GenDep(TestCase):
|
|
"""
|
|
Test suite for function gen_dep()
|
|
"""
|
|
|
|
def test_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = ['DEP1', 'DEP2']
|
|
dep_start, dep_end = gen_dependencies(dependencies)
|
|
preprocessor1, preprocessor2 = dep_start.splitlines()
|
|
endif1, endif2 = dep_end.splitlines()
|
|
self.assertEqual(preprocessor1, '#if defined(DEP1)',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(preprocessor2, '#if defined(DEP2)',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(endif1, '#endif /* DEP2 */',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(endif2, '#endif /* DEP1 */',
|
|
'Preprocessor generated incorrectly')
|
|
|
|
def test_disabled_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = ['!DEP1', '!DEP2']
|
|
dep_start, dep_end = gen_dependencies(dependencies)
|
|
preprocessor1, preprocessor2 = dep_start.splitlines()
|
|
endif1, endif2 = dep_end.splitlines()
|
|
self.assertEqual(preprocessor1, '#if !defined(DEP1)',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(preprocessor2, '#if !defined(DEP2)',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(endif1, '#endif /* !DEP2 */',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(endif2, '#endif /* !DEP1 */',
|
|
'Preprocessor generated incorrectly')
|
|
|
|
def test_mixed_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = ['!DEP1', 'DEP2']
|
|
dep_start, dep_end = gen_dependencies(dependencies)
|
|
preprocessor1, preprocessor2 = dep_start.splitlines()
|
|
endif1, endif2 = dep_end.splitlines()
|
|
self.assertEqual(preprocessor1, '#if !defined(DEP1)',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(preprocessor2, '#if defined(DEP2)',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(endif1, '#endif /* DEP2 */',
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(endif2, '#endif /* !DEP1 */',
|
|
'Preprocessor generated incorrectly')
|
|
|
|
def test_empty_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = []
|
|
dep_start, dep_end = gen_dependencies(dependencies)
|
|
self.assertEqual(dep_start, '', 'Preprocessor generated incorrectly')
|
|
self.assertEqual(dep_end, '', 'Preprocessor generated incorrectly')
|
|
|
|
def test_large_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = []
|
|
count = 10
|
|
for i in range(count):
|
|
dependencies.append('DEP%d' % i)
|
|
dep_start, dep_end = gen_dependencies(dependencies)
|
|
self.assertEqual(len(dep_start.splitlines()), count,
|
|
'Preprocessor generated incorrectly')
|
|
self.assertEqual(len(dep_end.splitlines()), count,
|
|
'Preprocessor generated incorrectly')
|
|
|
|
|
|
class GenDepOneLine(TestCase):
|
|
"""
|
|
Test Suite for testing gen_dependencies_one_line()
|
|
"""
|
|
|
|
def test_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = ['DEP1', 'DEP2']
|
|
dep_str = gen_dependencies_one_line(dependencies)
|
|
self.assertEqual(dep_str, '#if defined(DEP1) && defined(DEP2)',
|
|
'Preprocessor generated incorrectly')
|
|
|
|
def test_disabled_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = ['!DEP1', '!DEP2']
|
|
dep_str = gen_dependencies_one_line(dependencies)
|
|
self.assertEqual(dep_str, '#if !defined(DEP1) && !defined(DEP2)',
|
|
'Preprocessor generated incorrectly')
|
|
|
|
def test_mixed_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = ['!DEP1', 'DEP2']
|
|
dep_str = gen_dependencies_one_line(dependencies)
|
|
self.assertEqual(dep_str, '#if !defined(DEP1) && defined(DEP2)',
|
|
'Preprocessor generated incorrectly')
|
|
|
|
def test_empty_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = []
|
|
dep_str = gen_dependencies_one_line(dependencies)
|
|
self.assertEqual(dep_str, '', 'Preprocessor generated incorrectly')
|
|
|
|
def test_large_dependencies_list(self):
|
|
"""
|
|
Test that gen_dep() correctly creates dependencies for given
|
|
dependency list.
|
|
:return:
|
|
"""
|
|
dependencies = []
|
|
count = 10
|
|
for i in range(count):
|
|
dependencies.append('DEP%d' % i)
|
|
dep_str = gen_dependencies_one_line(dependencies)
|
|
expected = '#if ' + ' && '.join(['defined(%s)' %
|
|
x for x in dependencies])
|
|
self.assertEqual(dep_str, expected,
|
|
'Preprocessor generated incorrectly')
|
|
|
|
|
|
class GenFunctionWrapper(TestCase):
|
|
"""
|
|
Test Suite for testing gen_function_wrapper()
|
|
"""
|
|
|
|
def test_params_unpack(self):
|
|
"""
|
|
Test that params are properly unpacked in the function call.
|
|
|
|
:return:
|
|
"""
|
|
code = gen_function_wrapper('test_a', '', ('a', 'b', 'c', 'd'))
|
|
expected = '''
|
|
static void test_a_wrapper( void ** params )
|
|
{
|
|
|
|
test_a( a, b, c, d );
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
def test_local(self):
|
|
"""
|
|
Test that params are properly unpacked in the function call.
|
|
|
|
:return:
|
|
"""
|
|
code = gen_function_wrapper('test_a',
|
|
'int x = 1;', ('x', 'b', 'c', 'd'))
|
|
expected = '''
|
|
static void test_a_wrapper( void ** params )
|
|
{
|
|
int x = 1;
|
|
test_a( x, b, c, d );
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
def test_empty_params(self):
|
|
"""
|
|
Test that params are properly unpacked in the function call.
|
|
|
|
:return:
|
|
"""
|
|
code = gen_function_wrapper('test_a', '', ())
|
|
expected = '''
|
|
static void test_a_wrapper( void ** params )
|
|
{
|
|
(void)params;
|
|
|
|
test_a( );
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
|
|
class GenDispatch(TestCase):
|
|
"""
|
|
Test suite for testing gen_dispatch()
|
|
"""
|
|
|
|
def test_dispatch(self):
|
|
"""
|
|
Test that dispatch table entry is generated correctly.
|
|
:return:
|
|
"""
|
|
code = gen_dispatch('test_a', ['DEP1', 'DEP2'])
|
|
expected = '''
|
|
#if defined(DEP1) && defined(DEP2)
|
|
test_a_wrapper,
|
|
#else
|
|
NULL,
|
|
#endif
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
def test_empty_dependencies(self):
|
|
"""
|
|
Test empty dependency list.
|
|
:return:
|
|
"""
|
|
code = gen_dispatch('test_a', [])
|
|
expected = '''
|
|
test_a_wrapper,
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
|
|
class StringIOWrapper(StringIO):
|
|
"""
|
|
file like class to mock file object in tests.
|
|
"""
|
|
def __init__(self, file_name, data, line_no=0):
|
|
"""
|
|
Init file handle.
|
|
|
|
:param file_name:
|
|
:param data:
|
|
:param line_no:
|
|
"""
|
|
super(StringIOWrapper, self).__init__(data)
|
|
self.line_no = line_no
|
|
self.name = file_name
|
|
|
|
def next(self):
|
|
"""
|
|
Iterator method. This method overrides base class's
|
|
next method and extends the next method to count the line
|
|
numbers as each line is read.
|
|
|
|
:return: Line read from file.
|
|
"""
|
|
parent = super(StringIOWrapper, self)
|
|
line = parent.__next__()
|
|
return line
|
|
|
|
def readline(self, _length=0):
|
|
"""
|
|
Wrap the base class readline.
|
|
|
|
:param length:
|
|
:return:
|
|
"""
|
|
line = super(StringIOWrapper, self).readline()
|
|
if line is not None:
|
|
self.line_no += 1
|
|
return line
|
|
|
|
|
|
class ParseUntilPattern(TestCase):
|
|
"""
|
|
Test Suite for testing parse_until_pattern().
|
|
"""
|
|
|
|
def test_suite_headers(self):
|
|
"""
|
|
Test that suite headers are parsed correctly.
|
|
|
|
:return:
|
|
"""
|
|
data = '''#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
/* END_HEADER */
|
|
'''
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data, line_no=0)
|
|
headers = parse_until_pattern(stream, END_HEADER_REGEX)
|
|
self.assertEqual(headers, expected)
|
|
|
|
def test_line_no(self):
|
|
"""
|
|
Test that #line is set to correct line no. in source .function file.
|
|
|
|
:return:
|
|
"""
|
|
data = '''#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
/* END_HEADER */
|
|
'''
|
|
offset_line_no = 5
|
|
expected = '''#line %d "test_suite_ut.function"
|
|
#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
''' % (offset_line_no + 1)
|
|
stream = StringIOWrapper('test_suite_ut.function', data,
|
|
offset_line_no)
|
|
headers = parse_until_pattern(stream, END_HEADER_REGEX)
|
|
self.assertEqual(headers, expected)
|
|
|
|
def test_no_end_header_comment(self):
|
|
"""
|
|
Test that InvalidFileFormat is raised when end header comment is
|
|
missing.
|
|
:return:
|
|
"""
|
|
data = '''#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(GeneratorInputError, parse_until_pattern, stream,
|
|
END_HEADER_REGEX)
|
|
|
|
|
|
class ParseSuiteDependencies(TestCase):
|
|
"""
|
|
Test Suite for testing parse_suite_dependencies().
|
|
"""
|
|
|
|
def test_suite_dependencies(self):
|
|
"""
|
|
|
|
:return:
|
|
"""
|
|
data = '''
|
|
* depends_on:MBEDTLS_ECP_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
'''
|
|
expected = ['MBEDTLS_ECP_C']
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
dependencies = parse_suite_dependencies(stream)
|
|
self.assertEqual(dependencies, expected)
|
|
|
|
def test_no_end_dep_comment(self):
|
|
"""
|
|
Test that InvalidFileFormat is raised when end dep comment is missing.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
* depends_on:MBEDTLS_ECP_C
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(GeneratorInputError, parse_suite_dependencies,
|
|
stream)
|
|
|
|
def test_dependencies_split(self):
|
|
"""
|
|
Test that InvalidFileFormat is raised when end dep comment is missing.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
* depends_on:MBEDTLS_ECP_C:A:B: C : D :F : G: !H
|
|
* END_DEPENDENCIES
|
|
*/
|
|
'''
|
|
expected = ['MBEDTLS_ECP_C', 'A', 'B', 'C', 'D', 'F', 'G', '!H']
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
dependencies = parse_suite_dependencies(stream)
|
|
self.assertEqual(dependencies, expected)
|
|
|
|
|
|
class ParseFuncDependencies(TestCase):
|
|
"""
|
|
Test Suite for testing parse_function_dependencies()
|
|
"""
|
|
|
|
def test_function_dependencies(self):
|
|
"""
|
|
Test that parse_function_dependencies() correctly parses function
|
|
dependencies.
|
|
:return:
|
|
"""
|
|
line = '/* BEGIN_CASE ' \
|
|
'depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */'
|
|
expected = ['MBEDTLS_ENTROPY_NV_SEED', 'MBEDTLS_FS_IO']
|
|
dependencies = parse_function_dependencies(line)
|
|
self.assertEqual(dependencies, expected)
|
|
|
|
def test_no_dependencies(self):
|
|
"""
|
|
Test that parse_function_dependencies() correctly parses function
|
|
dependencies.
|
|
:return:
|
|
"""
|
|
line = '/* BEGIN_CASE */'
|
|
dependencies = parse_function_dependencies(line)
|
|
self.assertEqual(dependencies, [])
|
|
|
|
def test_tolerance(self):
|
|
"""
|
|
Test that parse_function_dependencies() correctly parses function
|
|
dependencies.
|
|
:return:
|
|
"""
|
|
line = '/* BEGIN_CASE depends_on:MBEDTLS_FS_IO: A : !B:C : F*/'
|
|
dependencies = parse_function_dependencies(line)
|
|
self.assertEqual(dependencies, ['MBEDTLS_FS_IO', 'A', '!B', 'C', 'F'])
|
|
|
|
|
|
class ParseFuncSignature(TestCase):
|
|
"""
|
|
Test Suite for parse_function_arguments().
|
|
"""
|
|
|
|
def test_int_and_char_params(self):
|
|
"""
|
|
Test int and char parameters parsing
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold( char * a, int b, int result )'
|
|
args, local, arg_dispatch = parse_function_arguments(line)
|
|
self.assertEqual(args, ['char*', 'int', 'int'])
|
|
self.assertEqual(local, '')
|
|
self.assertEqual(arg_dispatch,
|
|
['(char *) params[0]',
|
|
'((mbedtls_test_argument_t *) params[1])->sint',
|
|
'((mbedtls_test_argument_t *) params[2])->sint'])
|
|
|
|
def test_hex_params(self):
|
|
"""
|
|
Test hex parameters parsing
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold( char * a, data_t * h, int result )'
|
|
args, local, arg_dispatch = parse_function_arguments(line)
|
|
self.assertEqual(args, ['char*', 'hex', 'int'])
|
|
self.assertEqual(local,
|
|
' data_t data1 = {(uint8_t *) params[1], '
|
|
'((mbedtls_test_argument_t *) params[2])->len};\n')
|
|
self.assertEqual(arg_dispatch, ['(char *) params[0]',
|
|
'&data1',
|
|
'((mbedtls_test_argument_t *) params[3])->sint'])
|
|
|
|
def test_unsupported_arg(self):
|
|
"""
|
|
Test unsupported argument type
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold( char * a, data_t * h, unknown_t result )'
|
|
self.assertRaises(ValueError, parse_function_arguments, line)
|
|
|
|
def test_empty_params(self):
|
|
"""
|
|
Test no parameters (nothing between parentheses).
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold()'
|
|
args, local, arg_dispatch = parse_function_arguments(line)
|
|
self.assertEqual(args, [])
|
|
self.assertEqual(local, '')
|
|
self.assertEqual(arg_dispatch, [])
|
|
|
|
def test_blank_params(self):
|
|
"""
|
|
Test no parameters (space between parentheses).
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold( )'
|
|
args, local, arg_dispatch = parse_function_arguments(line)
|
|
self.assertEqual(args, [])
|
|
self.assertEqual(local, '')
|
|
self.assertEqual(arg_dispatch, [])
|
|
|
|
def test_void_params(self):
|
|
"""
|
|
Test no parameters (void keyword).
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold(void)'
|
|
args, local, arg_dispatch = parse_function_arguments(line)
|
|
self.assertEqual(args, [])
|
|
self.assertEqual(local, '')
|
|
self.assertEqual(arg_dispatch, [])
|
|
|
|
def test_void_space_params(self):
|
|
"""
|
|
Test no parameters (void with spaces).
|
|
:return:
|
|
"""
|
|
line = 'void entropy_threshold( void )'
|
|
args, local, arg_dispatch = parse_function_arguments(line)
|
|
self.assertEqual(args, [])
|
|
self.assertEqual(local, '')
|
|
self.assertEqual(arg_dispatch, [])
|
|
|
|
|
|
class ParseFunctionCode(TestCase):
|
|
"""
|
|
Test suite for testing parse_function_code()
|
|
"""
|
|
|
|
def test_no_function(self):
|
|
"""
|
|
Test no test function found.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
No
|
|
test
|
|
function
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
err_msg = 'file: test_suite_ut.function - Test functions not found!'
|
|
self.assertRaisesRegex(GeneratorInputError, err_msg,
|
|
parse_function_code, stream, [], [])
|
|
|
|
def test_no_end_case_comment(self):
|
|
"""
|
|
Test missing end case.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
void test_func()
|
|
{
|
|
}
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
err_msg = r'file: test_suite_ut.function - '\
|
|
'end case pattern .*? not found!'
|
|
self.assertRaisesRegex(GeneratorInputError, err_msg,
|
|
parse_function_code, stream, [], [])
|
|
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_function_called(self,
|
|
parse_function_arguments_mock):
|
|
"""
|
|
Test parse_function_code()
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
data = '''
|
|
void test_func()
|
|
{
|
|
}
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(GeneratorInputError, parse_function_code,
|
|
stream, [], [])
|
|
self.assertTrue(parse_function_arguments_mock.called)
|
|
parse_function_arguments_mock.assert_called_with('void test_func()\n')
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_return(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test generated code.
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''
|
|
void func()
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
name, arg, code, dispatch_code = parse_function_code(stream, [], [])
|
|
|
|
self.assertTrue(parse_function_arguments_mock.called)
|
|
parse_function_arguments_mock.assert_called_with('void func()\n')
|
|
gen_function_wrapper_mock.assert_called_with('test_func', '', [])
|
|
self.assertEqual(name, 'test_func')
|
|
self.assertEqual(arg, [])
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
static void test_func(void)
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
;
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
self.assertEqual(dispatch_code, "\n test_func_wrapper,\n")
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_with_exit_label(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test when exit label is present.
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''
|
|
void func()
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
_, _, code, _ = parse_function_code(stream, [], [])
|
|
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
static void test_func(void)
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
def test_non_void_function(self):
|
|
"""
|
|
Test invalid signature (non void).
|
|
:return:
|
|
"""
|
|
data = 'int entropy_threshold( char * a, data_t * h, int result )'
|
|
err_msg = 'file: test_suite_ut.function - Test functions not found!'
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaisesRegex(GeneratorInputError, err_msg,
|
|
parse_function_code, stream, [], [])
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_function_name_on_newline(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test with line break before the function name.
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''
|
|
void
|
|
|
|
|
|
func()
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
_, _, code, _ = parse_function_code(stream, [], [])
|
|
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
static void
|
|
|
|
|
|
test_func(void)
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_case_starting_with_comment(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test with comments before the function signature
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''/* comment */
|
|
/* more
|
|
* comment */
|
|
// this is\\
|
|
still \\
|
|
a comment
|
|
void func()
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
_, _, code, _ = parse_function_code(stream, [], [])
|
|
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void test_func(void)
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_comment_in_prototype(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test with comments in the function prototype
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''
|
|
void func( int x, // (line \\
|
|
comment)
|
|
int y /* lone closing parenthesis) */ )
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
_, _, code, _ = parse_function_code(stream, [], [])
|
|
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
static void test_func( int x,
|
|
|
|
int y )
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_line_comment_in_block_comment(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test with line comment in block comment.
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''
|
|
void func( int x /* // */ )
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
_, _, code, _ = parse_function_code(stream, [], [])
|
|
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
static void test_func( int x )
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
@patch("generate_test_code.gen_dispatch")
|
|
@patch("generate_test_code.gen_dependencies")
|
|
@patch("generate_test_code.gen_function_wrapper")
|
|
@patch("generate_test_code.parse_function_arguments")
|
|
def test_block_comment_in_line_comment(self, parse_function_arguments_mock,
|
|
gen_function_wrapper_mock,
|
|
gen_dependencies_mock,
|
|
gen_dispatch_mock):
|
|
"""
|
|
Test with block comment in line comment.
|
|
:return:
|
|
"""
|
|
parse_function_arguments_mock.return_value = ([], '', [])
|
|
gen_function_wrapper_mock.return_value = ''
|
|
gen_dependencies_mock.side_effect = gen_dependencies
|
|
gen_dispatch_mock.side_effect = gen_dispatch
|
|
data = '''
|
|
// /*
|
|
void func( int x )
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
_, _, code, _ = parse_function_code(stream, [], [])
|
|
|
|
expected = '''#line 1 "test_suite_ut.function"
|
|
|
|
|
|
static void test_func( int x )
|
|
{
|
|
ba ba black sheep
|
|
have you any wool
|
|
exit:
|
|
yes sir yes sir
|
|
3 bags full
|
|
}
|
|
'''
|
|
self.assertEqual(code, expected)
|
|
|
|
|
|
class ParseFunction(TestCase):
|
|
"""
|
|
Test Suite for testing parse_functions()
|
|
"""
|
|
|
|
@patch("generate_test_code.parse_until_pattern")
|
|
def test_begin_header(self, parse_until_pattern_mock):
|
|
"""
|
|
Test that begin header is checked and parse_until_pattern() is called.
|
|
:return:
|
|
"""
|
|
def stop(*_unused):
|
|
"""Stop when parse_until_pattern is called."""
|
|
raise Exception
|
|
parse_until_pattern_mock.side_effect = stop
|
|
data = '''/* BEGIN_HEADER */
|
|
#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
/* END_HEADER */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(Exception, parse_functions, stream)
|
|
parse_until_pattern_mock.assert_called_with(stream, END_HEADER_REGEX)
|
|
self.assertEqual(stream.line_no, 1)
|
|
|
|
@patch("generate_test_code.parse_until_pattern")
|
|
def test_begin_helper(self, parse_until_pattern_mock):
|
|
"""
|
|
Test that begin helper is checked and parse_until_pattern() is called.
|
|
:return:
|
|
"""
|
|
def stop(*_unused):
|
|
"""Stop when parse_until_pattern is called."""
|
|
raise Exception
|
|
parse_until_pattern_mock.side_effect = stop
|
|
data = '''/* BEGIN_SUITE_HELPERS */
|
|
static void print_hello_world()
|
|
{
|
|
printf("Hello World!\n");
|
|
}
|
|
/* END_SUITE_HELPERS */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(Exception, parse_functions, stream)
|
|
parse_until_pattern_mock.assert_called_with(stream,
|
|
END_SUITE_HELPERS_REGEX)
|
|
self.assertEqual(stream.line_no, 1)
|
|
|
|
@patch("generate_test_code.parse_suite_dependencies")
|
|
def test_begin_dep(self, parse_suite_dependencies_mock):
|
|
"""
|
|
Test that begin dep is checked and parse_suite_dependencies() is
|
|
called.
|
|
:return:
|
|
"""
|
|
def stop(*_unused):
|
|
"""Stop when parse_until_pattern is called."""
|
|
raise Exception
|
|
parse_suite_dependencies_mock.side_effect = stop
|
|
data = '''/* BEGIN_DEPENDENCIES
|
|
* depends_on:MBEDTLS_ECP_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(Exception, parse_functions, stream)
|
|
parse_suite_dependencies_mock.assert_called_with(stream)
|
|
self.assertEqual(stream.line_no, 1)
|
|
|
|
@patch("generate_test_code.parse_function_dependencies")
|
|
def test_begin_function_dep(self, func_mock):
|
|
"""
|
|
Test that begin dep is checked and parse_function_dependencies() is
|
|
called.
|
|
:return:
|
|
"""
|
|
def stop(*_unused):
|
|
"""Stop when parse_until_pattern is called."""
|
|
raise Exception
|
|
func_mock.side_effect = stop
|
|
|
|
dependencies_str = '/* BEGIN_CASE ' \
|
|
'depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */\n'
|
|
data = '''%sstatic void test_func()
|
|
{
|
|
}
|
|
''' % dependencies_str
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(Exception, parse_functions, stream)
|
|
func_mock.assert_called_with(dependencies_str)
|
|
self.assertEqual(stream.line_no, 1)
|
|
|
|
@patch("generate_test_code.parse_function_code")
|
|
@patch("generate_test_code.parse_function_dependencies")
|
|
def test_return(self, func_mock1, func_mock2):
|
|
"""
|
|
Test that begin case is checked and parse_function_code() is called.
|
|
:return:
|
|
"""
|
|
func_mock1.return_value = []
|
|
in_func_code = '''static void test_func()
|
|
{
|
|
}
|
|
'''
|
|
func_dispatch = '''
|
|
test_func_wrapper,
|
|
'''
|
|
func_mock2.return_value = 'test_func', [],\
|
|
in_func_code, func_dispatch
|
|
dependencies_str = '/* BEGIN_CASE ' \
|
|
'depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */\n'
|
|
data = '''%sstatic void test_func()
|
|
{
|
|
}
|
|
''' % dependencies_str
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
suite_dependencies, dispatch_code, func_code, func_info = \
|
|
parse_functions(stream)
|
|
func_mock1.assert_called_with(dependencies_str)
|
|
func_mock2.assert_called_with(stream, [], [])
|
|
self.assertEqual(stream.line_no, 5)
|
|
self.assertEqual(suite_dependencies, [])
|
|
expected_dispatch_code = '''/* Function Id: 0 */
|
|
|
|
test_func_wrapper,
|
|
'''
|
|
self.assertEqual(dispatch_code, expected_dispatch_code)
|
|
self.assertEqual(func_code, in_func_code)
|
|
self.assertEqual(func_info, {'test_func': (0, [])})
|
|
|
|
def test_parsing(self):
|
|
"""
|
|
Test case parsing.
|
|
:return:
|
|
"""
|
|
data = '''/* BEGIN_HEADER */
|
|
#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
/* END_HEADER */
|
|
|
|
/* BEGIN_DEPENDENCIES
|
|
* depends_on:MBEDTLS_ECP_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */
|
|
void func1()
|
|
{
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */
|
|
void func2()
|
|
{
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
suite_dependencies, dispatch_code, func_code, func_info = \
|
|
parse_functions(stream)
|
|
self.assertEqual(stream.line_no, 23)
|
|
self.assertEqual(suite_dependencies, ['MBEDTLS_ECP_C'])
|
|
|
|
expected_dispatch_code = '''/* Function Id: 0 */
|
|
|
|
#if defined(MBEDTLS_ECP_C) && defined(MBEDTLS_ENTROPY_NV_SEED) && defined(MBEDTLS_FS_IO)
|
|
test_func1_wrapper,
|
|
#else
|
|
NULL,
|
|
#endif
|
|
/* Function Id: 1 */
|
|
|
|
#if defined(MBEDTLS_ECP_C) && defined(MBEDTLS_ENTROPY_NV_SEED) && defined(MBEDTLS_FS_IO)
|
|
test_func2_wrapper,
|
|
#else
|
|
NULL,
|
|
#endif
|
|
'''
|
|
self.assertEqual(dispatch_code, expected_dispatch_code)
|
|
expected_func_code = '''#if defined(MBEDTLS_ECP_C)
|
|
#line 2 "test_suite_ut.function"
|
|
#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
#if defined(MBEDTLS_ENTROPY_NV_SEED)
|
|
#if defined(MBEDTLS_FS_IO)
|
|
#line 13 "test_suite_ut.function"
|
|
static void test_func1(void)
|
|
{
|
|
exit:
|
|
;
|
|
}
|
|
|
|
static void test_func1_wrapper( void ** params )
|
|
{
|
|
(void)params;
|
|
|
|
test_func1( );
|
|
}
|
|
#endif /* MBEDTLS_FS_IO */
|
|
#endif /* MBEDTLS_ENTROPY_NV_SEED */
|
|
#if defined(MBEDTLS_ENTROPY_NV_SEED)
|
|
#if defined(MBEDTLS_FS_IO)
|
|
#line 19 "test_suite_ut.function"
|
|
static void test_func2(void)
|
|
{
|
|
exit:
|
|
;
|
|
}
|
|
|
|
static void test_func2_wrapper( void ** params )
|
|
{
|
|
(void)params;
|
|
|
|
test_func2( );
|
|
}
|
|
#endif /* MBEDTLS_FS_IO */
|
|
#endif /* MBEDTLS_ENTROPY_NV_SEED */
|
|
#endif /* MBEDTLS_ECP_C */
|
|
'''
|
|
self.assertEqual(func_code, expected_func_code)
|
|
self.assertEqual(func_info, {'test_func1': (0, []),
|
|
'test_func2': (1, [])})
|
|
|
|
def test_same_function_name(self):
|
|
"""
|
|
Test name conflict.
|
|
:return:
|
|
"""
|
|
data = '''/* BEGIN_HEADER */
|
|
#include "mbedtls/ecp.h"
|
|
|
|
#define ECP_PF_UNKNOWN -1
|
|
/* END_HEADER */
|
|
|
|
/* BEGIN_DEPENDENCIES
|
|
* depends_on:MBEDTLS_ECP_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */
|
|
void func()
|
|
{
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE depends_on:MBEDTLS_ENTROPY_NV_SEED:MBEDTLS_FS_IO */
|
|
void func()
|
|
{
|
|
}
|
|
/* END_CASE */
|
|
'''
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
self.assertRaises(GeneratorInputError, parse_functions, stream)
|
|
|
|
|
|
class EscapedSplit(TestCase):
|
|
"""
|
|
Test suite for testing escaped_split().
|
|
Note: Since escaped_split() output is used to write back to the
|
|
intermediate data file. Any escape characters in the input are
|
|
retained in the output.
|
|
"""
|
|
|
|
def test_invalid_input(self):
|
|
"""
|
|
Test when input split character is not a character.
|
|
:return:
|
|
"""
|
|
self.assertRaises(ValueError, escaped_split, '', 'string')
|
|
|
|
def test_empty_string(self):
|
|
"""
|
|
Test empty string input.
|
|
:return:
|
|
"""
|
|
splits = escaped_split('', ':')
|
|
self.assertEqual(splits, [])
|
|
|
|
def test_no_escape(self):
|
|
"""
|
|
Test with no escape character. The behaviour should be same as
|
|
str.split()
|
|
:return:
|
|
"""
|
|
test_str = 'yahoo:google'
|
|
splits = escaped_split(test_str, ':')
|
|
self.assertEqual(splits, test_str.split(':'))
|
|
|
|
def test_escaped_input(self):
|
|
"""
|
|
Test input that has escaped delimiter.
|
|
:return:
|
|
"""
|
|
test_str = r'yahoo\:google:facebook'
|
|
splits = escaped_split(test_str, ':')
|
|
self.assertEqual(splits, [r'yahoo\:google', 'facebook'])
|
|
|
|
def test_escaped_escape(self):
|
|
"""
|
|
Test input that has escaped delimiter.
|
|
:return:
|
|
"""
|
|
test_str = r'yahoo\\:google:facebook'
|
|
splits = escaped_split(test_str, ':')
|
|
self.assertEqual(splits, [r'yahoo\\', 'google', 'facebook'])
|
|
|
|
def test_all_at_once(self):
|
|
"""
|
|
Test input that has escaped delimiter.
|
|
:return:
|
|
"""
|
|
test_str = r'yahoo\\:google:facebook\:instagram\\:bbc\\:wikipedia'
|
|
splits = escaped_split(test_str, ':')
|
|
self.assertEqual(splits, [r'yahoo\\', r'google',
|
|
r'facebook\:instagram\\',
|
|
r'bbc\\', r'wikipedia'])
|
|
|
|
|
|
class ParseTestData(TestCase):
|
|
"""
|
|
Test suite for parse test data.
|
|
"""
|
|
|
|
def test_parser(self):
|
|
"""
|
|
Test that tests are parsed correctly from data file.
|
|
:return:
|
|
"""
|
|
data = """
|
|
Diffie-Hellman full exchange #1
|
|
dhm_do_dhm:10:"23":10:"5"
|
|
|
|
Diffie-Hellman full exchange #2
|
|
dhm_do_dhm:10:"93450983094850938450983409623":10:"9345098304850938450983409622"
|
|
|
|
Diffie-Hellman full exchange #3
|
|
dhm_do_dhm:10:"9345098382739712938719287391879381271":10:"9345098792137312973297123912791271"
|
|
|
|
Diffie-Hellman selftest
|
|
dhm_selftest:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
# List of (name, function_name, dependencies, args)
|
|
tests = list(parse_test_data(stream))
|
|
test1, test2, test3, test4 = tests
|
|
self.assertEqual(test1[0], 3)
|
|
self.assertEqual(test1[1], 'Diffie-Hellman full exchange #1')
|
|
self.assertEqual(test1[2], 'dhm_do_dhm')
|
|
self.assertEqual(test1[3], [])
|
|
self.assertEqual(test1[4], ['10', '"23"', '10', '"5"'])
|
|
|
|
self.assertEqual(test2[0], 6)
|
|
self.assertEqual(test2[1], 'Diffie-Hellman full exchange #2')
|
|
self.assertEqual(test2[2], 'dhm_do_dhm')
|
|
self.assertEqual(test2[3], [])
|
|
self.assertEqual(test2[4], ['10', '"93450983094850938450983409623"',
|
|
'10', '"9345098304850938450983409622"'])
|
|
|
|
self.assertEqual(test3[0], 9)
|
|
self.assertEqual(test3[1], 'Diffie-Hellman full exchange #3')
|
|
self.assertEqual(test3[2], 'dhm_do_dhm')
|
|
self.assertEqual(test3[3], [])
|
|
self.assertEqual(test3[4], ['10',
|
|
'"9345098382739712938719287391879381271"',
|
|
'10',
|
|
'"9345098792137312973297123912791271"'])
|
|
|
|
self.assertEqual(test4[0], 12)
|
|
self.assertEqual(test4[1], 'Diffie-Hellman selftest')
|
|
self.assertEqual(test4[2], 'dhm_selftest')
|
|
self.assertEqual(test4[3], [])
|
|
self.assertEqual(test4[4], [])
|
|
|
|
def test_with_dependencies(self):
|
|
"""
|
|
Test that tests with dependencies are parsed.
|
|
:return:
|
|
"""
|
|
data = """
|
|
Diffie-Hellman full exchange #1
|
|
depends_on:YAHOO
|
|
dhm_do_dhm:10:"23":10:"5"
|
|
|
|
Diffie-Hellman full exchange #2
|
|
dhm_do_dhm:10:"93450983094850938450983409623":10:"9345098304850938450983409622"
|
|
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
# List of (name, function_name, dependencies, args)
|
|
tests = list(parse_test_data(stream))
|
|
test1, test2 = tests
|
|
self.assertEqual(test1[0], 4)
|
|
self.assertEqual(test1[1], 'Diffie-Hellman full exchange #1')
|
|
self.assertEqual(test1[2], 'dhm_do_dhm')
|
|
self.assertEqual(test1[3], ['YAHOO'])
|
|
self.assertEqual(test1[4], ['10', '"23"', '10', '"5"'])
|
|
|
|
self.assertEqual(test2[0], 7)
|
|
self.assertEqual(test2[1], 'Diffie-Hellman full exchange #2')
|
|
self.assertEqual(test2[2], 'dhm_do_dhm')
|
|
self.assertEqual(test2[3], [])
|
|
self.assertEqual(test2[4], ['10', '"93450983094850938450983409623"',
|
|
'10', '"9345098304850938450983409622"'])
|
|
|
|
def test_no_args(self):
|
|
"""
|
|
Test GeneratorInputError is raised when test function name and
|
|
args line is missing.
|
|
:return:
|
|
"""
|
|
data = """
|
|
Diffie-Hellman full exchange #1
|
|
depends_on:YAHOO
|
|
|
|
|
|
Diffie-Hellman full exchange #2
|
|
dhm_do_dhm:10:"93450983094850938450983409623":10:"9345098304850938450983409622"
|
|
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
err = None
|
|
try:
|
|
for _, _, _, _, _ in parse_test_data(stream):
|
|
pass
|
|
except GeneratorInputError as err:
|
|
self.assertEqual(type(err), GeneratorInputError)
|
|
|
|
def test_incomplete_data(self):
|
|
"""
|
|
Test GeneratorInputError is raised when test function name
|
|
and args line is missing.
|
|
:return:
|
|
"""
|
|
data = """
|
|
Diffie-Hellman full exchange #1
|
|
depends_on:YAHOO
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.function', data)
|
|
err = None
|
|
try:
|
|
for _, _, _, _, _ in parse_test_data(stream):
|
|
pass
|
|
except GeneratorInputError as err:
|
|
self.assertEqual(type(err), GeneratorInputError)
|
|
|
|
|
|
class GenDepCheck(TestCase):
|
|
"""
|
|
Test suite for gen_dep_check(). It is assumed this function is
|
|
called with valid inputs.
|
|
"""
|
|
|
|
def test_gen_dep_check(self):
|
|
"""
|
|
Test that dependency check code generated correctly.
|
|
:return:
|
|
"""
|
|
expected = """
|
|
case 5:
|
|
{
|
|
#if defined(YAHOO)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;"""
|
|
out = gen_dep_check(5, 'YAHOO')
|
|
self.assertEqual(out, expected)
|
|
|
|
def test_not_defined_dependency(self):
|
|
"""
|
|
Test dependency with !.
|
|
:return:
|
|
"""
|
|
expected = """
|
|
case 5:
|
|
{
|
|
#if !defined(YAHOO)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;"""
|
|
out = gen_dep_check(5, '!YAHOO')
|
|
self.assertEqual(out, expected)
|
|
|
|
def test_empty_dependency(self):
|
|
"""
|
|
Test invalid dependency input.
|
|
:return:
|
|
"""
|
|
self.assertRaises(GeneratorInputError, gen_dep_check, 5, '!')
|
|
|
|
def test_negative_dep_id(self):
|
|
"""
|
|
Test invalid dependency input.
|
|
:return:
|
|
"""
|
|
self.assertRaises(GeneratorInputError, gen_dep_check, -1, 'YAHOO')
|
|
|
|
|
|
class GenExpCheck(TestCase):
|
|
"""
|
|
Test suite for gen_expression_check(). It is assumed this function
|
|
is called with valid inputs.
|
|
"""
|
|
|
|
def test_gen_exp_check(self):
|
|
"""
|
|
Test that expression check code generated correctly.
|
|
:return:
|
|
"""
|
|
expected = """
|
|
case 5:
|
|
{
|
|
*out_value = YAHOO;
|
|
}
|
|
break;"""
|
|
out = gen_expression_check(5, 'YAHOO')
|
|
self.assertEqual(out, expected)
|
|
|
|
def test_invalid_expression(self):
|
|
"""
|
|
Test invalid expression input.
|
|
:return:
|
|
"""
|
|
self.assertRaises(GeneratorInputError, gen_expression_check, 5, '')
|
|
|
|
def test_negative_exp_id(self):
|
|
"""
|
|
Test invalid expression id.
|
|
:return:
|
|
"""
|
|
self.assertRaises(GeneratorInputError, gen_expression_check,
|
|
-1, 'YAHOO')
|
|
|
|
|
|
class WriteDependencies(TestCase):
|
|
"""
|
|
Test suite for testing write_dependencies.
|
|
"""
|
|
|
|
def test_no_test_dependencies(self):
|
|
"""
|
|
Test when test dependencies input is empty.
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_dependencies = []
|
|
dep_check_code = write_dependencies(stream, [], unique_dependencies)
|
|
self.assertEqual(dep_check_code, '')
|
|
self.assertEqual(len(unique_dependencies), 0)
|
|
self.assertEqual(stream.getvalue(), '')
|
|
|
|
def test_unique_dep_ids(self):
|
|
"""
|
|
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_dependencies = []
|
|
dep_check_code = write_dependencies(stream, ['DEP3', 'DEP2', 'DEP1'],
|
|
unique_dependencies)
|
|
expect_dep_check_code = '''
|
|
case 0:
|
|
{
|
|
#if defined(DEP3)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;
|
|
case 1:
|
|
{
|
|
#if defined(DEP2)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;
|
|
case 2:
|
|
{
|
|
#if defined(DEP1)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;'''
|
|
self.assertEqual(dep_check_code, expect_dep_check_code)
|
|
self.assertEqual(len(unique_dependencies), 3)
|
|
self.assertEqual(stream.getvalue(), 'depends_on:0:1:2\n')
|
|
|
|
def test_dep_id_repeat(self):
|
|
"""
|
|
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_dependencies = []
|
|
dep_check_code = ''
|
|
dep_check_code += write_dependencies(stream, ['DEP3', 'DEP2'],
|
|
unique_dependencies)
|
|
dep_check_code += write_dependencies(stream, ['DEP2', 'DEP1'],
|
|
unique_dependencies)
|
|
dep_check_code += write_dependencies(stream, ['DEP1', 'DEP3'],
|
|
unique_dependencies)
|
|
expect_dep_check_code = '''
|
|
case 0:
|
|
{
|
|
#if defined(DEP3)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;
|
|
case 1:
|
|
{
|
|
#if defined(DEP2)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;
|
|
case 2:
|
|
{
|
|
#if defined(DEP1)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;'''
|
|
self.assertEqual(dep_check_code, expect_dep_check_code)
|
|
self.assertEqual(len(unique_dependencies), 3)
|
|
self.assertEqual(stream.getvalue(),
|
|
'depends_on:0:1\ndepends_on:1:2\ndepends_on:2:0\n')
|
|
|
|
|
|
class WriteParams(TestCase):
|
|
"""
|
|
Test Suite for testing write_parameters().
|
|
"""
|
|
|
|
def test_no_params(self):
|
|
"""
|
|
Test with empty test_args
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_expressions = []
|
|
expression_code = write_parameters(stream, [], [], unique_expressions)
|
|
self.assertEqual(len(unique_expressions), 0)
|
|
self.assertEqual(expression_code, '')
|
|
self.assertEqual(stream.getvalue(), '\n')
|
|
|
|
def test_no_exp_param(self):
|
|
"""
|
|
Test when there is no macro or expression in the params.
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_expressions = []
|
|
expression_code = write_parameters(stream, ['"Yahoo"', '"abcdef00"',
|
|
'0'],
|
|
['char*', 'hex', 'int'],
|
|
unique_expressions)
|
|
self.assertEqual(len(unique_expressions), 0)
|
|
self.assertEqual(expression_code, '')
|
|
self.assertEqual(stream.getvalue(),
|
|
':char*:"Yahoo":hex:"abcdef00":int:0\n')
|
|
|
|
def test_hex_format_int_param(self):
|
|
"""
|
|
Test int parameter in hex format.
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_expressions = []
|
|
expression_code = write_parameters(stream,
|
|
['"Yahoo"', '"abcdef00"', '0xAA'],
|
|
['char*', 'hex', 'int'],
|
|
unique_expressions)
|
|
self.assertEqual(len(unique_expressions), 0)
|
|
self.assertEqual(expression_code, '')
|
|
self.assertEqual(stream.getvalue(),
|
|
':char*:"Yahoo":hex:"abcdef00":int:0xAA\n')
|
|
|
|
def test_with_exp_param(self):
|
|
"""
|
|
Test when there is macro or expression in the params.
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_expressions = []
|
|
expression_code = write_parameters(stream,
|
|
['"Yahoo"', '"abcdef00"', '0',
|
|
'MACRO1', 'MACRO2', 'MACRO3'],
|
|
['char*', 'hex', 'int',
|
|
'int', 'int', 'int'],
|
|
unique_expressions)
|
|
self.assertEqual(len(unique_expressions), 3)
|
|
self.assertEqual(unique_expressions, ['MACRO1', 'MACRO2', 'MACRO3'])
|
|
expected_expression_code = '''
|
|
case 0:
|
|
{
|
|
*out_value = MACRO1;
|
|
}
|
|
break;
|
|
case 1:
|
|
{
|
|
*out_value = MACRO2;
|
|
}
|
|
break;
|
|
case 2:
|
|
{
|
|
*out_value = MACRO3;
|
|
}
|
|
break;'''
|
|
self.assertEqual(expression_code, expected_expression_code)
|
|
self.assertEqual(stream.getvalue(),
|
|
':char*:"Yahoo":hex:"abcdef00":int:0:exp:0:exp:1'
|
|
':exp:2\n')
|
|
|
|
def test_with_repeat_calls(self):
|
|
"""
|
|
Test when write_parameter() is called with same macro or expression.
|
|
:return:
|
|
"""
|
|
stream = StringIOWrapper('test_suite_ut.data', '')
|
|
unique_expressions = []
|
|
expression_code = ''
|
|
expression_code += write_parameters(stream,
|
|
['"Yahoo"', 'MACRO1', 'MACRO2'],
|
|
['char*', 'int', 'int'],
|
|
unique_expressions)
|
|
expression_code += write_parameters(stream,
|
|
['"abcdef00"', 'MACRO2', 'MACRO3'],
|
|
['hex', 'int', 'int'],
|
|
unique_expressions)
|
|
expression_code += write_parameters(stream,
|
|
['0', 'MACRO3', 'MACRO1'],
|
|
['int', 'int', 'int'],
|
|
unique_expressions)
|
|
self.assertEqual(len(unique_expressions), 3)
|
|
self.assertEqual(unique_expressions, ['MACRO1', 'MACRO2', 'MACRO3'])
|
|
expected_expression_code = '''
|
|
case 0:
|
|
{
|
|
*out_value = MACRO1;
|
|
}
|
|
break;
|
|
case 1:
|
|
{
|
|
*out_value = MACRO2;
|
|
}
|
|
break;
|
|
case 2:
|
|
{
|
|
*out_value = MACRO3;
|
|
}
|
|
break;'''
|
|
self.assertEqual(expression_code, expected_expression_code)
|
|
expected_data_file = ''':char*:"Yahoo":exp:0:exp:1
|
|
:hex:"abcdef00":exp:1:exp:2
|
|
:int:0:exp:2:exp:0
|
|
'''
|
|
self.assertEqual(stream.getvalue(), expected_data_file)
|
|
|
|
|
|
class GenTestSuiteDependenciesChecks(TestCase):
|
|
"""
|
|
Test suite for testing gen_suite_dep_checks()
|
|
"""
|
|
def test_empty_suite_dependencies(self):
|
|
"""
|
|
Test with empty suite_dependencies list.
|
|
|
|
:return:
|
|
"""
|
|
dep_check_code, expression_code = \
|
|
gen_suite_dep_checks([], 'DEP_CHECK_CODE', 'EXPRESSION_CODE')
|
|
self.assertEqual(dep_check_code, 'DEP_CHECK_CODE')
|
|
self.assertEqual(expression_code, 'EXPRESSION_CODE')
|
|
|
|
def test_suite_dependencies(self):
|
|
"""
|
|
Test with suite_dependencies list.
|
|
|
|
:return:
|
|
"""
|
|
dep_check_code, expression_code = \
|
|
gen_suite_dep_checks(['SUITE_DEP'], 'DEP_CHECK_CODE',
|
|
'EXPRESSION_CODE')
|
|
expected_dep_check_code = '''
|
|
#if defined(SUITE_DEP)
|
|
DEP_CHECK_CODE
|
|
#endif
|
|
'''
|
|
expected_expression_code = '''
|
|
#if defined(SUITE_DEP)
|
|
EXPRESSION_CODE
|
|
#endif
|
|
'''
|
|
self.assertEqual(dep_check_code, expected_dep_check_code)
|
|
self.assertEqual(expression_code, expected_expression_code)
|
|
|
|
def test_no_dep_no_exp(self):
|
|
"""
|
|
Test when there are no dependency and expression code.
|
|
:return:
|
|
"""
|
|
dep_check_code, expression_code = gen_suite_dep_checks([], '', '')
|
|
self.assertEqual(dep_check_code, '')
|
|
self.assertEqual(expression_code, '')
|
|
|
|
|
|
class GenFromTestData(TestCase):
|
|
"""
|
|
Test suite for gen_from_test_data()
|
|
"""
|
|
|
|
@staticmethod
|
|
@patch("generate_test_code.write_dependencies")
|
|
@patch("generate_test_code.write_parameters")
|
|
@patch("generate_test_code.gen_suite_dep_checks")
|
|
def test_intermediate_data_file(func_mock1,
|
|
write_parameters_mock,
|
|
write_dependencies_mock):
|
|
"""
|
|
Test that intermediate data file is written with expected data.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
My test
|
|
depends_on:DEP1
|
|
func1:0
|
|
'''
|
|
data_f = StringIOWrapper('test_suite_ut.data', data)
|
|
out_data_f = StringIOWrapper('test_suite_ut.datax', '')
|
|
func_info = {'test_func1': (1, ('int',))}
|
|
suite_dependencies = []
|
|
write_parameters_mock.side_effect = write_parameters
|
|
write_dependencies_mock.side_effect = write_dependencies
|
|
func_mock1.side_effect = gen_suite_dep_checks
|
|
gen_from_test_data(data_f, out_data_f, func_info, suite_dependencies)
|
|
write_dependencies_mock.assert_called_with(out_data_f,
|
|
['DEP1'], ['DEP1'])
|
|
write_parameters_mock.assert_called_with(out_data_f, ['0'],
|
|
('int',), [])
|
|
expected_dep_check_code = '''
|
|
case 0:
|
|
{
|
|
#if defined(DEP1)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;'''
|
|
func_mock1.assert_called_with(
|
|
suite_dependencies, expected_dep_check_code, '')
|
|
|
|
def test_function_not_found(self):
|
|
"""
|
|
Test that AssertError is raised when function info in not found.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
My test
|
|
depends_on:DEP1
|
|
func1:0
|
|
'''
|
|
data_f = StringIOWrapper('test_suite_ut.data', data)
|
|
out_data_f = StringIOWrapper('test_suite_ut.datax', '')
|
|
func_info = {'test_func2': (1, ('int',))}
|
|
suite_dependencies = []
|
|
self.assertRaises(GeneratorInputError, gen_from_test_data,
|
|
data_f, out_data_f, func_info, suite_dependencies)
|
|
|
|
def test_different_func_args(self):
|
|
"""
|
|
Test that AssertError is raised when no. of parameters and
|
|
function args differ.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
My test
|
|
depends_on:DEP1
|
|
func1:0
|
|
'''
|
|
data_f = StringIOWrapper('test_suite_ut.data', data)
|
|
out_data_f = StringIOWrapper('test_suite_ut.datax', '')
|
|
func_info = {'test_func2': (1, ('int', 'hex'))}
|
|
suite_dependencies = []
|
|
self.assertRaises(GeneratorInputError, gen_from_test_data, data_f,
|
|
out_data_f, func_info, suite_dependencies)
|
|
|
|
def test_output(self):
|
|
"""
|
|
Test that intermediate data file is written with expected data.
|
|
:return:
|
|
"""
|
|
data = '''
|
|
My test 1
|
|
depends_on:DEP1
|
|
func1:0:0xfa:MACRO1:MACRO2
|
|
|
|
My test 2
|
|
depends_on:DEP1:DEP2
|
|
func2:"yahoo":88:MACRO1
|
|
'''
|
|
data_f = StringIOWrapper('test_suite_ut.data', data)
|
|
out_data_f = StringIOWrapper('test_suite_ut.datax', '')
|
|
func_info = {'test_func1': (0, ('int', 'int', 'int', 'int')),
|
|
'test_func2': (1, ('char*', 'int', 'int'))}
|
|
suite_dependencies = []
|
|
dep_check_code, expression_code = \
|
|
gen_from_test_data(data_f, out_data_f, func_info,
|
|
suite_dependencies)
|
|
expected_dep_check_code = '''
|
|
case 0:
|
|
{
|
|
#if defined(DEP1)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;
|
|
case 1:
|
|
{
|
|
#if defined(DEP2)
|
|
ret = DEPENDENCY_SUPPORTED;
|
|
#else
|
|
ret = DEPENDENCY_NOT_SUPPORTED;
|
|
#endif
|
|
}
|
|
break;'''
|
|
expected_data = '''My test 1
|
|
depends_on:0
|
|
0:int:0:int:0xfa:exp:0:exp:1
|
|
|
|
My test 2
|
|
depends_on:0:1
|
|
1:char*:"yahoo":int:88:exp:0
|
|
|
|
'''
|
|
expected_expression_code = '''
|
|
case 0:
|
|
{
|
|
*out_value = MACRO1;
|
|
}
|
|
break;
|
|
case 1:
|
|
{
|
|
*out_value = MACRO2;
|
|
}
|
|
break;'''
|
|
self.assertEqual(dep_check_code, expected_dep_check_code)
|
|
self.assertEqual(out_data_f.getvalue(), expected_data)
|
|
self.assertEqual(expression_code, expected_expression_code)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest_main()
|