mirror of
				https://github.com/smaeul/u-boot.git
				synced 2025-10-31 03:58:17 +00:00 
			
		
		
		
	Generate events when devices are probed or removed, allowing hooks to be added for these situations. This is controlled by the DM_EVENT config option. Signed-off-by: Simon Glass <sjg@chromium.org>
		
			
				
	
	
		
			86 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			86 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // SPDX-License-Identifier: GPL-2.0+
 | |
| /*
 | |
|  * Unit tests for event handling
 | |
|  *
 | |
|  * Copyright 2021 Google LLC
 | |
|  * Written by Simon Glass <sjg@chromium.org>
 | |
|  */
 | |
| 
 | |
| #include <common.h>
 | |
| #include <dm.h>
 | |
| #include <event.h>
 | |
| #include <test/common.h>
 | |
| #include <test/test.h>
 | |
| #include <test/ut.h>
 | |
| 
 | |
| struct test_state {
 | |
| 	struct udevice *dev;
 | |
| 	int val;
 | |
| };
 | |
| 
 | |
| static int h_adder(void *ctx, struct event *event)
 | |
| {
 | |
| 	struct event_data_test *data = &event->data.test;
 | |
| 	struct test_state *test_state = ctx;
 | |
| 
 | |
| 	test_state->val += data->signal;
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int test_event_base(struct unit_test_state *uts)
 | |
| {
 | |
| 	struct test_state state;
 | |
| 	int signal;
 | |
| 
 | |
| 	state.val = 12;
 | |
| 	ut_assertok(event_register("wibble", EVT_TEST, h_adder, &state));
 | |
| 
 | |
| 	signal = 17;
 | |
| 
 | |
| 	/* Check that the handler is called */
 | |
| 	ut_assertok(event_notify(EVT_TEST, &signal, sizeof(signal)));
 | |
| 	ut_asserteq(12 + 17, state.val);
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| COMMON_TEST(test_event_base, 0);
 | |
| 
 | |
| static int h_probe(void *ctx, struct event *event)
 | |
| {
 | |
| 	struct test_state *test_state = ctx;
 | |
| 
 | |
| 	test_state->dev = event->data.dm.dev;
 | |
| 	switch (event->type) {
 | |
| 	case EVT_DM_PRE_PROBE:
 | |
| 		test_state->val |= 1;
 | |
| 		break;
 | |
| 	case EVT_DM_POST_PROBE:
 | |
| 		test_state->val |= 2;
 | |
| 		break;
 | |
| 	default:
 | |
| 		break;
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static int test_event_probe(struct unit_test_state *uts)
 | |
| {
 | |
| 	struct test_state state;
 | |
| 	struct udevice *dev;
 | |
| 
 | |
| 	state.val = 0;
 | |
| 	ut_assertok(event_register("pre", EVT_DM_PRE_PROBE, h_probe, &state));
 | |
| 	ut_assertok(event_register("post", EVT_DM_POST_PROBE, h_probe, &state));
 | |
| 
 | |
| 	/* Probe a device */
 | |
| 	ut_assertok(uclass_first_device_err(UCLASS_TEST_FDT, &dev));
 | |
| 
 | |
| 	/* Check that the handler is called */
 | |
| 	ut_asserteq(3, state.val);
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| COMMON_TEST(test_event_probe, UT_TESTF_DM | UT_TESTF_SCAN_FDT);
 |