1 /* SPDX-License-Identifier: GPL-2.0+ */ 2 /* 3 * Copyright (c) 2013 Google, Inc. 4 */ 5 6 #ifndef __DM_TEST_H 7 #define __DM_TEST_H 8 9 #include <dm.h> 10 #include <test/test.h> 11 12 /** 13 * struct dm_test_cdata - configuration data for test instance 14 * 15 * @ping_add: Amonut to add each time we get a ping 16 * @base: Base address of this device 17 */ 18 struct dm_test_pdata { 19 int ping_add; 20 uint32_t base; 21 }; 22 23 /** 24 * struct test_ops - Operations supported by the test device 25 * 26 * @ping: Ping operation 27 * @dev: Device to operate on 28 * @pingval: Value to ping the device with 29 * @pingret: Returns resulting value from driver 30 * @return 0 if OK, -ve on error 31 */ 32 struct test_ops { 33 int (*ping)(struct udevice *dev, int pingval, int *pingret); 34 }; 35 36 /* Operations that our test driver supports */ 37 enum { 38 DM_TEST_OP_BIND = 0, 39 DM_TEST_OP_UNBIND, 40 DM_TEST_OP_PROBE, 41 DM_TEST_OP_REMOVE, 42 43 /* For uclass */ 44 DM_TEST_OP_POST_BIND, 45 DM_TEST_OP_PRE_UNBIND, 46 DM_TEST_OP_PRE_PROBE, 47 DM_TEST_OP_POST_PROBE, 48 DM_TEST_OP_PRE_REMOVE, 49 DM_TEST_OP_INIT, 50 DM_TEST_OP_DESTROY, 51 52 DM_TEST_OP_COUNT, 53 }; 54 55 /* Test driver types */ 56 enum { 57 DM_TEST_TYPE_FIRST = 0, 58 DM_TEST_TYPE_SECOND, 59 }; 60 61 /* The number added to the ping total on each probe */ 62 #define DM_TEST_START_TOTAL 5 63 64 /** 65 * struct dm_test_priv - private data for the test devices 66 */ 67 struct dm_test_priv { 68 int ping_total; 69 int op_count[DM_TEST_OP_COUNT]; 70 int uclass_flag; 71 int uclass_total; 72 }; 73 74 /** 75 * struct dm_test_perdev_class_priv - private per-device data for test uclass 76 */ 77 struct dm_test_uclass_perdev_priv { 78 int base_add; 79 }; 80 81 /** 82 * struct dm_test_uclass_priv - private data for test uclass 83 */ 84 struct dm_test_uclass_priv { 85 int total_add; 86 }; 87 88 /** 89 * struct dm_test_parent_data - parent's information on each child 90 * 91 * @sum: Test value used to check parent data works correctly 92 * @flag: Used to track calling of parent operations 93 * @uclass_flag: Used to track calling of parent operations by uclass 94 */ 95 struct dm_test_parent_data { 96 int sum; 97 int flag; 98 }; 99 100 /* Test values for test device's uclass platform data */ 101 enum { 102 TEST_UC_PDATA_INTVAL1 = 2, 103 TEST_UC_PDATA_INTVAL2 = 334, 104 TEST_UC_PDATA_INTVAL3 = 789452, 105 }; 106 107 /** 108 * struct dm_test_uclass_platda - uclass's information on each device 109 * 110 * @intval1: set to TEST_UC_PDATA_INTVAL1 in .post_bind method of test uclass 111 * @intval2: set to TEST_UC_PDATA_INTVAL2 in .post_bind method of test uclass 112 * @intval3: set to TEST_UC_PDATA_INTVAL3 in .post_bind method of test uclass 113 */ 114 struct dm_test_perdev_uc_pdata { 115 int intval1; 116 int intval2; 117 int intval3; 118 }; 119 120 /* 121 * Operation counts for the test driver, used to check that each method is 122 * called correctly 123 */ 124 extern int dm_testdrv_op_count[DM_TEST_OP_COUNT]; 125 126 extern struct unit_test_state global_dm_test_state; 127 128 /* 129 * struct dm_test_state - Entire state of dm test system 130 * 131 * This is often abreviated to dms. 132 * 133 * @root: Root device 134 * @testdev: Test device 135 * @force_fail_alloc: Force all memory allocs to fail 136 * @skip_post_probe: Skip uclass post-probe processing 137 * @removed: Used to keep track of a device that was removed 138 */ 139 struct dm_test_state { 140 struct udevice *root; 141 struct udevice *testdev; 142 int force_fail_alloc; 143 int skip_post_probe; 144 struct udevice *removed; 145 }; 146 147 /* Test flags for each test */ 148 enum { 149 DM_TESTF_SCAN_PDATA = 1 << 0, /* test needs platform data */ 150 DM_TESTF_PROBE_TEST = 1 << 1, /* probe test uclass */ 151 DM_TESTF_SCAN_FDT = 1 << 2, /* scan device tree */ 152 DM_TESTF_FLAT_TREE = 1 << 3, /* test needs flat DT */ 153 DM_TESTF_LIVE_TREE = 1 << 4, /* needs live device tree */ 154 }; 155 156 /* Declare a new driver model test */ 157 #define DM_TEST(_name, _flags) UNIT_TEST(_name, _flags, dm_test) 158 159 /* This platform data is needed in tests, so declare it here */ 160 struct sandbox_sdl_plat { 161 int xres; 162 int yres; 163 int bpix; 164 int rot; 165 const char *vidconsole_drv_name; 166 int font_size; 167 }; 168 169 /* Declare ping methods for the drivers */ 170 int test_ping(struct udevice *dev, int pingval, int *pingret); 171 int testfdt_ping(struct udevice *dev, int pingval, int *pingret); 172 173 /** 174 * dm_check_operations() - Check that we can perform ping operations 175 * 176 * This checks that the ping operations work as expected for a device 177 * 178 * @dms: Overall test state 179 * @dev: Device to test 180 * @base: Base address, used to check ping return value 181 * @priv: Pointer to private test information 182 * @return 0 if OK, -ve on error 183 */ 184 int dm_check_operations(struct unit_test_state *uts, struct udevice *dev, 185 uint32_t base, struct dm_test_priv *priv); 186 187 /** 188 * dm_check_devices() - check the devices respond to operations correctly 189 * 190 * @dms: Overall test state 191 * @num_devices: Number of test devices to check 192 * @return 0 if OK, -ve on error 193 */ 194 int dm_check_devices(struct unit_test_state *uts, int num_devices); 195 196 /** 197 * dm_leak_check_start() - Prepare to check for a memory leak 198 * 199 * Call this before allocating memory to record the amount of memory being 200 * used. 201 * 202 * @dms: Overall test state 203 */ 204 void dm_leak_check_start(struct unit_test_state *uts); 205 206 /** 207 * dm_leak_check_end() - Check that no memory has leaked 208 * 209 * Call this after dm_leak_check_start() and after you have hopefuilly freed 210 * all the memory that was allocated. This function will print an error if 211 * it sees a different amount of total memory allocated than before. 212 * 213 * @dms: Overall test state 214 */int dm_leak_check_end(struct unit_test_state *uts); 215 216 #endif 217