Add test for OAuth2 prompting config

Reviewed-by: Iker Pedrosa <ipedrosa@redhat.com>
Reviewed-by: Sumit Bose <sbose@redhat.com>

Author: eisenmann-b1

src/tests/cmocka/test_prompt_config.c

@@ -134,6 +134,22 @@ void test_pc_list_add_smartcard(void **state)
     pc_list_free(pc_list);
 }
 
+void test_pc_list_add_oauth2(void **state)
+{
+    int ret;
+    struct prompt_config **pc_list = NULL;
+
+    ret = pc_list_add_oauth2(&pc_list, "inter");
+    assert_int_equal(ret, EOK);
+    assert_non_null(pc_list);
+    assert_non_null(pc_list[0]);
+    assert_int_equal(PC_TYPE_OAUTH2, pc_get_type(pc_list[0]));
+    assert_string_equal("inter", pc_get_oauth2_inter_prompt(pc_list[0]));
+    assert_null(pc_list[1]);
+
+    pc_list_free(pc_list);
+}
+
 void test_pam_get_response_prompt_config(void **state)
 {
     int ret;
@@ -156,21 +172,26 @@ void test_pam_get_response_prompt_config(void **state)
     ret = pc_list_add_smartcard(&pc_list, "init", "PIN");
     assert_int_equal(ret, EOK);
 
+    ret = pc_list_add_oauth2(&pc_list, "inter");
+    assert_int_equal(ret, EOK);
+
     ret = pam_get_response_prompt_config(pc_list, &len, &data);
     pc_list_free(pc_list);
     assert_int_equal(ret, EOK);
-    assert_int_equal(len, 96);
+    assert_int_equal(len, 109);
 
 #if __BYTE_ORDER == __LITTLE_ENDIAN
-    assert_memory_equal(data, "\5\0\0\0\1\0\0\0\10\0\0\0" "password\2\0\0\0\5\0\0\0"
+    assert_memory_equal(data, "\6\0\0\0\1\0\0\0\10\0\0\0" "password\2\0\0\0\5\0\0\0"
                         "first\6\0\0\0" "second\3\0\0\0\6\0\0\0" "single\6\0\0\0\4\0\0\0"
                         "init\4\0\0\0" "link\5\0\0\0\4\0\0\0"
-                        "init\3\0\0\0" "PIN", len);
+                        "init\3\0\0\0" "PIN\7\0\0\0\5\0\0\0"
+                        "inter", len);
 #else
-    assert_memory_equal(data, "\0\0\0\5\0\0\0\1\0\0\0\10" "password\0\0\0\2\0\0\0\5"
+    assert_memory_equal(data, "\0\0\0\6\0\0\0\1\0\0\0\10" "password\0\0\0\2\0\0\0\5"
                         "first\0\0\0\6" "second\0\0\0\3\0\0\0\6" "single\0\0\0\6\0\0\0\4"
                         "init\0\0\0\4" "link\0\0\0\5\0\0\0\4"
-                        "init\0\0\0\3" "PIN", len);
+                        "init\0\0\0\3" "PIN\0\0\0\7\0\0\0\5"
+                        "inter", len);
 #endif
 
     free(data);
@@ -198,10 +219,13 @@ void test_pc_list_from_response(void **state)
     ret = pc_list_add_smartcard(&pc_list, "init", "PIN");
     assert_int_equal(ret, EOK);
 
+    ret = pc_list_add_oauth2(&pc_list, "inter");
+    assert_int_equal(ret, EOK);
+
     ret = pam_get_response_prompt_config(pc_list, &len, &data);
     pc_list_free(pc_list);
     assert_int_equal(ret, EOK);
-    assert_int_equal(len, 96);
+    assert_int_equal(len, 109);
 
     pc_list = NULL;
 
@@ -233,7 +257,11 @@ void test_pc_list_from_response(void **state)
     assert_string_equal("init", pc_get_smartcard_init_prompt(pc_list[4]));
     assert_string_equal("PIN", pc_get_smartcard_pin_prompt(pc_list[4]));
 
-    assert_null(pc_list[5]);
+    assert_non_null(pc_list[5]);
+    assert_int_equal(PC_TYPE_OAUTH2, pc_get_type(pc_list[5]));
+    assert_string_equal("inter", pc_get_oauth2_inter_prompt(pc_list[5]));
+
+    assert_null(pc_list[6]);
 
     pc_list_free(pc_list);
 }
@@ -254,6 +282,7 @@ int main(int argc, const char *argv[])
         cmocka_unit_test(test_pc_list_add_2fa),
         cmocka_unit_test(test_pc_list_add_eidp),
         cmocka_unit_test(test_pc_list_add_smartcard),
+        cmocka_unit_test(test_pc_list_add_oauth2),
         cmocka_unit_test(test_pam_get_response_prompt_config),
         cmocka_unit_test(test_pc_list_from_response),
     };