FHSS unit tests: Test removing parent info and get neighbor info

Author: Jarkko Paso

test/nanostack/unittest/service_libs/fhss_config/fhssconfigtest.cpp

@@ -53,3 +53,8 @@ TEST(fhss_config, test_ns_fhss_ws_set_parent)
 {
     CHECK(test_ns_fhss_ws_set_parent());
 }
+
+TEST(fhss_config, test_ns_fhss_ws_remove_parent)
+{
+    CHECK(test_ns_fhss_ws_remove_parent());
+}

test/nanostack/unittest/service_libs/fhss_config/test_fhss_config.c

@@ -125,6 +125,7 @@ bool test_ns_fhss_ws_set_parent()
     uint8_t dest_address[8] = {1,2,3,4,5,6,7,8};
     broadcast_timing_info_t bc_timing_info;
     // Test without api
+    fhss_common_stub.bool_value = false;
     if (-1 != ns_fhss_ws_set_parent(NULL, dest_address, &bc_timing_info)) {
         return false;
     }
@@ -135,3 +136,20 @@ bool test_ns_fhss_ws_set_parent()
     }
     return true;
 }
+
+bool test_ns_fhss_ws_remove_parent()
+{
+    fhss_api_t api;
+    uint8_t dest_address[8] = {1,2,3,4,5,6,7,8};
+    // Test without api
+    fhss_common_stub.bool_value = false;
+    if (-1 != ns_fhss_ws_remove_parent(NULL, dest_address)) {
+        return false;
+    }
+    // Test success
+    fhss_common_stub.bool_value = true;
+    if (0 != ns_fhss_ws_remove_parent(&api, dest_address)) {
+        return false;
+    }
+    return true;
+}

test/nanostack/unittest/service_libs/fhss_config/test_fhss_config.h

@@ -33,6 +33,8 @@ bool test_ns_fhss_delete();
 bool test_ns_fhss_configuration_set();
 // Test setting WS parent
 bool test_ns_fhss_ws_set_parent();
+// Test removing WS parent
+bool test_ns_fhss_ws_remove_parent();
 
 
 #ifdef __cplusplus

test/nanostack/unittest/service_libs/fhss_ws/fhsswstest.cpp

@@ -93,3 +93,8 @@ TEST(fhssws, test_fhss_ws_set_parent)
 {
     CHECK(test_fhss_ws_set_parent());
 }
+
+TEST(fhssws, test_fhss_ws_remove_parent)
+{
+    CHECK(test_fhss_ws_remove_parent());
+}

test/nanostack/unittest/service_libs/fhss_ws/test_fhss_ws.c

@@ -54,6 +54,11 @@ static int32_t app_channel_function(const fhss_api_t *api, uint16_t slot, uint8_
     return VENDOR_CHANNEL;
 }
 
