mirror of
https://github.com/Telecominfraproject/OpenCellular.git
synced 2025-12-29 18:11:05 +00:00
dbefb29f02
Recently, there have been several changes to LPC and host command modules. This CL fixes unit tests after these changes. BUG=none TEST=All test passed. Change-Id: I263716899af78a61e324fcd0b2636b948617a264 Reviewed-on: https://gerrit.chromium.org/gerrit/27354 Reviewed-by: Rong Chang <rongchang@chromium.org> Tested-by: Vic Yang <victoryang@chromium.org> Commit-Ready: Vic Yang <victoryang@chromium.org>
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
#
|
|
# Utility functions for flash related test
|
|
#
|
|
|
|
import random
|
|
import re
|
|
|
|
# Fixed random seed.
|
|
random.seed(1234)
|
|
|
|
def get_flash_size(helper):
|
|
helper.ec_command("hcflashinfo")
|
|
return int(helper.wait_output("flash_size = (?P<f>\d+)", use_re=True)["f"])
|
|
|
|
def get_ro_size(helper):
|
|
helper.ec_command("rosize")
|
|
return int(helper.wait_output(
|
|
"RO image size = (?P<ro>0x[0-9a-f]+)", use_re=True)["ro"], 16)
|
|
|
|
def xor_sum(size, seed, mult, add):
|
|
ret = 0
|
|
for i in xrange(size):
|
|
ret ^= (seed & 0xff)
|
|
seed = seed * mult + add
|
|
return ret
|
|
|
|
def test_erase(helper, offset, size):
|
|
helper.ec_command("hcflasherase %d %d" % (offset, size))
|
|
helper.wait_output("Flash erase at %x size %x" % (offset, size))
|
|
|
|
def _get_read_ref(helper, offset, size):
|
|
ret = []
|
|
retsub = []
|
|
assert size % 4 == 0
|
|
while size > 0:
|
|
helper.ec_command("rw %d" % offset)
|
|
h = helper.wait_output("read.*=\s+0x(?P<h>[0-9a-f]+)", use_re=True)["h"]
|
|
# Change endianess here
|
|
retsub.append(re.sub('(..)(..)(..)(..)', r'\4\3\2\1', h))
|
|
if len(retsub) == 8:
|
|
ret.append(''.join(retsub))
|
|
retsub = []
|
|
size = size - 4
|
|
offset = offset + 4
|
|
if retsub:
|
|
ret.append(''.join(retsub))
|
|
return ret
|
|
|
|
def test_read(helper, offset, size):
|
|
ref = _get_read_ref(helper, offset, size)
|
|
helper.ec_command("hcflashread %d %d" % (offset, size))
|
|
for line in ref:
|
|
helper.wait_output(line)
|
|
|
|
def test_write(helper, offset, size, expect_fail=False):
|
|
seed = random.randint(2, 10000)
|
|
mult = random.randint(2, 10000)
|
|
add = random.randint(2, 10000)
|
|
helper.ec_command("hcflashwrite %d %d %d %d %d" %
|
|
(offset, size, seed, mult, add))
|
|
if expect_fail:
|
|
helper.wait_output("Command returned error")
|
|
else:
|
|
expected_sum = xor_sum(size, seed, mult, add)
|
|
helper.wait_output("Flash write at %x size %x XOR %x" %
|
|
(offset, size, expected_sum))
|