ref: 734e60aeeb1c5500826ac4e4745898ddd8e96b9a
dir: /test/EncUT_MemoryAlloc.cpp/
#include "../gtest/include/gtest/gtest.h" #include <string.h> // use memset/memcmp #include <time.h> #include "../codec/encoder/core/inc/memory_align.h" using namespace WelsSVCEnc; //Tests of WelsGetCacheLineSize Begin TEST(MemoryAlignTest, GetCacheLineSize_LoopWithin16K) { const unsigned int kuiTestBoundary16K = 16 * 1024; unsigned int uiTargetAlign = 1; while (uiTargetAlign < kuiTestBoundary16K) { CMemoryAlign cTestMa(uiTargetAlign); ASSERT_EQ( (uiTargetAlign & 0x0F)?16:uiTargetAlign, cTestMa.WelsGetCacheLineSize() ); ++ uiTargetAlign; } } TEST(MemoryAlignTest, GetCacheLineSize_Zero) { CMemoryAlign cTestMa(0); ASSERT_EQ( 16, cTestMa.WelsGetCacheLineSize() ); } TEST(MemoryAlignTest, GetCacheLineSize_MaxUINT) { CMemoryAlign cTestMa(0xFFFFFFFF); ASSERT_EQ( 16, cTestMa.WelsGetCacheLineSize() ); } //Tests of WelsGetCacheLineSize End //Tests of WelsMallocAndFree Begin TEST(MemoryAlignTest, WelsMallocAndFreeOnceFunctionVerify) { const uint32_t kuiTargetAlignSize[4] = {32, 16, 64, 8}; srand((uint32_t)time(NULL)); for (int i=0; i<4; i++) { const uint32_t kuiTestAlignSize = kuiTargetAlignSize[i]; const uint32_t kuiTestDataSize = abs(rand()); CMemoryAlign cTestMa(kuiTestAlignSize); const uint32_t uiSize = kuiTestDataSize; const char strUnitTestTag[100] = "pUnitTestData"; const uint32_t kuiUsedCacheLineSize = ((kuiTestAlignSize == 0) || (kuiTestAlignSize & 0x0F)) ? (16) : (kuiTestAlignSize); const uint32_t kuiExtraAlignSize = kuiUsedCacheLineSize-1; const uint32_t kuiExpectedSize = sizeof( void ** ) + sizeof( int32_t ) + kuiExtraAlignSize + uiSize; uint8_t *pUnitTestData = static_cast<uint8_t *>(cTestMa.WelsMalloc(uiSize, strUnitTestTag)); if ( pUnitTestData != NULL ) { ASSERT_TRUE( (((int64_t)(static_cast<void *>(pUnitTestData))) & kuiExtraAlignSize) == 0 ); EXPECT_EQ( kuiExpectedSize, cTestMa.WelsGetMemoryUsage() ); cTestMa.WelsFree( pUnitTestData, strUnitTestTag ); EXPECT_EQ( 0, cTestMa.WelsGetMemoryUsage() ); } else { EXPECT_EQ( NULL, pUnitTestData ); EXPECT_EQ( 0, cTestMa.WelsGetMemoryUsage() ); cTestMa.WelsFree( pUnitTestData, strUnitTestTag ); EXPECT_EQ( 0, cTestMa.WelsGetMemoryUsage() ); } } }