third_party/txt/tests/render_test.cc

/*
 * Copyright 2017 Google, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "render_test.h"

#include <string>

#include "lib/fxl/logging.h"
#include "third_party/skia/include/core/SkImageEncoder.h"
#include "third_party/skia/include/core/SkStream.h"
#include "txt/asset_font_manager.h"
#include "txt/directory_asset_data_provider.h"
#include "txt/font_collection.h"
#include "utils.h"

namespace txt {

RenderTest::RenderTest()
    : snapshots_(0), font_collection_(std::make_shared<FontCollection>()) {
  font_collection_->PushBack(sk_make_sp<AssetFontManager>(
      std::make_unique<txt::DirectoryAssetDataProvider>(GetFontDir())));
}

RenderTest::~RenderTest() = default;

SkCanvas* RenderTest::GetCanvas() {
  return canvas_ == nullptr ? nullptr : canvas_.get();
}

std::string RenderTest::GetNextSnapshotName() {
  const auto& test_info =
      ::testing::UnitTest::GetInstance()->current_test_info();

  std::stringstream stream;
  stream << "snapshots/" << test_info->test_case_name() << "_"
         << test_info->name();
  stream << "_" << ++snapshots_ << ".png";

  return stream.str();
}

bool RenderTest::Snapshot() {
  if (!canvas_ || !bitmap_) {
    return false;
  }
  std::string snapshot_dir = "snapshots";
  int error = 0;
// _WIN32 defined by Windows Visual compiler.
#if defined(_WIN32)
  // Handle windows path creation.
  error = _mkdir(snapshot_dir.c_str());
#else
  // Handle non-windows path creation with Unix permissions.
  mode_t permissions = 0733;
  error = mkdir(snapshot_dir.c_str(), permissions);
#endif
  if (error > 0) {
    FXL_LOG(ERROR) << "'snapshot/' Directory not available and could not be "
                      "created. Please create manually to save snapshot.";
    return false;
  }
  auto file_name = GetNextSnapshotName();
  SkFILEWStream file(file_name.c_str());
  return SkEncodeImage(&file, *bitmap_, SkEncodedImageFormat::kPNG, 100);
}

size_t RenderTest::GetTestCanvasWidth() const {
  return 1000;
}

size_t RenderTest::GetTestCanvasHeight() const {
  return 600;
}

void RenderTest::SetUp() {
  bitmap_ = std::make_unique<SkBitmap>();
  bitmap_->allocN32Pixels(GetTestCanvasWidth(), GetTestCanvasHeight());
  canvas_ = std::make_unique<SkCanvas>(*bitmap_);
  canvas_->clear(SK_ColorWHITE);
}

std::shared_ptr<FontCollection> RenderTest::GetTestFontCollection() const {
  return font_collection_;
}

void RenderTest::TearDown() {
  canvas_ = nullptr;
  bitmap_ = nullptr;
}

}  // namespace txt