1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2015 Thomas Chou <thomas@wytron.com.tw>
4  */
5 
6 #include <dm.h>
7 #include <timer.h>
8 #include <dm/test.h>
9 #include <dm/device-internal.h>
10 #include <test/test.h>
11 #include <test/ut.h>
12 #include <asm/cpu.h>
13 
14 /*
15  * Basic test of the timer uclass.
16  */
dm_test_timer_base(struct unit_test_state * uts)17 static int dm_test_timer_base(struct unit_test_state *uts)
18 {
19 	struct udevice *dev;
20 
21 	ut_assertok(uclass_get_device_by_name(UCLASS_TIMER, "timer@0", &dev));
22 	ut_asserteq(1000000, timer_get_rate(dev));
23 
24 	return 0;
25 }
26 DM_TEST(dm_test_timer_base, UTF_SCAN_PDATA | UTF_SCAN_FDT);
27 
28 /*
29  * Test of timebase fallback
30  */
dm_test_timer_timebase_fallback(struct unit_test_state * uts)31 static int dm_test_timer_timebase_fallback(struct unit_test_state *uts)
32 {
33 	struct udevice *dev;
34 
35 	cpu_sandbox_set_current("cpu@1");
36 	ut_assertok(uclass_get_device_by_name(UCLASS_TIMER, "timer@1", &dev));
37 	ut_asserteq(3000000, timer_get_rate(dev));
38 	ut_assertok(device_remove(dev, DM_REMOVE_NORMAL));
39 
40 	cpu_sandbox_set_current("cpu@2");
41 	ut_assertok(uclass_get_device_by_name(UCLASS_TIMER, "timer@1", &dev));
42 	ut_asserteq(2000000, timer_get_rate(dev));
43 
44 	cpu_sandbox_set_current("cpu@1");
45 
46 	return 0;
47 }
48 DM_TEST(dm_test_timer_timebase_fallback,
49 	UTF_SCAN_PDATA | UTF_SCAN_FDT);
50