Adjust the code to use the common test-execution function.
Signed-off-by: NSimon Glass <sjg@chromium.org>