Forgetting to save really sucks. Sorry again for the wasted commit.

[reactos.git] / rostests / drivers / kmtest / kmtest.h

diff --git a/rostests/drivers/kmtest/kmtest.h b/rostests/drivers/kmtest/kmtest.h
index 658bf8b..f57e85d 100644 (file)
--- a/rostests/drivers/kmtest/kmtest.h
+++ b/rostests/drivers/kmtest/kmtest.h
@@ -1,9 +1,11 @@
-#ifndef PNPTEST_H
-#define PNPTEST_H
+#ifndef KMTEST_H
+#define KMTEST_H
 
 #include <stdio.h>
 #include <stdarg.h>
 #include <string.h>
+#include "ntddk.h"
+
 
 /*
     Some macros, structs, and vars are based or inspired from the great
@@ -28,7 +30,7 @@ typedef struct
 extern tls_data glob_data;
 
 VOID StartTest();
-VOID FinishTest(LPSTR TestName);
+VOID FinishTest(HANDLE KeyHandle, LPWSTR TestName);
 void kmtest_set_location(const char* file, int line);
 
 #ifdef __GNUC__
@@ -42,7 +44,11 @@ extern int kmtest_ok( int condition, const char *msg, ... );
 #endif /* __GNUC__ */
 
 
-#define ok_(file, line)     (kmtest_set_location(file, line), 0) ? 0 : kmtest_ok
+#define ok_(file, line)     (kmtest_set_location(file, line), 0) ? (void)0 : kmtest_ok
 #define ok     ok_(__FILE__, __LINE__)
 
-#endif /* PNPTEST_H */
+PDEVICE_OBJECT AttachDeviceObject;
+PDEVICE_OBJECT MainDeviceObject;
+PDRIVER_OBJECT ThisDriverObject;
+
+#endif /* KMTEST_H */