Added ImagePipe

mojo/services/gfx/images/cpp/image_pipe_apptest.cc

Index: mojo/services/gfx/images/cpp/image_pipe_apptest.cc
diff --git a/mojo/services/gfx/images/cpp/image_pipe_apptest.cc b/mojo/services/gfx/images/cpp/image_pipe_apptest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..bc2c9a0cb34945624413ecfd2298e2629f298d2d
--- /dev/null
+++ b/mojo/services/gfx/images/cpp/image_pipe_apptest.cc
@@ -0,0 +1,337 @@
+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "mojo/public/cpp/application/application_impl.h"
+#include "mojo/public/cpp/application/application_test_base.h"
+#include "mojo/public/cpp/environment/logging.h"
+#include "mojo/public/cpp/system/macros.h"
+#include "mojo/public/cpp/utility/run_loop.h"
+#include "mojo/services/gfx/images/cpp/image_pipe_consumer_endpoint.h"
+#include "mojo/services/gfx/images/cpp/image_pipe_producer_endpoint.h"
+
+namespace mojo {
+namespace {
+
+using namespace image_pipe;
+
+class ImagePipeApplicationTest : public test::ApplicationTestBase {
+ public:
+  ImagePipeApplicationTest() : ApplicationTestBase() {}
+  ~ImagePipeApplicationTest() override {}
+
+ private:
+  MOJO_DISALLOW_COPY_AND_ASSIGN(ImagePipeApplicationTest);
+};
+
+class TestImagePipe : private ImagePipeConsumerDelegate {
+ public:
+  TestImagePipe(bool producer_checked, bool consumer_checked);
+  ~TestImagePipe() override;
+  mojo::gfx::ImagePtr CreateTestImage();
+
+  ImagePipeProducerEndpoint* producer_endpoint;
+  ImagePipeConsumerEndpoint* consumer_endpoint;
+  mojo::gfx::SupportedImagePropertiesPtr supported_properties_;
+
+  bool HasExperiencedError() { return has_error_; }
+
+  std::function<void(mojo::gfx::ImagePtr image, uint32_t id)>
+      add_image_override;
+  std::function<void(uint32_t id)> remove_image_override;
+  std::function<void(uint32_t id)> present_image_override;
+
+ private:
+  // Inherited from ImagePipeConsumerDelegate //
+  void AddImage(mojo::gfx::ImagePtr image, uint32_t id) override;
+  void RemoveImage(uint32_t id) override;
+  void PresentImage(uint32_t id) override;
+  void HandleEndpointClosed() override {
+    has_error_ = true;
+    mojo::RunLoop::current()->Quit();
+  }
+  bool has_error_;
+};
+
+void TestImagePipe::AddImage(mojo::gfx::ImagePtr image, uint32_t id) {
+  add_image_override(image.Pass(), id);
+}
+void TestImagePipe::RemoveImage(uint32_t id) {
+  remove_image_override(id);
+}
+void TestImagePipe::PresentImage(uint32_t id) {
+  present_image_override(id);
+}
+
+TestImagePipe::TestImagePipe(bool producer_checked, bool consumer_checked) {
+  present_image_override = [](uint32_t id) {};
+  remove_image_override = [](uint32_t id) {};
+  add_image_override = [](mojo::gfx::ImagePtr image, uint32_t id) {};
+
+  has_error_ = false;
+  supported_properties_ = mojo::gfx::SupportedImageProperties::New();
+  supported_properties_->size = mojo::Size::New();
+  supported_properties_->size->width = 256;
+  supported_properties_->size->height = 256;
+
+  supported_properties_->formats =
+      mojo::Array<mojo::gfx::ColorFormatPtr>::New(0);
+  mojo::gfx::ColorFormatPtr format = mojo::gfx::ColorFormat::New();
+  format->layout = mojo::gfx::PixelLayout::BGRA_8888;
+  format->color_space = mojo::gfx::ColorSpace::SRGB;
+  supported_properties_->formats.push_back(format.Pass());
+
+  mojo::gfx::ImagePipePtr image_pipe_ptr;
+  consumer_endpoint = new ImagePipeConsumerEndpoint(
+      GetProxy(&image_pipe_ptr), supported_properties_.Clone(), this,
+      consumer_checked);
+
+  producer_endpoint =
+      new ImagePipeProducerEndpoint(image_pipe_ptr.Pass(), [this] {
+        has_error_ = true;
+        mojo::RunLoop::current()->Quit();
+      }, producer_checked);
+}
+
+TestImagePipe::~TestImagePipe() {
+  delete producer_endpoint;
+  delete consumer_endpoint;
+}
+
+mojo::gfx::ImagePtr TestImagePipe::CreateTestImage() {
+  mojo::MessagePipe pipe;
+
+  uint32_t bytes_per_pixel = 32;
+
+  mojo::gfx::ImageBufferPtr image_buffer = mojo::gfx::ImageBuffer::New();
+  image_buffer->size = supported_properties_->size->width *
+                       supported_properties_->size->width * bytes_per_pixel;
+  image_buffer->data = mojo::ScopedHandle(pipe.handle0.Pass());
+
+  mojo::gfx::ImagePtr image = mojo::gfx::Image::New();
+  image->buffer = image_buffer.Pass();
+  image->format = supported_properties_->formats[0].Clone();
+  image->size = supported_properties_->size.Clone();
+  image->pitch = supported_properties_->size->width;
+  image->stride = image->pitch * bytes_per_pixel;
+
+  return image;
+}
+
+// Tests that the usual flow for creating, adding, presenting, and removing
+// an image doesnt crash/break/cause-errors etc
+TEST_F(ImagePipeApplicationTest, NormalImageLifeCycle) {
+  TestImagePipe image_pipe(true, true);
+
+  image_pipe.present_image_override = [&image_pipe](uint32_t id) {
+    uint32_t acquired_id;
+    MOJO_CHECK(image_pipe.consumer_endpoint->AcquireNextImage(acquired_id));
+    MOJO_CHECK(acquired_id == id);
+    image_pipe.consumer_endpoint->ReleaseImage(
+        id, mojo::gfx::PresentationStatus::PRESENTED);
+  };
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id, [&image_pipe](unsigned int id, mojo::gfx::PresentationStatus status) {
+        EXPECT_EQ(mojo::gfx::PresentationStatus::PRESENTED, status);
+        mojo::RunLoop::current()->Quit();
+      });
+
+  mojo::RunLoop::current()->Run();
+  acquired_id = UINT32_MAX;
+  EXPECT_TRUE(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->RemoveImage(acquired_id);
+
+  EXPECT_FALSE(image_pipe.HasExperiencedError());
+}
+
+// Tests that flushing returns images to the producer with NOT_PRESENTED_FLUSHED
+TEST_F(ImagePipeApplicationTest, FlushImages) {
+  TestImagePipe image_pipe(true, true);
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id,
+      [&id](unsigned int presented_id, mojo::gfx::PresentationStatus status) {
+        EXPECT_EQ(mojo::gfx::PresentationStatus::NOT_PRESENTED_FLUSHED, status);
+        EXPECT_TRUE(presented_id == id);
+        mojo::RunLoop::current()->Quit();
+      });
+  image_pipe.producer_endpoint->FlushImages();
+
+  mojo::RunLoop::current()->Run();
+  acquired_id = UINT32_MAX;
+  EXPECT_TRUE(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  EXPECT_TRUE(acquired_id == id);
+
+  EXPECT_FALSE(image_pipe.HasExperiencedError());
+}
+
+// Tests that you can safely try to acquire an image when none are available,
+// and that you will safely fail
+TEST_F(ImagePipeApplicationTest, AcquireImageFromEmptyPool) {
+  TestImagePipe image_pipe(true, true);
+
+  uint32_t id = 0xDEADBEEF, acquired_id = id;
+  EXPECT_FALSE(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  EXPECT_TRUE(acquired_id == id);
+  EXPECT_FALSE(image_pipe.HasExperiencedError());
+}
+
+// Tests that adding an image with an existing ID causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ProducerError_AddImageWithReusedID) {
+  TestImagePipe image_pipe(false, true);
+
+  uint32_t id = 0;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that removing an image that hasnt been added causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ProducerError_RemoveImageBeforeAdded) {
+  TestImagePipe image_pipe(false, true);
+
+  uint32_t id = 0;
+  image_pipe.producer_endpoint->RemoveImage(id);
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that removing an image that has already been removed causes the pipe to
+// error (essentially that removing and image takes it out of the pool)
+TEST_F(ImagePipeApplicationTest, ProducerError_AddImageThenRemoveTwice) {
+  TestImagePipe image_pipe(false, true);
+
+  uint32_t id = 0;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  image_pipe.producer_endpoint->RemoveImage(id);
+  image_pipe.producer_endpoint->RemoveImage(id);
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that removing an image owned by the consumer causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ProducerError_RemoveImageOwnedByConsumer) {
+  TestImagePipe image_pipe(false, true);
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+  image_pipe.producer_endpoint->RemoveImage(id);
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that presenting an image that hasnt been added causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ProducerError_PresentImageNotAdded) {
+  TestImagePipe image_pipe(false, true);
+
+  uint32_t id = 0;
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that presenting an image that has already been presented causes the
+// pipe to error
+TEST_F(ImagePipeApplicationTest, ProducerError_PresentImageTwice) {
+  TestImagePipe image_pipe(false, true);
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that releasing an image that hasnt been added causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ConsumerError_ReleaseImageNotInPool) {
+  TestImagePipe image_pipe(true, false);
+
+  image_pipe.present_image_override = [&image_pipe](uint32_t id) {
+    image_pipe.consumer_endpoint->ReleaseImage(
+        id + 1, mojo::gfx::PresentationStatus::PRESENTED);
+  };
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that releasing an image before acquiring it causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ConsumerError_ReleaseBeforeAcquire) {
+  TestImagePipe image_pipe(true, false);
+
+  image_pipe.present_image_override = [&image_pipe](uint32_t id) {
+    image_pipe.consumer_endpoint->ReleaseImage(
+        id, mojo::gfx::PresentationStatus::PRESENTED);
+  };
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+// Tests that releasing an image before its presented causes the pipe to error
+TEST_F(ImagePipeApplicationTest, ConsumerError_ReleaseImageNotPresented) {
+  TestImagePipe image_pipe(true, false);
+
+  image_pipe.present_image_override = [&image_pipe](uint32_t id) {
+    uint32_t acquired_id;
+    MOJO_CHECK(image_pipe.consumer_endpoint->AcquireNextImage(acquired_id));
+    MOJO_CHECK(acquired_id == id);
+    image_pipe.consumer_endpoint->ReleaseImage(
+        id + 1, mojo::gfx::PresentationStatus::PRESENTED);
+  };
+
+  uint32_t id = 0, acquired_id = UINT32_MAX;
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id);
+  image_pipe.producer_endpoint->AddImage(image_pipe.CreateTestImage(), id + 1);
+  MOJO_CHECK(image_pipe.producer_endpoint->AcquireImage(acquired_id));
+  MOJO_CHECK(acquired_id == id);
+  image_pipe.producer_endpoint->PresentImage(
+      id, [](unsigned int, mojo::gfx::PresentationStatus) {});
+
+  mojo::RunLoop::current()->Run();
+  EXPECT_TRUE(image_pipe.HasExperiencedError());
+}
+
+}  // namespace image_pipe
+}  // namespace mojo