Home | History | Annotate | Download | only in dm
      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