[Comgr] Fix bugs with mangled name test

Check after linking bitcodes instead of before. Also fix issue with
wrong variable being printed after check

Change-Id: I14d219a2f5656c6789d5abbf164a04c27d4ab395

Author: lamb-j

amd/comgr/docs/ReleaseNotes.md

@@ -128,6 +128,7 @@ unbundling.
 - Add git branch and commit hash for Comgr, and commit hash for LLVM to log
 output for Comgr actions. This can help us debug issues more quickly in cases
 where reporters provide Comgr logs.
+- Fix multiple bugs with mangled names test
 
 New Targets
 -----------

amd/comgr/test/mangled_names_test.c

@@ -116,10 +116,28 @@ int main(int argc, char *argv[]) {
     exit(1);
   }
 
+  Status = amd_comgr_create_data_set(&DataSetLinked);
+  checkError(Status, "amd_comgr_create_data_set");
+
+  Status = amd_comgr_do_action(AMD_COMGR_ACTION_LINK_BC_TO_BC, DataAction,
+                               DataSetBc, DataSetLinked);
+  checkError(Status, "amd_comgr_do_action");
+
+  Status = amd_comgr_action_data_count(DataSetLinked, AMD_COMGR_DATA_KIND_BC,
+                                       &Count);
+  checkError(Status, "amd_comgr_action_data_count");
+
+  if (Count != 1) {
+    printf("AMD_COMGR_ACTION_LINK_BC_TO_BC Failed: "
+           "produced %zu BC objects (expected 1)\n",
+           Count);
+    exit(1);
+  }
+
   // Get bitcode mangled names
   amd_comgr_data_t DataBc;
 
-  Status = amd_comgr_action_data_get_data(DataSetBc,
+  Status = amd_comgr_action_data_get_data(DataSetLinked,
                                           AMD_COMGR_DATA_KIND_BC,
                                           0, &DataBc);
   checkError(Status, "amd_comgr_action_data_get_data");
@@ -158,7 +176,7 @@ int main(int argc, char *argv[]) {
   if (numNames != 2) {
     printf("amd_populate_mangled_names Failed: "
            "produced %zu bitcode names (expected 2)\n",
-           Count);
+           numNames);
     exit(1);
   }
 
@@ -183,24 +201,6 @@ int main(int argc, char *argv[]) {
     free(mName);
   }
 
-  Status = amd_comgr_create_data_set(&DataSetLinked);
-  checkError(Status, "amd_comgr_create_data_set");
-
-  Status = amd_comgr_do_action(AMD_COMGR_ACTION_LINK_BC_TO_BC, DataAction,
-                               DataSetBc, DataSetLinked);
-  checkError(Status, "amd_comgr_do_action");
-
-  Status = amd_comgr_action_data_count(DataSetLinked, AMD_COMGR_DATA_KIND_BC,
-                                       &Count);
-  checkError(Status, "amd_comgr_action_data_count");
-
-  if (Count != 1) {
-    printf("AMD_COMGR_ACTION_LINK_BC_TO_BC Failed: "
-           "produced %zu BC objects (expected 1)\n",
-           Count);
-    exit(1);
-  }
-
   Status = amd_comgr_create_data_set(&DataSetReloc);
   checkError(Status, "amd_comgr_create_data_set");