Engine/source/testing/memoryTester.h

   1 //-----------------------------------------------------------------------------
   2 // Copyright (c) 2013 Stephan Brenner https://github.com/ymx/gtest_mem
   3 //
   4 // Permission is hereby granted, free of charge, to any person obtaining a copy
   5 // of this software and associated documentation files (the "Software"), to
   6 // deal in the Software without restriction, including without limitation the
   7 // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
   8 // sell copies of the Software, and to permit persons to whom the Software is
   9 // furnished to do so, subject to the following conditions:
  10 //
  11 // The above copyright notice and this permission notice shall be included in
  12 // all copies or substantial portions of the Software.
  13 //
  14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  20 // IN THE SOFTWARE.
  21 //-----------------------------------------------------------------------------
  22
  23 #include <iostream>
  24 #include <gtest/gtest.h>
  25
  26 #if defined(TORQUE_OS_WIN)
  27 #define _CRTDBG_MAP_ALLOC
  28 #include <crtdbg.h>
  29 #endif
  30
  31 namespace testing
  32 {
  33    class MemoryLeakDetector : public EmptyTestEventListener
  34    {
  35    public:
  36       virtual void OnTestStart(const TestInfo&)
  37       {
  38 #if defined(TORQUE_OS_WIN)
  39          _CrtMemCheckpoint(&memState_);
  40 #endif
  41       }
  42
  43       virtual void OnTestEnd(const TestInfo& test_info)
  44       {
  45          if(test_info.result()->Passed())
  46          {
  47 #if defined(TORQUE_OS_WIN)
  48             _CrtMemState stateNow, stateDiff;
  49             _CrtMemCheckpoint(&stateNow);
  50             int diffResult = _CrtMemDifference(&stateDiff, &memState_, &stateNow);
  51             if (diffResult)
  52             {
  53                FAIL() << "Memory leak of " << stateDiff.lSizes[1] << " byte(s) detected.";
  54             }
  55 #endif
  56          }
  57       }
  58
  59    private:
  60 #if defined(TORQUE_OS_WIN)
  61       _CrtMemState memState_;
  62 #endif
  63    };
  64 }