X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=EdkModulePkg%2FUniversal%2FPCD%2FTest%2FPcdTest.c;h=6f2a5f1f1f4122a7566b3fdfeb8714879f6dfcc0;hp=21fa67a6aa9789e1a9d9642cf251c37b400f9af7;hb=0653eb895dfb8d059c114120a0779b561c5578ef;hpb=19b362e879de90ef07ba67cd4f8aa97122f625b0

diff --git a/EdkModulePkg/Universal/PCD/Test/PcdTest.c b/EdkModulePkg/Universal/PCD/Test/PcdTest.c
index 21fa67a6aa..6f2a5f1f1f 100644
--- a/EdkModulePkg/Universal/PCD/Test/PcdTest.c
+++ b/EdkModulePkg/Universal/PCD/Test/PcdTest.c
@@ -23,9 +23,10 @@ CONST GUID Guid1  = GUID1;
 EFI_STATUS
 EFIAPI
 OnsetCallback1 (
-  IN  UINT32    CallBackToken,
-  IN  VOID      *TokenData,
-  IN  UINTN     TokenDataSize
+  IN  CONST EFI_GUID  *Guid,
+  IN  UINT32          CallBackToken,
+  IN  VOID            *TokenData,
+  IN  UINTN           TokenDataSize
   )
 {
   DebugPrint (0x80000000, "In CallbackOnSet %x %d\n", * ((UINT32 *)TokenData), TokenDataSize);    
@@ -42,6 +43,10 @@ DoTest(
   UINT16    u16;
   UINT32    u32;
   UINT64    u64;
+  PCD_TOKEN_NUMBER Token;
+  
+
+  LibPcdCallbackOnSet (NULL, PcdToken(PcdTestDynamicUint32), OnsetCallback1);
   
   u32 = 0xafafafaf;
   PcdSet32(PcdTestDynamicUint32, u32);
@@ -57,6 +62,15 @@ DoTest(
   ASSERT (u16 == 0x1234);
   ASSERT (u64 == PcdGet64(PcdTestDynamicUint64));
   ASSERT (u32 == PcdGet32(PcdTestDynamicUint32));
+
+
+  Token = PCD_INVALID_TOKEN_NUMBER;
+
+  do {
+    Token = LibPcdGetNextToken (NULL, Token);
+    DebugPrint (EFI_D_ERROR, "Next Token Number is %d\n", Token);
+  } while (Token != PCD_INVALID_TOKEN_NUMBER);
+  
   
   return;
 }