Move mock driver model classes to shared

Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>

shared/test/common/mocks/mock_driver_model.h

@@ -13,6 +13,8 @@
 #include <cstdint>
 #include <functional>
 
+namespace NEO {
+
 class MockDriverModel : public NEO::DriverModel {
   public:
     MockDriverModel() : NEO::DriverModel(NEO::DriverModelType::UNKNOWN) {}
@@ -24,7 +26,7 @@ class MockDriverModel : public NEO::DriverModel {
     NEO::PhysicalDevicePciBusInfo getPciBusInfo() const override { return pciBusInfo; }
 
     size_t getMaxMemAllocSize() const override {
-        return 0;
+        return maxAllocSize;
     }
 
     bool isGpuHangDetected(NEO::OsContext &osContext) override {
@@ -35,9 +37,26 @@ class MockDriverModel : public NEO::DriverModel {
         return isGpuHangDetectedToReturn;
     }
 
-    PhyicalDevicePciSpeedInfo getPciSpeedInfo() const override { return {}; }
+    PhyicalDevicePciSpeedInfo getPciSpeedInfo() const override { return pciSpeedInfo; }
 
-    NEO::PhysicalDevicePciBusInfo pciBusInfo{};
+    PhyicalDevicePciSpeedInfo pciSpeedInfo{};
+    PhysicalDevicePciBusInfo pciBusInfo{};
     bool isGpuHangDetectedToReturn{};
     std::function<void()> isGpuHangDetectedSideEffect{};
-};
+    size_t maxAllocSize = 0;
+};
+
+class MockDriverModelWDDM : public MockDriverModel {
+  public:
+    MockDriverModelWDDM() : MockDriverModel() {
+        driverModelType = DriverModelType::WDDM;
+    }
+};
+
+class MockDriverModelDRM : public MockDriverModel {
+  public:
+    MockDriverModelDRM() : MockDriverModel() {
+        driverModelType = DriverModelType::DRM;
+    }
+};
+} // namespace NEO

shared/test/common/mocks/mock_memory_manager.h

@@ -411,4 +411,13 @@ class MockMemoryManagerWithCapacity : public MockMemoryManager {
     size_t capacity = 0u;
 };
 
+class MemoryManagerMemHandleMock : public MockMemoryManager {
+  public:
+    bool isNTHandle(osHandle handle, uint32_t rootDeviceIndex) override {
+        return NTHandle;
+    };
+
+    bool NTHandle = false;
+};
+
 } // namespace NEO