Add test for updating publish

diff --git a/test/wifi-aware-test.c b/test/wifi-aware-test.c
index b82e337..e257567 100644 (file)
--- a/test/wifi-aware-test.c
+++ b/test/wifi-aware-test.c
@@ -82,6 +82,7 @@
        } while (0)
 
 #define SERVICE_NAME "CtsVerifierTestService"
+#define UPDATE_SERVICE_NAME "UpdatedServiceName"
 #define MATCH_FILTER "bytes used for matching"
 #define PUB_SSI "Extra bytes in the publisher discovery"
 #define SUB_SSI "Arbitrary bytes for the subscribe discovery"
@@ -103,6 +104,7 @@ enum {
        CMD_ENABLE,
        CMD_PUBLISH,
        CMD_CANCEL_PUBLISH,
+       CMD_UPDATE_PUBLISH,
        CMD_SUBSCRIBE,
        CMD_CANCEL_SUBSCRIBE,
 
@@ -129,6 +131,8 @@ static char *g_menu_str[] = {
                        = "PUBLISH",
                [CMD_CANCEL_PUBLISH]
                        = "CANCEL PUBLISH",
+               [CMD_UPDATE_PUBLISH]
+                       = "UPDATE PUBLISH",
                [CMD_SUBSCRIBE]
                        = "SUBSCRIBE",
                [CMD_CANCEL_SUBSCRIBE]
@@ -417,6 +421,39 @@ void test_cancel_publish()
        __print_result(ret, "wifi_aware_cancel_publish");
 }
 
+static void __updated_publish_cb(wifi_aware_error_e error, wifi_aware_publish_h publish, void *user_data)
+{
+       if (error == WIFI_AWARE_ERROR_NONE)
+               printf("Update Publish Request is success. Publish ID: [*]\n");
+       else
+               printf("Update Publish Request is fail.\n");
+}
+
+void test_update_publish()
+{
+       int ret = 0;
+       unsigned char service_name[MAX_SERVICE_NAME_LEN + 1] = {0, };
+       size_t len = 0;
+
+       RET_IF_LOOP_IS_NULL();
+       RET_IF_HANDLE_IS_NULL();
+
+       if (g_publish_handle == NULL) {
+               printf("publish should be done\n");
+               return;
+       }
+
+       len = strlen(UPDATE_SERVICE_NAME);
+       memcpy(service_name, UPDATE_SERVICE_NAME, len);
+       ret = wifi_aware_publish_set_service_name(g_publish_handle,
+                       service_name, len);
+       if (ret != WIFI_AWARE_ERROR_NONE)
+               return;
+
+       ret = wifi_aware_update_publish(g_wifi_aware_handle, g_publish_handle, __updated_publish_cb, NULL);
+       __print_result(ret, "wifi_aware_update_publish");
+}
+
 static void __subscribed_cb(wifi_aware_error_e error, unsigned int subscribe_id, void *user_data)
 {
        g_sub_id = subscribe_id;
@@ -519,6 +556,7 @@ test_func g_menu_func[] = {
        [CMD_ENABLE]                    = test_enable,
        [CMD_PUBLISH]                   = test_publish,
        [CMD_CANCEL_PUBLISH]    = test_cancel_publish,
+       [CMD_UPDATE_PUBLISH]    = test_update_publish,
        [CMD_SUBSCRIBE]                 = test_subscribe,
        [CMD_CANCEL_SUBSCRIBE]  = test_cancel_subscribe,