require "barnyard_harvester"
require "yaml"
require "mongo"
require "redis"
require "logger"
require "json"

CROP_NUMBER = 1

REDIS_SETTINGS = {
    :host => "localhost",
    :port => 6379,
    :db => CROP_NUMBER
}

MONGODB_SETTINGS = {
    :host => "localhost",
    :collection => "test_collection",
    :db => "test_database",
    :collection => "test_collection"
}

MONGODB_REPLICA_SET_SETTINGS = {
    :host => ["ip-172-19-31-44.c.qaapollogrp.edu:27017", "ip-172-19-30-49.c.qaapollogrp.edu:27017", "ip-172-19-31-202.c.qaapollogrp.edu:27017"],
    :collection => "test_collection",
    :db => "test_database",
    :user => "honeybadger",
    :password => "0joQuk35vJM05Hj",
    :collection => "test_collection"
}

$mongo_settings = MONGODB_SETTINGS

describe BarnyardHarvester do

  def load_and_process_file(file, backend)

    data = YAML::load_file file

    my_logger = Logger.new(STDOUT)
    my_logger.level = Logger::INFO

    h = BarnyardHarvester::Sync.new(:backend => backend,
                                    :queueing => :resque,
                                    :debug => false,
                                    :mongodb_settings => $mongo_settings,
                                    :crop_number => CROP_NUMBER,
                                    :redis_settings => REDIS_SETTINGS,
                                    :logger => my_logger)

    h.run do
      data.each do |primary_key, value|
        h.process primary_key, value
      end
    end

    h
  end

  def flush

    require "barnyard_harvester/mongodb_helper"

    my_logger = Logger.new(STDOUT)
    my_logger.level = Logger::INFO

    mongo = BarnyardHarvester::MongoDbHelper.connect $mongo_settings.merge(logger: my_logger)
    collection_name = $mongo_settings[:collection]
    collection = mongo.collection(collection_name)

    collection.find.each do |row|
      collection.remove("_id" => row["_id"])
    end

  end

  before(:each) do

    flush

    @crop_number = 1

    file = "spec/fixtures/data-init.yml"

    data = YAML::load_file file

    h = load_and_process_file(file, :mongodb)

    h.add_count.should eq(data.count)
    h.delete_count.should eq(0)
    h.change_count.should eq(0)
    h.source_count.should eq(data.count)
    h.cache_count.should eq(data.count)

  end

  it "test initial load of records" do

    data = YAML::load_file "spec/fixtures/data-init.yml"

    mongo = BarnyardHarvester::MongoDbHelper.connect $mongo_settings
    collection = mongo.collection($mongo_settings[:collection])

    data.each do |primary_key, value|

      doc = collection.find("_id" => primary_key).to_a[0]

      doc.delete("_id")

      value.should eq(doc)
    end

  end

  it "test add one record" do

    file = "spec/fixtures/data-add.yml"
    data = YAML::load_file file

    h = load_and_process_file(file, :mongodb)

    h.add_count.should eq(1)
    h.delete_count.should eq(0)
    h.change_count.should eq(0)
    h.source_count.should eq(data.count)
    h.cache_count.should eq(data.count)

    h.my_barn.log_run("#{file}-#{Random.rand(100)}", @crop_number, Time.now, Time.now, h.source_count, h.change_count, h.add_count, h.delete_count)

  end

  it "test change one record" do

    file = "spec/fixtures/data-change.yml"

    data = YAML::load_file file

    h = load_and_process_file(file, :mongodb)

    h.add_count.should eq(0)
    h.delete_count.should eq(0)
    h.change_count.should eq(1)
    h.source_count.should eq(data.count)
    h.cache_count.should eq(data.count)

    h.my_barn.log_run("#{file}-#{Random.rand(100)}", @crop_number, Time.now, Time.now, h.source_count, h.change_count, h.add_count, h.delete_count)

  end

  it "test delete one record" do

    file = "spec/fixtures/data-delete.yml"

    data = YAML::load_file file

    h = load_and_process_file(file, :mongodb)

    h.add_count.should eq(0)
    h.delete_count.should eq(1)
    h.change_count.should eq(0)
    h.source_count.should eq(data.count)
    h.cache_count.should eq(data.count + 1)

    h.my_barn.log_run("#{file}-#{Random.rand(100)}", @crop_number, Time.now, Time.now, h.source_count, h.change_count, h.add_count, h.delete_count)

  end

  it "test delete all records and add one" do

    init_file = "spec/fixtures/data-init.yml"
    init_data = YAML::load_file init_file

    file = "spec/fixtures/data-delete-all-records-add-one.yml"
    #data = YAML::load_file file

    h = load_and_process_file(file, :mongodb)

    h.add_count.should eq(1)
    h.delete_count.should eq(5)
    h.change_count.should eq(0)
    h.source_count.should eq(1)
    h.cache_count.should eq(init_data.count + 1)

    h.my_barn.log_run("#{file}-#{Random.rand(100)}", @crop_number, Time.now, Time.now, h.source_count, h.change_count, h.add_count, h.delete_count)

  end


  after(:each) do
  end


end