canvas_test.h

// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef TESTING_CANVAS_TEST_H_
#define TESTING_CANVAS_TEST_H_

#include "flutter/fml/macros.h"
#include "flutter/testing/mock_canvas.h"
#include "gtest/gtest.h"

namespace flutter {
namespace testing {

// This fixture allows creating tests that make use of a mock |SkCanvas|.
template <typename BaseT>
class CanvasTestBase : public BaseT {
 public:
  CanvasTestBase() = default;

  MockCanvas& mock_canvas() { return canvas_; }
  SkColorSpace* mock_color_space() { return canvas_.imageInfo().colorSpace(); }
  SkNWayCanvas* mock_internal_canvas() { return canvas_.internal_canvas(); }

 private:
  MockCanvas canvas_;

  FML_DISALLOW_COPY_AND_ASSIGN(CanvasTestBase);
};
using CanvasTest = CanvasTestBase<::testing::Test>;

}  // namespace testing
}  // namespace flutter

#endif  // TESTING_CANVAS_TEST_H_