Fix memory leak in BitmapHeapTest

BUG=skia:

Review URL: https://codereview.chromium.org/715383003

1 files changed, 8 insertions, 2 deletions

diff --git a/tests/BitmapHeapTest.cpp b/tests/BitmapHeapTest.cpp
index f3385644bd..c7d097278b 100644
--- a/tests/BitmapHeapTest.cpp
+++ b/tests/BitmapHeapTest.cpp
@@ -17,9 +17,15 @@
 
 struct SimpleFlatController : public SkFlatController {
     SimpleFlatController() : SkFlatController() {}
-    virtual void* allocThrow(size_t bytes) SK_OVERRIDE { return sk_malloc_throw(bytes); }
-    virtual void unalloc(void* ptr) SK_OVERRIDE { sk_free(ptr); }
+    ~SimpleFlatController() { fAllocations.freeAll(); }
+    virtual void* allocThrow(size_t bytes) SK_OVERRIDE {
+        fAllocations.push(sk_malloc_throw(bytes));
+        return fAllocations.top();
+    }
+    virtual void unalloc(void*) SK_OVERRIDE { }
     void setBitmapStorage(SkBitmapHeap* h) { this->setBitmapHeap(h); }
+private:
+    SkTDArray<void*> fAllocations;
 };
 
 struct SkShaderTraits {