cartodb-4.42/spec/models/organization_shared_examples.rb

require_relative '../spec_helper'

# Tests should define the following methods:
# - get_organization: returns a correspoding Organization instance
# - get_twitter_imports_count_by_organization_id: returns organization import count. Needed because implementations don't share a common interface
shared_examples_for "organization models" do
  include_context 'users helper'
  include_context 'organization with users helper'

  describe "#get_geocoding_calls" do

    it "counts all geocodings within the org" do
      base_line = get_geocoding_calls_by_organization_id(@organization.id)

      get_organization.owner.geocoder_provider = 'heremaps'
      org_user_1_geocoder_metrics = CartoDB::GeocoderUsageMetrics.new(
        @org_user_1.username,
        @org_user_1.organization.name
      )
      org_user_1_geocoder_metrics.incr(:geocoder_here, :success_responses, 2)
      org_user_1_geocoder_metrics.incr(:geocoder_cache, :success_responses, 3)

      org_user_2_geocoder_metrics = CartoDB::GeocoderUsageMetrics.new(
        @org_user_2.username,
        @org_user_2.organization.name
      )
      org_user_2_geocoder_metrics.incr(:geocoder_here, :success_responses, 4)
      org_user_2_geocoder_metrics.incr(:geocoder_cache, :success_responses, 5)

      user1_geocoder_metrics = CartoDB::GeocoderUsageMetrics.new(@user1.username, nil)
      user1_geocoder_metrics.incr(:geocoder_here, :success_responses, 2)
      user1_geocoder_metrics.incr(:geocoder_cache, :success_responses, 3)

      ::User.any_instance.expects(:get_geocoding_calls).never
      get_geocoding_calls_by_organization_id(@organization.id).should == base_line + 14
    end

    it "counts all geocodings within the org using the org provider" do
      base_line = get_geocoding_calls_by_organization_id(@organization.id)

      get_organization.owner.geocoder_provider = 'tomtom'
      org_user_1_geocoder_metrics = CartoDB::GeocoderUsageMetrics.new(
        @org_user_1.username,
        @org_user_1.organization.name
      )
      org_user_1_geocoder_metrics.incr(:geocoder_here, :success_responses, 2)
      org_user_1_geocoder_metrics.incr(:geocoder_cache, :success_responses, 3)

      org_user_2_geocoder_metrics = CartoDB::GeocoderUsageMetrics.new(
        @org_user_2.username,
        @org_user_2.organization.name
      )
      org_user_2_geocoder_metrics.incr(:geocoder_here, :success_responses, 4)
      org_user_2_geocoder_metrics.incr(:geocoder_cache, :success_responses, 5)

      user1_geocoder_metrics = CartoDB::GeocoderUsageMetrics.new(@user1.username, nil)
      user1_geocoder_metrics.incr(:geocoder_here, :success_responses, 2)
      user1_geocoder_metrics.incr(:geocoder_cache, :success_responses, 3)

      ::User.any_instance.expects(:get_geocoding_calls).never
      get_geocoding_calls_by_organization_id(get_organization.id).should == base_line + 14
    end

  end

  describe "#signup_page_enabled" do
    it 'is true if domain whitelist is not empty' do
      organization = get_organization
      organization.auth_username_password_enabled = true
      organization.whitelisted_email_domains = ['carto.com']
      organization.signup_page_enabled.should == true
    end

    it 'is false if domain whitelist is empty' do
      organization = get_organization
      organization.auth_username_password_enabled = true
      organization.whitelisted_email_domains = []
      organization.signup_page_enabled.should == false
    end

    it 'is false if no authentication is enabled' do
      organization = get_organization
      organization.auth_username_password_enabled = false
      organization.auth_google_enabled = false
      organization.auth_github_enabled = false
      organization.whitelisted_email_domains = ['carto.com']
      organization.signup_page_enabled.should be_false
    end
  end

  it 'generates auth_tokens and save them for future accesses' do
    token = get_organization.get_auth_token
    token.should be
    get_organization.reload
    get_organization.get_auth_token.should eq token
  end
end