submodule/skia/tests/DiscardableMemoryTest.cpp

   1 /*
   2  * Copyright 2013 Google Inc.
   3  *
   4  * Use of this source code is governed by a BSD-style license that can be
   5  * found in the LICENSE file.
   6  */
   7
   8 #include "include/core/SkRefCnt.h"
   9 #include "src/core/SkDiscardableMemory.h"
  10 #include "src/lazy/SkDiscardableMemoryPool.h"
  11 #include "tests/Test.h"
  12
  13 #include <cstring>
  14 #include <memory>
  15
  16 namespace {
  17 constexpr char kTestString[] = "HELLO, WORLD!";
  18 constexpr size_t kTestStringLength = sizeof(kTestString);
  19 }  // namespace
  20
  21 static void test_dm(skiatest::Reporter* reporter,
  22                     SkDiscardableMemory* dm,
  23                     bool assertRelock) {
  24     REPORTER_ASSERT(reporter, dm);
  25     if (!dm) {
  26         return;
  27     }
  28     void* ptr = dm->data();
  29     REPORTER_ASSERT(reporter, ptr);
  30     if (!ptr) {
  31         return;
  32     }
  33     memcpy(ptr, kTestString, sizeof(kTestString));
  34     dm->unlock();
  35     bool relockSuccess = dm->lock();
  36     if (assertRelock) {
  37         REPORTER_ASSERT(reporter, relockSuccess);
  38     }
  39     if (!relockSuccess) {
  40         return;
  41     }
  42     ptr = dm->data();
  43     REPORTER_ASSERT(reporter, ptr);
  44     if (!ptr) {
  45         return;
  46     }
  47     REPORTER_ASSERT(reporter, 0 == memcmp(ptr, kTestString, kTestStringLength));
  48     dm->unlock();
  49 }
  50
  51 DEF_TEST(DiscardableMemory_global, reporter) {
  52     std::unique_ptr<SkDiscardableMemory> dm(SkDiscardableMemory::Create(kTestStringLength));
  53     // lock() test is allowed to fail, since other threads could be
  54     // using global pool.
  55     test_dm(reporter, dm.get(), false);
  56 }
  57
  58 DEF_TEST(DiscardableMemory_nonglobal, reporter) {
  59     sk_sp<SkDiscardableMemoryPool> pool(
  60         SkDiscardableMemoryPool::Make(1024));
  61     std::unique_ptr<SkDiscardableMemory> dm(pool->create(kTestStringLength));
  62     test_dm(reporter, dm.get(), true);
  63 }
  64