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 */
17static 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}
26DM_TEST(dm_test_timer_base, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
27
28/*
29 * Test of timebase fallback
30 */
31static 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}
48DM_TEST(dm_test_timer_timebase_fallback,
49	UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT);
50