[reactos] 04/13: [KMTESTS] Fix tests for uniprocessor checked build

[Jérôme Gardou]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=8d665f895900f2658c623…

commit 8d665f895900f2658c62317f7727da7075f62cdc
Author:     Jérôme Gardou <jerome.gardou(a)reactos.org>
AuthorDate: Wed May 26 17:06:11 2021 +0200
Commit:     Jérôme Gardou <zefklop(a)users.noreply.github.com>
CommitDate: Tue Jun 29 11:49:20 2021 +0200

    [KMTESTS] Fix tests for uniprocessor checked build
---
 modules/rostests/kmtests/ntos_ke/KeSpinLock.c | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/modules/rostests/kmtests/ntos_ke/KeSpinLock.c
b/modules/rostests/kmtests/ntos_ke/KeSpinLock.c
index d761d8e9a25..9b16648573f 100644
--- a/modules/rostests/kmtests/ntos_ke/KeSpinLock.c
+++ b/modules/rostests/kmtests/ntos_ke/KeSpinLock.c
@@ -167,12 +167,16 @@ BOOLEAN TryNoRaise(PKSPIN_LOCK SpinLock, PCHECK_DATA CheckData) {
 {                                                                                   \
     PKTHREAD Thread = KeGetCurrentThread();                                         \
     (VOID)Thread;                                                                   \
-    if (KmtIsMultiProcessorBuild)                                                   \
+    if (KmtIsMultiProcessorBuild || KmtIsCheckedBuild)                              \
     {                                                                               \
         ok_eq_bool(Ret, (Value) == 0);                                              \
         if (SpinLock)                                                               \
-            ok_eq_ulongptr(*(SpinLock),                                             \
-                        (Value) ? (ULONG_PTR)Thread | 1 : 0);                       \
+        {                                                                           \
+            if (KmtIsCheckedBuild)                                                  \
+                ok_eq_ulongptr(*(SpinLock), (Value) ? (ULONG_PTR)Thread | 1 : 0);   \
+            else                                                                    \
+                ok_eq_ulongptr(*(SpinLock), (Value) ? 1 : 0);                       \
+        }                                                                           \
     }                                                                               \
     else                                                                            \
     {                                                                               \
@@ -192,7 +196,7 @@ BOOLEAN TryNoRaise(PKSPIN_LOCK SpinLock, PCHECK_DATA CheckData) {
 
 #define CheckSpinLockQueueHandle(SpinLock, CheckData, Value) do                     \
 {                                                                                   \
-    if (KmtIsMultiProcessorBuild)                                                   \
+    if (KmtIsMultiProcessorBuild || KmtIsCheckedBuild)                              \
     {                                                                               \
         ok_eq_bool(Ret, (Value) == 0);                                              \
         if (SpinLock)                                                               \