Add mock function into MockIoLib

microsoft · Nov 8, 2024 · 345357c · 345357c
1 parent a418ed9
commit 345357c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/MdePkg/Test/Mock/Include/GoogleTest/Library/MockIoLib.h b/MdePkg/Test/Mock/Include/GoogleTest/Library/MockIoLib.h
@@ -129,6 +129,15 @@ struct MockIoLib {
      IN      UINT64  Value
     )
     );
+
+  MOCK_FUNCTION_DECLARATION (
+    UINT16,
+    IoWrite16,
+    (
+     IN      UINTN   Port,
+     IN      UINT16  Value
+    )
+    );
 };
 
 #endif
diff --git a/MdePkg/Test/Mock/Library/GoogleTest/MockIoLib/MockIoLib.cpp b/MdePkg/Test/Mock/Library/GoogleTest/MockIoLib/MockIoLib.cpp
@@ -22,3 +22,4 @@ MOCK_FUNCTION_DEFINITION (MockIoLib, MmioRead32, 1, EFIAPI);
 MOCK_FUNCTION_DEFINITION (MockIoLib, MmioWrite32, 2, EFIAPI);
 MOCK_FUNCTION_DEFINITION (MockIoLib, MmioRead64, 1, EFIAPI);
 MOCK_FUNCTION_DEFINITION (MockIoLib, MmioWrite64, 2, EFIAPI);
+MOCK_FUNCTION_DEFINITION (MockIoLib, IoWrite16, 2, EFIAPI);