require 'spec_helper'
require 'flydata/compatibility_check'

module Flydata
  describe AgentCompatibilityCheck do
    let(:default_data_port) do
      {
        "servers"=>["sample-test-site.com"]
      }
     end

    describe "#check" do
      subject { AgentCompatibilityCheck.new("servers" => ['localhost']) }
      context "runs all check methods" do
        context "when all ports are accessible" do
          let(:sock) { double('sock') }
          before do
            allow(TCPSocket).to receive(:new).and_return(sock)
            allow(sock).to receive(:close)
          end
          it "does nothing" do
            subject.check
          end
        end
        context "when a port access fails" do
          before do
            allow(TCPSocket).to receive(:new).and_raise(Errno::ETIMEDOUT)
          end
          it do
            expect{subject.check_outgoing_ports}.to raise_error(Flydata::AgentCompatibilityCheck::AgentCompatibilityError, /ports/)
          end
        end
      end
    end
  end

  describe MysqlCompatibilityCheck do
    let(:default_mysql_cred) do
      {
        "host" => "test",
        "port" => 1234,
        "username" => "test",
        "password" => "password",
        "database" => "test_db"
      }
    end
    describe "#check_mysql_binlog_retention" do
      context "on on-premise mysql server" do
        subject { MysqlCompatibilityCheck.new({},:default_mysql_cred) }
        context "where retention is below limit" do
          let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).and_return([{"@@expire_logs_days"=>1}])
            allow(client).to receive(:close)
            allow(subject).to receive(:is_rds?).and_return(false)
          end
          it do
            expect{subject.check_mysql_binlog_retention}.to raise_error(Flydata::MysqlCompatibilityCheck::MysqlCompatibilityError, /expire_logs_days/)
          end
        end
        context "where retention is 0" do
          let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).and_return([{"@@expire_logs_days"=>0}])
            allow(subject).to receive(:is_rds?).and_return(false)
          end
          it do
            expect{subject.check_mysql_binlog_retention}
          end
        end
        context "where retention is above limit" do
          let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).and_return([{"@@expire_logs_days"=>11}])
            allow(subject).to receive(:is_rds?).and_return(false)
          end
          it do
            expect{subject.check_mysql_binlog_retention}
          end
        end
      end
      context "on RDS" do
        subject { MysqlCompatibilityCheck.new({},:default_mysql_cred) }
        context "where retention period is nil" do
          let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).with("SELECT @@expire_logs_days").and_return([{"@@expire_logs_days"=>0}])
            allow(client).to receive(:query).with("call mysql.rds_show_configuration;").and_return([{"name"=>"binlog retention hours", "value"=>nil}])
            allow(client).to receive(:close)
            allow(subject).to receive(:is_rds?).and_return(true)
          end
          it do
            expect{subject.check_mysql_binlog_retention}.to raise_error(Flydata::MysqlCompatibilityCheck::MysqlCompatibilityError, /rds_set_config/)
          end
        end
        context "where retention period is too low" do
           let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).with("SELECT @@expire_logs_days").and_return([{"@@expire_logs_days"=>0}])
            allow(client).to receive(:query).with("call mysql.rds_show_configuration;").and_return([{"name"=>"binlog retention hours", "value"=>4}])
            allow(client).to receive(:close)
            allow(subject).to receive(:is_rds?).and_return(true)
          end
          it do
            expect{subject.check_mysql_binlog_retention}.to raise_error(Flydata::MysqlCompatibilityCheck::MysqlCompatibilityError, /rds_set_config/)
          end
        end
        context "where retention period is over recommended limit" do
         let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).with("SELECT @@expire_logs_days").and_return([{"@@expire_logs_days"=>0}])
            allow(client).to receive(:query).with("call mysql.rds_show_configuration;").and_return([{"name"=>"binlog retention hours", "value"=>120}])
            allow(client).to receive(:close)
            allow(subject).to receive(:is_rds?).and_return(true)
          end
          it do
            expect{subject.check_mysql_binlog_retention}
          end
        end
        context "where user has no access to rds configuration" do
          let(:client) { double('client') }
          before do
            allow(Mysql2::Client).to receive(:new).and_return(client)
            allow(client).to receive(:query).with("SELECT @@expire_logs_days").and_return([{"@@expire_logs_days"=>0}])
            allow(client).to receive(:query).with("call mysql.rds_show_configuration;").and_raise(Mysql2::Error, "execute command denied to user")
            allow(client).to receive(:close)
            allow(subject).to receive(:is_rds?).and_return(true)
          end
          it do
            expect{subject.check_mysql_binlog_retention}
          end
        end
      end
    end
  end

end