Fix SkCanvas::wouldOverwriteEntireSurface() contains test

tests/ImageTest.cpp

 /*
  * Copyright 2015 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "SkCanvas.h"
 #include "SkData.h"
 #include "SkDevice.h"
@@ skipping 112 matching lines @@
     const SkImageInfo dstInfo = SkImageInfo::MakeN32Premul(2, 2);
     const size_t dstRowBytes = 2 * sizeof(SkPMColor);
     SkPMColor pixels[4];
     memset(pixels, 0xFF, sizeof(pixels));   // init with values we don't expect
     image->readPixels(dstInfo, pixels, dstRowBytes, 0, 0);
     REPORTER_ASSERT(reporter, red == pixels[0]);
     REPORTER_ASSERT(reporter, green == pixels[1]);
     REPORTER_ASSERT(reporter, blue == pixels[2]);
     REPORTER_ASSERT(reporter, 0 == pixels[3]);
 }
+
+// Test that a draw that only partially covers the drawing surface isn't
+// interpreted as covering the entire drawing surface (i.e., exercise one of the
+// conditions of SkCanvas::wouldOverwriteEntireSurface()).
+DEF_TEST(Image_RetainSnapshot, reporter) {
+    const SkPMColor red   = SkPackARGB32(0xFF, 0xFF, 0, 0);
+    const SkPMColor green = SkPackARGB32(0xFF, 0, 0xFF, 0);
+    SkImageInfo info = SkImageInfo::MakeN32Premul(2, 2);
+    SkAutoTUnref<SkSurface> surface(SkSurface::NewRaster(info));
+    surface->getCanvas()->clear(0xFF00FF00);
+
+    SkPMColor pixels[4];
+    memset(pixels, 0xFF, sizeof(pixels));   // init with values we don't expect
+    const SkImageInfo dstInfo = SkImageInfo::MakeN32Premul(2, 2);
+    const size_t dstRowBytes = 2 * sizeof(SkPMColor);
+
+    SkAutoTUnref<SkImage> image1(surface->newImageSnapshot());
+    REPORTER_ASSERT(reporter, image1->readPixels(dstInfo, pixels, dstRowBytes, 0, 0));
+    for (size_t i = 0; i < SK_ARRAY_COUNT(pixels); ++i) {
+        REPORTER_ASSERT(reporter, pixels[i] == green);
+    }
+
+    SkPaint paint;
+    paint.setXfermodeMode(SkXfermode::kSrc_Mode);
+    paint.setColor(SK_ColorRED);
+
+    surface->getCanvas()->drawRect(SkRect::MakeXYWH(1, 1, 1, 1), paint);
+
+    SkAutoTUnref<SkImage> image2(surface->newImageSnapshot());
+    REPORTER_ASSERT(reporter, image2->readPixels(dstInfo, pixels, dstRowBytes, 0, 0));
+    REPORTER_ASSERT(reporter, pixels[0] == green);
+    REPORTER_ASSERT(reporter, pixels[1] == green);
+    REPORTER_ASSERT(reporter, pixels[2] == green);
+    REPORTER_ASSERT(reporter, pixels[3] == red);
+}