GIF decoding: Add a unit test for repetition count

third_party/WebKit/Source/platform/graphics/BitmapImageTest.cpp

 /*
  * Copyright (c) 2013, Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 97 matching lines @@
         for (size_t i = 0; i < decodedFramesCount(); ++i)
             size += frameDecodedSize(i);
         return size;
     }
 
     void advanceAnimation()
     {
         m_image->advanceAnimation(0);
     }
 
+    int repetitionCount()
+    {
+        return m_image->repetitionCount(true);
+    }
+
+    int animationFinished()
+    {
+        return m_image->m_animationFinished;
+    }
+
     PassRefPtr<Image> imageForDefaultFrame()
     {
         return m_image->imageForDefaultFrame();
     }
 
 protected:
     void SetUp() override
     {
         DeferredImageDecoder::setEnabled(m_enableDeferredDecoding);
         m_imageObserver = adoptPtrWillBeNoop(new FakeImageObserver);
@@ skipping 32 matching lines @@
 TEST_F(BitmapImageTest, maybeAnimated)
 {
     loadImage("/LayoutTests/fast/images/resources/gif-loop-count.gif");
     for (size_t i = 0; i < frameCount(); ++i) {
         EXPECT_TRUE(m_image->maybeAnimated());
         advanceAnimation();
     }
     EXPECT_FALSE(m_image->maybeAnimated());
 }
 
+TEST_F(BitmapImageTest, animationRepetitions)
+{
+    loadImage("/LayoutTests/fast/images/resources/full2loop.gif");
+    int expectedRepetitionCount = 2;
+    EXPECT_EQ(expectedRepetitionCount, repetitionCount());
+
+    // We actually loop once more than stored repetition count.
+    for (int repeat = 0; repeat < expectedRepetitionCount + 1; ++repeat) {

[Peter Kasting, 2016/03/11 01:23:46]

Nit: repeat <= expectedRepetitionCount

[urvang, 2016/03/11 18:32:17]

Done

+        for (size_t i = 0; i < frameCount(); ++i) {
+            EXPECT_FALSE(animationFinished());
+            advanceAnimation();
+        }
+    }
+    EXPECT_TRUE(animationFinished());
+}
+
 TEST_F(BitmapImageTest, isAllDataReceived)
 {
     RefPtr<SharedBuffer> imageData = readFile("/LayoutTests/fast/images/resources/green.jpg");
     ASSERT_TRUE(imageData.get());
 
     RefPtr<BitmapImage> image = BitmapImage::create();
     EXPECT_FALSE(image->isAllDataReceived());
 
     image->setData(imageData, false);
     EXPECT_FALSE(image->isAllDataReceived());
@@ skipping 101 matching lines @@
     frameAtIndex(0);
 
     // Trying to destroy all data except an undecoded frame should go ahead and
     // destroy all other frames.
     setCurrentFrame(2);
     destroyDecodedData(false);
     EXPECT_EQ(-frameSize * 2, m_imageObserver->m_lastDecodedSizeChangedDelta);
 }
 
 } // namespace blink