#include "visualdl/storage/storage.h"

#include <gtest/gtest.h>
#include <memory>

namespace visualdl {
class StorageTest : public ::testing::Test {
public:
  void SetUp() { storage.reset(new Storage(&data_)); }

  storage::Storage data_;
  std::unique_ptr<Storage> storage;
};

TEST_F(StorageTest, main) {
  storage->AddMode("train");
  storage->AddMode("test");

  auto tag0 = storage->AddTablet("tag0");
  auto tag1 = storage->AddTablet("tag1");

  auto modes = storage->Modes();
  ASSERT_EQ(modes.size(), 2);
  ASSERT_EQ(modes[0], "train");
  ASSERT_EQ(modes[1], "test");
}

}  // namespace visualdl