+static fhss_ws_neighbor_timing_info_t *app_no_neighbor_info(const fhss_api_t *api, uint8_t eui64[8])
+{
+    return NULL;
+}
+
 static fhss_ws_neighbor_timing_info_t *app_get_neighbor_info(const fhss_api_t *api, uint8_t eui64[8])
 {
     return &neighbor_info;
@@ -85,6 +90,10 @@ static fhss_api_t *test_generate_fhss_api(void)
     memset(&ws, 0, sizeof(fhss_ws_t));
     memset(&fhss_callbacks_stub, 0, sizeof(fhss_callbacks_stub_def));
     memset(&fhss_common_stub, 0, sizeof(fhss_common_stub_def));
+    neighbor_info.uc_timing_info.ufsi = 1000;
+    neighbor_info.uc_timing_info.unicast_channel_function = WS_TR51CF;
+    neighbor_info.uc_timing_info.unicast_dwell_interval = 200;
+    neighbor_info.uc_timing_info.utt_rx_timestamp = 100000;
     fhss_common_stub.fhss_struct.fhss_api = &fhss_api;
     fhss_common_stub.fhss_struct.ws = &ws;
     fhss_common_stub.fhss_struct.ws->fhss_configuration.fhss_uc_dwell_interval = 200;
@@ -170,8 +179,14 @@ bool test_fhss_ws_tx_handle_callback()
     if (fhss_common_stub.fhss_struct.fhss_api->tx_handle(api, true, dest_address, DEFAULT_FRAME_TYPE, DEFAULT_FRAME_LENGTH, DEFAULT_PHY_HEAD_LENGTH, DEFAULT_PHY_TAIL_LENGTH, DEFAULT_TX_TIME) != 0) {
         return false;
     }
-    // Test fixed channel
+    // Test when no neighbor info found
     fhss_common_stub.fhss_struct.fhss_state = DEFAULT_FHSS_STATE;
+    fhss_common_stub.fhss_struct.ws->fhss_configuration.get_neighbor_info = &app_no_neighbor_info;
+    if (fhss_common_stub.fhss_struct.fhss_api->tx_handle(api, DEFAULT_IS_BC_DEST, dest_address, DEFAULT_FRAME_TYPE, DEFAULT_FRAME_LENGTH, DEFAULT_PHY_HEAD_LENGTH, DEFAULT_PHY_TAIL_LENGTH, DEFAULT_TX_TIME) != -1) {
+        return false;
+    }
+    // Test fixed channel
+    fhss_common_stub.fhss_struct.ws->fhss_configuration.get_neighbor_info = &app_get_neighbor_info;
     fhss_common_stub.fhss_struct.ws->fhss_configuration.ws_channel_function = WS_FIXED_CHANNEL;
     if (fhss_common_stub.fhss_struct.fhss_api->tx_handle(api, DEFAULT_IS_BC_DEST, dest_address, DEFAULT_FRAME_TYPE, DEFAULT_FRAME_LENGTH, DEFAULT_PHY_HEAD_LENGTH, DEFAULT_PHY_TAIL_LENGTH, DEFAULT_TX_TIME) != -1) {
         return false;
@@ -181,6 +196,8 @@ bool test_fhss_ws_tx_handle_callback()
     if (fhss_common_stub.fhss_struct.fhss_api->tx_handle(api, DEFAULT_IS_BC_DEST, dest_address, DEFAULT_FRAME_TYPE, DEFAULT_FRAME_LENGTH, DEFAULT_PHY_HEAD_LENGTH, DEFAULT_PHY_TAIL_LENGTH, DEFAULT_TX_TIME) != 0) {
         return false;
     }
+    // For more coverage, set neighbor info NULL
+    memset(&neighbor_info, 0, sizeof(fhss_ws_neighbor_timing_info_t));
     // Test direct hash channel function
     fhss_common_stub.fhss_struct.ws->fhss_configuration.ws_channel_function = WS_DH1CF;
     if (fhss_common_stub.fhss_struct.fhss_api->tx_handle(api, DEFAULT_IS_BC_DEST, dest_address, DEFAULT_FRAME_TYPE, DEFAULT_FRAME_LENGTH, DEFAULT_PHY_HEAD_LENGTH, DEFAULT_PHY_TAIL_LENGTH, DEFAULT_TX_TIME) != 0) {
@@ -457,3 +474,23 @@ bool test_fhss_ws_set_parent()
     }
     return true;
 }
+
+bool test_fhss_ws_remove_parent()
+{
+    uint8_t dest_address[8] = {1,2,3,4,5,6,7,8};
+    fhss_api_t *api = test_generate_fhss_api();
+    broadcast_timing_info_t bc_timing_info;
+    fhss_common_stub.fhss_struct.ws->parent_bc_info = &bc_timing_info;
+
+    // Test without WS enabled FHSS
+    fhss_structure_t fake_fhss_structure;
+    memset(&fake_fhss_structure, 0, sizeof(fhss_structure_t));
+    if (-1 != fhss_ws_remove_parent(&fake_fhss_structure, dest_address)) {
+        return false;
+    }
+    // Test success
+    if ((0 != fhss_ws_remove_parent(&fhss_common_stub.fhss_struct, dest_address)) || (fhss_common_stub.fhss_struct.ws->parent_bc_info != NULL)) {
+        return false;
+    }
+    return true;
+}

test/nanostack/unittest/service_libs/fhss_ws/test_fhss_ws.h

@@ -36,6 +36,7 @@ bool test_fhss_broadcast_handler();
 bool test_fhss_ws_update_uc_channel_callback();
 bool test_fhss_unicast_handler();
 bool test_fhss_ws_set_parent();
+bool test_fhss_ws_remove_parent();
 
 #ifdef __cplusplus
 }