#include <stdlib.h>
#include "gtest/gtest.h"
#include "base/image_manager.h"
#include "image_loader/image_loader_stb.h"
#include <string>

using std::string;

TEST(ImageManager, basic) {
  bitmap_t bmp;
  memset(&bmp, 0x00, sizeof(bmp));
  ASSERT_EQ(image_manager_get_bitmap(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_get_bitmap(image_manager(), "not found", &bmp), RET_NOT_FOUND);
  ASSERT_EQ(image_manager_lookup(image_manager(), "unchecked", &bmp), RET_NOT_FOUND);

  ASSERT_EQ(image_manager_unload_unused(image_manager(), 0), RET_OK);
}

TEST(ImageManager, unload) {
  bitmap_t bmp;
  memset(&bmp, 0x00, sizeof(bmp));

  ASSERT_EQ(image_manager_get_bitmap(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_unload_bitmap(image_manager(), &bmp), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), "checked", &bmp), RET_NOT_FOUND);
}

TEST(ImageManager, unload_unused) {
  bitmap_t bmp;
  memset(&bmp, 0x00, sizeof(bmp));

  ASSERT_EQ(image_manager_get_bitmap(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_unload_unused(image_manager(), 0), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), "checked", &bmp), RET_NOT_FOUND);

  ASSERT_EQ(image_manager_unload_unused(image_manager(), 0), RET_OK);
}

TEST(ImageManager, add) {
  bitmap_t bmp;
  memset(&bmp, 0x00, sizeof(bmp));

  bmp.w = 10;
  bmp.h = 10;

  ASSERT_EQ(image_manager_add(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), "checked", &bmp), RET_OK);
  ASSERT_EQ(image_manager_unload_unused(image_manager(), 0), RET_OK);
}

TEST(ImageManager, locale) {
  bitmap_t bmp;
  memset(&bmp, 0x00, sizeof(bmp));
  image_manager_t* imm = image_manager_create();
  assets_manager_t* am = assets_manager_create(0);

  assets_manager_set_res_root(am, "tests/testdata");
  image_manager_set_assets_manager(imm, am);

  ASSERT_EQ(image_manager_get_bitmap(imm, "locale_$locale$", &bmp), RET_OK);
  ASSERT_EQ(string(bmp.name), string("locale_en"));

  ASSERT_EQ(image_manager_get_bitmap(imm, "locale1_$locale$", &bmp), RET_OK);
  ASSERT_EQ(string(bmp.name), string("locale1_en_US"));

  ASSERT_EQ(image_manager_get_bitmap(imm, "locale_$locale$", &bmp), RET_OK);
  ASSERT_EQ(string(bmp.name), string("locale_en"));

  ASSERT_EQ(image_manager_get_bitmap(imm, "locale1_$locale$", &bmp), RET_OK);
  ASSERT_EQ(string(bmp.name), string("locale1_en_US"));
  ASSERT_EQ(image_manager_unload_unused(imm, 0), RET_OK);

  assets_manager_destroy(am);
  image_manager_destroy(imm);
}

#ifdef WITH_FS_RES
TEST(ImageManager, fs) {
  bitmap_t bmp;
  const char* filename = "file://./demos/assets/default/raw/images/xx/flag_CN.png";

  memset(&bmp, 0x00, sizeof(bmp));
  ASSERT_EQ(image_manager_get_bitmap(image_manager(), filename, &bmp), RET_OK);
  ASSERT_EQ(image_manager_lookup(image_manager(), filename, &bmp), RET_OK);
  ASSERT_EQ(image_manager_unload_bitmap(image_manager(), &bmp), RET_OK);
}
#endif /*WITH_FS_RES*/