test: cpu: test current cpu

Add test whether the CPU is U-Boot is running from.

Reviewed-by: Simon Glass <sjg@chromium.org>
Signed-off-by: Peng Fan <peng.fan@nxp.com>

diff --git a/test/dm/cpu.c b/test/dm/cpu.c
index e6dc576ea3c3d9d262a8a0126ebe2f8bb0be998d..def9b64a282fd575f19d5805a7726b45de77ec2d 100644 (file)
--- a/test/dm/cpu.c
+++ b/test/dm/cpu.c
@@ -26,6 +26,8 @@ static int dm_test_cpu(struct unit_test_state *uts)
                ut_assert(dev->flags & DM_FLAG_ACTIVATED);
 
        ut_assertok(uclass_get_device_by_name(UCLASS_CPU, "cpu-test1", &dev));
+       ut_asserteq_ptr(cpu_get_current_dev(), dev);
+       ut_asserteq(cpu_is_current(dev), 1);
 
        ut_assertok(cpu_get_desc(dev, text, sizeof(text)));
        ut_assertok(strcmp(text, "LEG Inc. SuperMegaUltraTurbo CPU No. 1"));