Fix for heapHelper

-add mockable_virtual for memoryManager class

Change-Id: Ie3ff592c7d61759efa7c2955de6b4e8d1731a2c4

Author: kamdiedrich

core/helpers/heap_helper.cpp

@@ -17,13 +17,8 @@ GraphicsAllocation *HeapHelper::getHeapAllocation(size_t heapSize, size_t alignm
     if (allocation) {
         return allocation.release();
     }
-    NEO::AllocationProperties properties{rootDeviceIndex,
-                                         true /* allocateMemory*/,
-                                         alignment,
-                                         GraphicsAllocation::AllocationType::INTERNAL_HEAP,
-                                         isMultiOsContextCapable,
-                                         false /* isMultiStorageAllocation */,
-                                         {}};
+    NEO::AllocationProperties properties{rootDeviceIndex, true, heapSize, GraphicsAllocation::AllocationType::INTERNAL_HEAP, isMultiOsContextCapable, false, {}};
+    properties.alignment = alignment;
 
     return this->memManager->allocateGraphicsMemoryWithProperties(properties);
 }

runtime/memory_manager/memory_manager.h

@@ -69,7 +69,7 @@ class MemoryManager {
         return allocateGraphicsMemoryInPreferredPool(properties, ptr);
     }
 
-    GraphicsAllocation *allocateGraphicsMemoryInPreferredPool(const AllocationProperties &properties, const void *hostPtr);
+    MOCKABLE_VIRTUAL GraphicsAllocation *allocateGraphicsMemoryInPreferredPool(const AllocationProperties &properties, const void *hostPtr);
 
     virtual GraphicsAllocation *createGraphicsAllocationFromSharedHandle(osHandle handle, const AllocationProperties &properties, bool requireSpecificBitness) = 0;
 
@@ -90,7 +90,7 @@ class MemoryManager {
     void freeSystemMemory(void *ptr);
 
     virtual void freeGraphicsMemoryImpl(GraphicsAllocation *gfxAllocation) = 0;
-    void freeGraphicsMemory(GraphicsAllocation *gfxAllocation);
+    MOCKABLE_VIRTUAL void freeGraphicsMemory(GraphicsAllocation *gfxAllocation);
     virtual void handleFenceCompletion(GraphicsAllocation *allocation){};
 
     void checkGpuUsageAndDestroyGraphicsAllocations(GraphicsAllocation *gfxAllocation);