// Copyright 2018 The Fuchsia 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 PERIDOT_BIN_CLOUD_PROVIDER_FIRESTORE_FIRESTORE_TESTING_TEST_FIRESTORE_SERVICE_H_
#define PERIDOT_BIN_CLOUD_PROVIDER_FIRESTORE_FIRESTORE_TESTING_TEST_FIRESTORE_SERVICE_H_

#include <string>

#include <google/firestore/v1beta1/document.pb.h>
#include <google/firestore/v1beta1/firestore.grpc.pb.h>
#include <lib/fit/function.h>

#include "peridot/bin/cloud_provider_firestore/firestore/firestore_service.h"

namespace cloud_provider_firestore {

struct GetDocumentRecord {
  google::firestore::v1beta1::GetDocumentRequest request;
  fit::function<void(grpc::Status, google::firestore::v1beta1::Document)>
      callback;
};

struct ListDocumentsRecord {
  google::firestore::v1beta1::ListDocumentsRequest request;
  fit::function<void(grpc::Status,
                     google::firestore::v1beta1::ListDocumentsResponse)>
      callback;
};

struct CreateDocumentRecord {
  google::firestore::v1beta1::CreateDocumentRequest request;
  fit::function<void(grpc::Status, google::firestore::v1beta1::Document)>
      callback;
};

struct DeleteDocumentRecord {
  google::firestore::v1beta1::DeleteDocumentRequest request;
  fit::function<void(grpc::Status)> callback;
};

struct CommitRecord {
  google::firestore::v1beta1::CommitRequest request;
  fit::function<void(grpc::Status, google::firestore::v1beta1::CommitResponse)>
      callback;
};

struct RunQueryRecord {
  google::firestore::v1beta1::RunQueryRequest request;
  fit::function<void(grpc::Status,
                     std::vector<google::firestore::v1beta1::RunQueryResponse>)>
      callback;
};

class TestFirestoreService : public FirestoreService {
 public:
  TestFirestoreService();
  ~TestFirestoreService() override;

  // FirestoreService:
  const std::string& GetDatabasePath() override;

  const std::string& GetRootPath() override;

  void GetDocument(
      google::firestore::v1beta1::GetDocumentRequest request,
      std::shared_ptr<grpc::CallCredentials> call_credentials,
      fit::function<void(grpc::Status, google::firestore::v1beta1::Document)>
          callback) override;

  void ListDocuments(
      google::firestore::v1beta1::ListDocumentsRequest request,
      std::shared_ptr<grpc::CallCredentials> call_credentials,
      fit::function<void(grpc::Status,
                         google::firestore::v1beta1::ListDocumentsResponse)>
          callback) override;

  void CreateDocument(
      google::firestore::v1beta1::CreateDocumentRequest request,
      std::shared_ptr<grpc::CallCredentials> call_credentials,
      fit::function<void(grpc::Status, google::firestore::v1beta1::Document)>
          callback) override;

  void DeleteDocument(google::firestore::v1beta1::DeleteDocumentRequest request,
                      std::shared_ptr<grpc::CallCredentials> call_credentials,
                      fit::function<void(grpc::Status)> callback) override;

  void RunQuery(google::firestore::v1beta1::RunQueryRequest request,
                std::shared_ptr<grpc::CallCredentials> call_credentials,
                fit::function<void(
                    grpc::Status,
                    std::vector<google::firestore::v1beta1::RunQueryResponse>)>
                    callback) override;

  void Commit(google::firestore::v1beta1::CommitRequest request,
              std::shared_ptr<grpc::CallCredentials> call_credentials,
              fit::function<void(grpc::Status,
                                 google::firestore::v1beta1::CommitResponse)>
                  callback) override;

  std::unique_ptr<ListenCallHandler> Listen(
      std::shared_ptr<grpc::CallCredentials> call_credentials,
      ListenCallClient* client) override;

  void ShutDown(fit::closure callback) override;

  std::vector<GetDocumentRecord> get_document_records;
  std::vector<ListDocumentsRecord> list_documents_records;
  std::vector<CreateDocumentRecord> create_document_records;
  std::vector<DeleteDocumentRecord> delete_document_records;
  std::vector<CommitRecord> commit_records;
  std::vector<RunQueryRecord> run_query_records;
  std::vector<ListenCallClient*> listen_clients;

  fit::closure shutdown_callback;

 private:
  const std::string db_path_;
  const std::string root_path_;
};

}  // namespace cloud_provider_firestore

#endif  // PERIDOT_BIN_CLOUD_PROVIDER_FIRESTORE_FIRESTORE_TESTING_TEST_FIRESTORE_SERVICE_H_