require_relative '../../../spec_helper_min'
require_relative '../../../support/helpers'
require_relative '../../../factories/organizations_contexts'

describe Carto::Superadmin::OrganizationsController do
  include HelperMethods

  let(:superadmin_headers) do
    credentials = Cartodb.config[:superadmin]
    {
      'HTTP_AUTHORIZATION' => ActionController::HttpAuthentication::Basic.encode_credentials(
        credentials['username'],
        credentials['password']
      ),
      'HTTP_ACCEPT' => "application/json"
    }
  end

  let(:invalid_headers) do
    {
      'HTTP_AUTHORIZATION' => ActionController::HttpAuthentication::Basic.encode_credentials('not', 'trusworthy'),
      'HTTP_ACCEPT' => "application/json"
    }
  end

  describe '#usage' do
    include_context 'organization with users helper'

    it 'fails without authorization' do
      get_json(usage_superadmin_user_url(@organization.id), {}, invalid_headers) do |response|
        response.status.should eq 401
      end
    end

    shared_examples_for 'dataservices usage metrics' do
      before(:each) do
        @date = Date.today
        usage_metrics = @class.new(@org_user_owner.username, @organization.name, MockRedis.new)
        @class.stubs(:new).returns(usage_metrics)
        usage_metrics.incr(@service, :success_responses, 10, @date)
        usage_metrics.incr(@service, :success_responses, 100, @date - 2)
        usage_metrics.incr(@service, :empty_responses, 20, @date - 2)
        usage_metrics.incr(@service, :failed_responses, 30, @date - 1)
        usage_metrics.incr(@service, :total_requests, 40, @date)
      end

      it 'returns usage metrics' do
        get_json(usage_superadmin_organization_url(@organization.id), { from: Date.today - 5 }, superadmin_headers) do |response|
          success = response.body[@service][:success_responses]
          success.find { |h| h[:date] == @date.to_s }[:value].should eq 10
          success.find { |h| h[:date] == (@date - 2).to_s }[:value].should eq 100

          empty = response.body[@service][:empty_responses]
          empty.find { |h| h[:date] == (@date - 2).to_s }[:value].should eq 20

          error = response.body[@service][:failed_responses]
          error.find { |h| h[:date] == (@date - 1).to_s }[:value].should eq 30

          total = response.body[@service][:total_requests]
          total.find { |h| h[:date] == @date.to_s }[:value].should eq 40
        end
      end

      it 'returns totals' do
        get_json(usage_superadmin_organization_url(@organization.id), { from: Date.today - 5, totals: true }, superadmin_headers) do |response|
          response.body[@service][:success_responses].should eq 110
          response.body[@service][:empty_responses].should eq 20
          response.body[@service][:failed_responses].should eq 30
          response.body[@service][:total_requests].should eq 40
        end
      end
    end

    describe 'geocoder_internal' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_internal
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'geocoder_here' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_here
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'geocoder_google' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_google
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'geocoder_cache' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_cache
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'geocoder_mapzen' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_mapzen
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'geocoder_mapbox' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_mapbox
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'geocoder_tomtom' do
      before(:all) do
        @class = CartoDB::GeocoderUsageMetrics
        @service = :geocoder_tomtom
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'here_isolines' do
      before(:all) do
        @class = CartoDB::IsolinesUsageMetrics
        @service = :here_isolines
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'mapzen_isolines' do
      before(:all) do
        @class = CartoDB::IsolinesUsageMetrics
        @service = :mapzen_isolines
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'mapbox_isolines' do
      before(:all) do
        @class = CartoDB::IsolinesUsageMetrics
        @service = :mapbox_isolines
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'tomtom_isolines' do
      before(:all) do
        @class = CartoDB::IsolinesUsageMetrics
        @service = :tomtom_isolines
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'obs_general' do
      before(:all) do
        @class = CartoDB::ObservatoryGeneralUsageMetrics
        @service = :obs_general
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'obs_snapshot' do
      before(:all) do
        @class = CartoDB::ObservatorySnapshotUsageMetrics
        @service = :obs_snapshot
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'routing_mapzen' do
      before(:all) do
        @class = CartoDB::RoutingUsageMetrics
        @service = :routing_mapzen
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'routing_mapbox' do
      before(:all) do
        @class = CartoDB::RoutingUsageMetrics
        @service = :routing_mapbox
      end

      it_behaves_like 'dataservices usage metrics'
    end

    describe 'routing_tomtom' do
      before(:all) do
        @class = CartoDB::RoutingUsageMetrics
        @service = :routing_tomtom
      end

      it_behaves_like 'dataservices usage metrics'
    end

    it 'returns mapviews' do
      key = CartoDB::Stats::APICalls.new.redis_api_call_key(@org_user_owner.username, 'mapviews')
      $users_metadata.ZADD(key, 23, "20160915")
      get_json(usage_superadmin_organization_url(@organization.id), { from: '2016-09-14' }, superadmin_headers) do |response|
        mapviews = response.body[:mapviews][:total_views]
        mapviews.find { |h| h[:date] == "2016-09-15" }[:value].should eq 23
      end
    end

    it 'returns Twitter imports' do
      st1 = SearchTweet.create(
        user_id: @org_user_owner.id,
        table_id: '96a86fb7-0270-4255-a327-15410c2d49d4',
        data_import_id: '96a86fb7-0270-4255-a327-15410c2d49d4',
        service_item_id: '555',
        retrieved_items: 42,
        state: ::SearchTweet::STATE_COMPLETE
      )
      st2 = SearchTweet.create(
        user_id: @org_user_1.id,
        table_id: '96a86fb7-0270-4255-a327-15410c2d49d4',
        data_import_id: '96a86fb7-0270-4255-a327-15410c2d49d4',
        service_item_id: '555',
        retrieved_items: 628,
        state: ::SearchTweet::STATE_COMPLETE
      )
      get_json(usage_superadmin_organization_url(@organization.id), { from: Date.today - 5 }, superadmin_headers) do |response|
        tweets = response.body[:twitter_imports][:retrieved_items]
        formatted_date = st1.created_at.to_date.to_s
        tweets.find { |h| h[:date] == formatted_date }[:value].should eq st1.retrieved_items + st2.retrieved_items
        tweets.find { |h| h[:date] == (Date.today - 5).to_s }[:value].should eq 0
      end
      st1.destroy
      st2.destroy
    end

    it 'does not return data outside the date range' do
      key = CartoDB::Stats::APICalls.new.redis_api_call_key(@org_user_owner.username, 'mapviews')
      $users_metadata.ZADD(key, 23, "20160915")
      get_json(usage_superadmin_organization_url(@organization.id), { from: '2016-09-16' }, superadmin_headers) do |response|
        mapviews = response.body[:mapviews][:total_views]
        mapviews.should_not include :"2016-09-15"
      end
    end

    it 'returns only requested services' do
      payload = { services: [:mapviews] }
      get_json(usage_superadmin_organization_url(@organization.id), payload, superadmin_headers) do |response|
        response.body.keys.should eq [:mapviews]
      end
    end

    it 'returns an error for invalid services array format' do
      payload = { services: 'wadus' }
      get_json(usage_superadmin_organization_url(@organization.id), payload, superadmin_headers) do |response|
        response.status.should eq 422
      end
    end

    it 'returns an error for invalid date format' do
      get_json(usage_superadmin_organization_url(@organization.id), { from: 'wadus' }, superadmin_headers) do |response|
        response.status.should eq 422
      end
    end

    it 'returns an error for unknown organization' do
      get_json(usage_superadmin_user_url(UUIDTools::UUID.random_create.to_s), {}, superadmin_headers) do |response|
        response.status.should eq 404
      end
    end
  end
end