#ifndef BIERGARTEN_PIPELINE_DATA_GENERATION_MOCK_GENERATOR_H_
#define BIERGARTEN_PIPELINE_DATA_GENERATION_MOCK_GENERATOR_H_

#include "data_generation/data_generator.h"
#include <string>
#include <vector>

class MockGenerator final : public DataGenerator {
public:
  void Load(const std::string &model_path) override;
  BreweryResult GenerateBrewery(const std::string &city_name,
                                const std::string &country_name,
                                const std::string &region_context) override;
  UserResult GenerateUser(const std::string &locale) override;

private:
  static std::size_t DeterministicHash(const std::string &a,
                                       const std::string &b);

  static const std::vector<std::string> kBreweryAdjectives;
  static const std::vector<std::string> kBreweryNouns;
  static const std::vector<std::string> kBreweryDescriptions;
  static const std::vector<std::string> kUsernames;
  static const std::vector<std::string> kBios;
};

#endif  // BIERGARTEN_PIPELINE_DATA_GENERATION_MOCK_GENERATOR_H_