#ifndef DBS_SRELATION_TEST_FACTORY_H
#define DBS_SRELATION_TEST_FACTORY_H

#include "Algebras/DBService2/Relation.hpp"
#include "Algebras/DBService2/Node.hpp"
#include "Algebras/DBService2/NodeTestFactory.hpp"

namespace DBServiceTest {

  //TODO Some of these factory methods - without their default values - could
  //  be moved to Relation.
  class RelationTestFactory {
    public: 

    static std::shared_ptr<DBService::Relation> buildRelationWithOriginalNode(
      std::string recordDatabase, 
      std::string relationName = "relationWithOriginalNode",
      int originalNodePort = 4712,
      bool save = true) {
      
      // Relation
      std::shared_ptr<DBService::Relation> relation = DBService::Relation::build(recordDatabase, relationName);
      relation->setDatabase(recordDatabase);

      // Original Node
      std::shared_ptr<DBService::Node> node = std::make_shared<DBService::Node>(
        "localhost", originalNodePort, 
        "", "/home/doesnt_exist/secondo", 9941, 9942);
      node->setDatabase(recordDatabase);
      node->setType(DBService::Node::nodeTypeOriginal());      
      relation->setOriginalNode(node); 

      if (save == true)
        relation->save();

      return relation;
    }

    /*
      Builds an Relation with an original Node, a targetNode and a Replica.      
    */
    static std::shared_ptr<DBService::Relation> buildRelationWithReplica(
      std::string recordDatabase, 
      std::string relationName = "relationWithReplica",
      int originalNodePort = 4712,
      int targetNodePort = 4713, 
      bool save = true) {

      std::shared_ptr<DBService::Relation> relationWithReplica = 
        buildRelationWithOriginalNode(
          recordDatabase, relationName, originalNodePort, false);      

      // Target Node
      std::shared_ptr<DBService::Node> targetNode = NodeTestFactory::buildTargetNode(recordDatabase, targetNodePort);

      // Replica
      relationWithReplica->addReplica(targetNode);

      // Save the replica by saving the relation
      if (save == true)
        relationWithReplica->save();

      return relationWithReplica;
    }
  };
}

#endif