Add UTs for new API

FreeRTOS · Jun 6, 2024 · 09ebd92 · 09ebd92
1 parent 3282b35
commit 09ebd92
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/test/unit-test/FreeRTOS_DNS/FreeRTOS_DNS_utest.c b/test/unit-test/FreeRTOS_DNS/FreeRTOS_DNS_utest.c
@@ -2244,3 +2244,35 @@ void test_FreeRTOS_freeaddrinfo_NullInput( void )
 {
     FreeRTOS_freeaddrinfo( NULL );
 }
+
+/**
+ * @brief Preference is IPv6
+ */
+void test_FreeRTOS_SetDNSIPPreference_IPv4( void )
+{
+    BaseType_t xReturn;
+    xReturn = FreeRTOS_SetDNSIPPreference( xPreferenceIPv4 );
+    TEST_ASSERT_EQUAL( pdPASS, xReturn );
+    TEST_ASSERT_EQUAL( xPreferenceIPv4, xDNS_IP_Preference );
+}
+
+/**
+ * @brief Preference is IPv6
+ */
+void test_FreeRTOS_SetDNSIPPreference_IPv6( void )
+{
+    BaseType_t xReturn;
+    xReturn = FreeRTOS_SetDNSIPPreference( xPreferenceIPv6 );
+    TEST_ASSERT_EQUAL( pdPASS, xReturn );
+    TEST_ASSERT_EQUAL( xPreferenceIPv6, xDNS_IP_Preference );
+}
+
+/**
+ * @brief Preference is None
+ */
+void test_FreeRTOS_SetDNSIPPreference_None( void )
+{
+    BaseType_t xReturn;
+    xReturn = FreeRTOS_SetDNSIPPreference( xPreferenceNone );
+    TEST_ASSERT_EQUAL( pdFAIL, xReturn );
+}