From 4919b89ab83f8eee43b89ff4786eae46bf2fa30e Mon Sep 17 00:00:00 2001
From: "Akihiko Odaki (@fn_aki@pawoo.net)" <akihiko.odaki.4i@stu.hosei.ac.jp>
Date: Sat, 10 Jun 2017 16:44:02 +0900
Subject: [PATCH] Improve default language decision and spec (#3351)

* Improve default language decision

This change allows to takes account of accepted language determined by
the user agent even if the custom default locale of the instance is
configured.

* Cover Localized more

* Fix code style
---
 app/controllers/concerns/localized.rb       | 22 ++++--
 spec/controllers/concerns/localized_spec.rb | 88 +++++++++++++++++++++
 2 files changed, 105 insertions(+), 5 deletions(-)
 create mode 100644 spec/controllers/concerns/localized_spec.rb

diff --git a/app/controllers/concerns/localized.rb b/app/controllers/concerns/localized.rb
index 5501b6793be..a9ea60f7dc6 100644
--- a/app/controllers/concerns/localized.rb
+++ b/app/controllers/concerns/localized.rb
@@ -17,12 +17,24 @@ module Localized
   end
 
   def default_locale
-    ENV.fetch('DEFAULT_LOCALE') do
-      user_supplied_locale || I18n.default_locale
-    end
+    request_locale || env_locale || I18n.default_locale
   end
 
-  def user_supplied_locale
-    http_accept_language.language_region_compatible_from(I18n.available_locales)
+  def env_locale
+    ENV['DEFAULT_LOCALE']
+  end
+
+  def request_locale
+    preferred_locale || compatible_locale
+  end
+
+  def preferred_locale
+    http_accept_language.preferred_language_from([env_locale]) ||
+      http_accept_language.preferred_language_from(I18n.available_locales)
+  end
+
+  def compatible_locale
+    http_accept_language.compatible_language_from([env_locale]) ||
+      http_accept_language.compatible_language_from(I18n.available_locales)
   end
 end
diff --git a/spec/controllers/concerns/localized_spec.rb b/spec/controllers/concerns/localized_spec.rb
new file mode 100644
index 00000000000..18fe508e44b
--- /dev/null
+++ b/spec/controllers/concerns/localized_spec.rb
@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+describe ApplicationController, type: :controller do
+  controller do
+    include Localized
+
+    def success
+      head 200
+    end
+  end
+
+  before do
+    routes.draw { get 'success' => 'anonymous#success' }
+  end
+
+  shared_examples 'default locale' do
+    context 'when DEFAULT_LOCALE environment variable is set' do
+      around do |example|
+        ClimateControl.modify 'DEFAULT_LOCALE' => 'ca', &example.method(:run)
+      end
+
+      it 'sets language specified by ENV if preferred' do
+        request.headers['Accept-Language'] = 'ca, fa'
+        get 'success'
+        expect(I18n.locale).to eq :ca
+      end
+
+      it 'sets available and preferred language if language specified by ENV is not preferred' do
+        request.headers['Accept-Language'] = 'ca-ES, fa'
+        get 'success'
+        expect(I18n.locale).to eq :fa
+      end
+
+      it 'sets language specified by ENV if it is compatible and none of available languages are preferred' do
+        request.headers['Accept-Language'] = 'ca-ES, fa-IR'
+        get 'success'
+        expect(I18n.locale).to eq :ca
+      end
+
+      it 'sets available and compatible langauge if language specified by ENV is not compatible none of available languages are preferred' do
+        request.headers['Accept-Language'] = 'fa-IR'
+        get 'success'
+        expect(I18n.locale).to eq :fa
+      end
+
+      it 'sets language specified by ENV if none of available languages are compatible' do
+        request.headers['Accept-Language'] = ''
+        get 'success'
+        expect(I18n.locale).to eq :ca
+      end
+    end
+
+    context 'when DEFAULT_LOCALE environment variable is not set' do
+      it 'sets default locale if none of available languages are compatible' do
+        request.headers['Accept-Language'] = ''
+        get 'success'
+        expect(I18n.locale).to eq :en
+      end
+    end
+  end
+
+  context 'user with valid locale has signed in' do
+    it "sets user's locale" do
+      user = Fabricate(:user, locale: :ca)
+
+      sign_in(user)
+      get 'success'
+
+      expect(I18n.locale).to eq :ca
+    end
+  end
+
+  context 'user with invalid locale has signed in' do
+    before do
+      user = Fabricate.build(:user, locale: :invalid)
+      user.save!(validate: false)
+      sign_in(user)
+    end
+
+    include_examples 'default locale'
+  end
+
+  context 'user has not signed in' do
+    include_examples 'default locale'
+  end
+end