1
0
mirror of https://github.com/ARM-software/workload-automation.git synced 2024-10-06 02:41:11 +01:00
workload-automation/wlauto/tests/test_utils.py
2015-03-10 13:09:31 +00:00

64 lines
2.1 KiB
Python

# Copyright 2013-2015 ARM Limited
#
# Licensed 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.
#
# pylint: disable=R0201
from unittest import TestCase
from nose.tools import raises, assert_equal # pylint: disable=E0611
from wlauto.utils.android import check_output
from wlauto.utils.misc import merge_dicts, TimeoutError
class TestCheckOutput(TestCase):
def test_ok(self):
check_output("python -c 'import time; time.sleep(0.1)'", timeout=0.5, shell=True)
@raises(TimeoutError)
def test_bad(self):
check_output("python -c 'import time; time.sleep(1)'", timeout=0.5, shell=True)
class TestMerge(TestCase):
def test_dict_merge(self):
base = {'a': 1, 'b': {'x': 9, 'z': 10}}
other = {'b': {'x': 7, 'y': 8}, 'c': [1, 2, 3]}
result = merge_dicts(base, other)
assert_equal(result['a'], 1)
assert_equal(result['b']['x'], 7)
assert_equal(result['b']['y'], 8)
assert_equal(result['b']['z'], 10)
assert_equal(result['c'], [1, 2, 3])
def test_merge_dict_lists(self):
base = {'a': [1, 3, 2]}
other = {'a': [3, 4, 5]}
result = merge_dicts(base, other)
assert_equal(result['a'], [1, 3, 2, 3, 4, 5])
result = merge_dicts(base, other, list_duplicates='first')
assert_equal(result['a'], [1, 3, 2, 4, 5])
result = merge_dicts(base, other, list_duplicates='last')
assert_equal(result['a'], [1, 2, 3, 4, 5])
@raises(ValueError)
def test_type_mismatch(self):
base = {'a': [1, 2, 3]}
other = {'a': 'test'}
merge_dicts(base, other, match_types=True)