add runtime option to provide data->imagegenerator factory

BUG=skia:

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

1 files changed, 32 insertions, 0 deletions

diff --git a/tests/ImageGeneratorTest.cpp b/tests/ImageGeneratorTest.cpp
index 94867bc945..c856d8be60 100644
--- a/tests/ImageGeneratorTest.cpp
+++ b/tests/ImageGeneratorTest.cpp
@@ -5,9 +5,39 @@
  * found in the LICENSE file.
  */
 
+#include "SkData.h"
+#include "SkGraphics.h"
 #include "SkImageGenerator.h"
 #include "Test.h"
 
+static SkImageGenerator* my_factory(SkData* data) {
+    int* ptr = *(int**)data->data();
+    *ptr = 1;   // signal that we were called
+    return NULL;
+}
+
+static void test_imagegenerator_factory(skiatest::Reporter* reporter) {
+    int factoryHasBeenCalled = 0;
+    int* sentinelPtr = &factoryHasBeenCalled;
+    SkData* data = SkData::NewWithCopy(&sentinelPtr, sizeof(sentinelPtr));
+
+    SkImageGenerator* gen;
+    REPORTER_ASSERT(reporter, 0 == *sentinelPtr);
+
+    gen = SkImageGenerator::NewFromEncoded(data);
+    REPORTER_ASSERT(reporter, NULL == gen);
+    REPORTER_ASSERT(reporter, 0 == *sentinelPtr);
+
+    // Test is racy, in that it hopes no other thread is changing this global...
+    SkGraphics::ImageGeneratorFromEncodedFactory prev =
+                                                SkGraphics::GetImageGeneratorFromEncodedFactory();
+    SkGraphics::SetImageGeneratorFromEncodedFactory(my_factory);
+    gen = SkImageGenerator::NewFromEncoded(data);
+    REPORTER_ASSERT(reporter, NULL == gen);
+    REPORTER_ASSERT(reporter, 1 == *sentinelPtr);
+    SkGraphics::SetImageGeneratorFromEncodedFactory(prev);
+}
+
 class MyImageGenerator : public SkImageGenerator {
 public:
     MyImageGenerator() : SkImageGenerator(SkImageInfo::MakeN32Premul(0, 0)) {}
@@ -36,4 +66,6 @@ DEF_TEST(ImageGenerator, reporter) {
     rowBytes[0] = rowBytes[1] = rowBytes[2] = 250;
 
     ig.getYUV8Planes(sizes, planes, rowBytes, &colorSpace);
+
+    test_imagegenerator_factory(reporter);
 }