tests: lib: devicetree: Add sub-partition tests

Adds tests for sub-partition macros

Signed-off-by: Jamie McCrae <jamie.mccrae@nordicsemi.no>

Author: nordicjm

tests/lib/devicetree/api/app.overlay

@@ -861,10 +861,30 @@
 						reg = < 0x0 0xc0 >;
 						label = "test-partition-0";
 					};
+
 					partition@c0 {
 						reg = < 0xc0 0x40 >;
 						label = "test-partition-1";
 					};
+
+					partition@100 {
+						compatible = "fixed-subpartitions";
+						label = "test-subpartitions";
+						reg = <0x00000100 0x100>;
+						ranges = <0x0 0x100 0x100>;
+						#address-cells = <1>;
+						#size-cells = <1>;
+
+						partition@0 {
+							label = "test-subpartition-0";
+							reg = <0x00000000 0x40>;
+						};
+
+						partition@40 {
+							label = "test-subpartition-1";
+							reg = <0x00000040 0xc0>;
+						};
+					};
 				};
 			};
 		};

tests/lib/devicetree/api/src/main.c

@@ -104,6 +104,13 @@
 #define TEST_PARTITION_1 DT_PATH(test, test_mtd_ffeeddcc, flash_20000000, partitions, partition_c0)
 #define TEST_PARTITION_2 DT_PATH(test, test_mtd_33221100, partitions, partition_6ff80)
 
+#define TEST_SUBPARTITION_COMBINED DT_PATH(test, test_mtd_ffeeddcc, flash_20000000, partitions, \
+					   partition_100)
+#define TEST_SUBPARTITION_0 DT_PATH(test, test_mtd_ffeeddcc, flash_20000000, partitions, \
+				    partition_100, partition_0)
+#define TEST_SUBPARTITION_1 DT_PATH(test, test_mtd_ffeeddcc, flash_20000000, partitions, \
+				    partition_100, partition_40)
+
 #define ZEPHYR_USER DT_PATH(zephyr_user)
 
 #define TA_HAS_COMPAT(compat) DT_NODE_HAS_COMPAT(TEST_ARRAYS, compat)
@@ -3215,6 +3222,34 @@ ZTEST(devicetree_api, test_fixed_partitions)
 #undef FIXED_PARTITION_ID_COMMA
 }
 
+ZTEST(devicetree_api, test_fixed_subpartitions)
+{
+	zassert_true(DT_FIXED_PARTITION_EXISTS(TEST_SUBPARTITION_COMBINED));
+	zassert_true(DT_FIXED_SUBPARTITION_EXISTS(TEST_SUBPARTITION_0));
+	zassert_true(DT_FIXED_SUBPARTITION_EXISTS(TEST_SUBPARTITION_1));
+
+	/* Test DT_MEM_FROM_FIXED_SUBPARTITION. */
+	zassert_true(DT_NODE_EXISTS(DT_MEM_FROM_FIXED_PARTITION(TEST_SUBPARTITION_COMBINED)));
+	zassert_true(DT_NODE_EXISTS(DT_MEM_FROM_FIXED_SUBPARTITION(TEST_SUBPARTITION_0)));
+	zassert_true(DT_NODE_EXISTS(DT_MEM_FROM_FIXED_SUBPARTITION(TEST_SUBPARTITION_1)));
+
+	/* Test DT_MTD_FROM_FIXED_SUBPARTITION. */
+	zassert_true(DT_NODE_EXISTS(DT_MTD_FROM_FIXED_PARTITION(TEST_SUBPARTITION_COMBINED)));
+	zassert_true(DT_NODE_EXISTS(DT_MTD_FROM_FIXED_SUBPARTITION(TEST_SUBPARTITION_0)));
+	zassert_true(DT_NODE_EXISTS(DT_MTD_FROM_FIXED_SUBPARTITION(TEST_SUBPARTITION_1)));
+
+	/* Test DT_FIXED_SUBPARTITION_ADDR. */
+	zassert_equal(DT_FIXED_PARTITION_ADDR(TEST_SUBPARTITION_COMBINED), 0x20000100);
+	zassert_equal(DT_FIXED_SUBPARTITION_ADDR(TEST_SUBPARTITION_0),
+		      DT_FIXED_PARTITION_ADDR(TEST_SUBPARTITION_COMBINED));
+	zassert_equal(DT_FIXED_SUBPARTITION_ADDR(TEST_SUBPARTITION_0), 0x20000100);
+	zassert_equal(DT_FIXED_SUBPARTITION_ADDR(TEST_SUBPARTITION_1), 0x20000140);
+
+	/* Check sizes match */
+	zassert_equal(DT_REG_SIZE(TEST_SUBPARTITION_COMBINED),
+		      (DT_REG_SIZE(TEST_SUBPARTITION_0) + DT_REG_SIZE(TEST_SUBPARTITION_1)));
+}
+
 ZTEST(devicetree_api, test_string_token)
 {
 #undef DT_DRV_COMPAT