wovnrb 3.8.0 → 3.10.0

data/test/lib/api_translator_test.rb

@@ -1,215 +1,217 @@
-require 'test_helper'
-
-module Wovnrb
-  REQUEST_UUID = 'ABCD'.freeze
-
-  class ApiTranslatorTest < WovnMiniTest
-    def test_translate
-      assert_translation('test.html', 'test_translated.html', true)
-    end
-
-    def test_translate_falls_back_to_original_body_if_exception
-      Net::HTTP.any_instance.expects(:request).raises
-      assert_translation('test.html', 'test_translated.html', false, response: nil)
-    end
-
-    def test_translate_falls_back_to_original_body_if_api_error
-      assert_translation('test.html', 'test_translated.html', false, response: { encoding: 'text/json', status_code: 500 })
-    end
-
-    def test_translate_continues_if_api_response_is_not_compressed
-      assert_translation('test.html', 'test_translated.html', true, response: { encoding: 'unknown', status: 200 }, compress_data: false)
-    end
-
-    def test_translate_continues_if_api_response_is_compressed
-      assert_translation('test.html', 'test_translated.html', true, response: { encoding: 'unknown', status: 200 })
-    end
-
-    def test_translate_accepts_uncompressed_response_from_api_in_dev_mode
-      Wovnrb::Store.instance.update_settings('wovn_dev_mode' => true)
-      assert_translation('test.html', 'test_translated.html', true, response: { encoding: 'text/json', status: 200 }, compress_data: false)
-    end
-
-    def test_translate_without_api_compression_sends_json
-      Wovnrb::Store.instance.update_settings('compress_api_requests' => false)
-      sut, _store, _headers = create_sut
-      html_body = 'foo'
-
-      stub_request(:post, %r{http://wovn\.global\.ssl\.fastly\.net/v0/translation\?cache_key=.*})
-        .to_return(status: 200, body: { 'body' => 'translated_body' }.to_json)
-
-      sut.translate(html_body)
-
-      assert_requested :post, %r{http://wovn\.global\.ssl\.fastly\.net/v0/translation\?cache_key=.*},
-                       headers: {
-                         'Accept' => '*/*',
-                         'Accept-Encoding' => 'gzip',
-                         'Content-Type' => 'application/json',
-                         'User-Agent' => 'Ruby',
-                         'X-Request-Id' => REQUEST_UUID
-                       },
-                       body: {
-                         'url' => 'http://wovn.io/test',
-                         'token' => '123456',
-                         'lang_code' => 'fr',
-                         'url_pattern' => 'subdomain',
-                         'lang_param_name' => 'lang',
-                         'translate_canonical_tag' => true,
-                         'product' => 'WOVN.rb',
-                         'version' => VERSION,
-                         'body' => 'foo',
-                         'custom_lang_aliases' => { 'ja' => 'Japanese' }.to_json
-                       }.to_json,
-                       times: 1
-    end
-
-    def test_api_timeout_is_search_engine_user_higher_default
-      settings = {
-        'project_token' => '123456',
-        'custom_lang_aliases' => { 'ja' => 'Japanese' },
-        'default_lang' => 'en',
-        'url_pattern' => 'subdomain',
-        'url_pattern_reg' => '^(?<lang>[^.]+)\.',
-        'lang_param_name' => 'lang'
-      }
-      store = Wovnrb::Store.instance
-      store.update_settings(settings)
-      headers = Wovnrb::Headers.new(
-        Wovnrb.get_env('url' => 'http://fr.wovn.io/test', 'HTTP_USER_AGENT' => 'Mozilla/5.0 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)'),
-        Wovnrb.get_settings(settings),
-        UrlLanguageSwitcher.new(store)
-      )
-      api_translator = ApiTranslator.new(store, headers, REQUEST_UUID)
-      assert_equal(5.0, api_translator.send(:api_timeout))
-    end
-
-    def test_api_timeout_no_user_agent_use_normal_default
-      settings = {
-        'project_token' => '123456',
-        'custom_lang_aliases' => { 'ja' => 'Japanese' },
-        'default_lang' => 'en',
-        'url_pattern' => 'subdomain',
-        'url_pattern_reg' => '^(?<lang>[^.]+)\.',
-        'lang_param_name' => 'lang'
-      }
-      store = Wovnrb::Store.instance
-      store.update_settings(settings)
-      env = Wovnrb.get_env('url' => 'http://fr.wovn.io/test')
-      env.delete('HTTP_USER_AGENT')
-      headers = Wovnrb::Headers.new(
-        env,
-        Wovnrb.get_settings(settings),
-        UrlLanguageSwitcher.new(store)
-      )
-      api_translator = ApiTranslator.new(store, headers, REQUEST_UUID)
-      assert_equal(1.0, api_translator.send(:api_timeout))
-    end
-
-    private
-
-    def assert_translation(original_html_fixture, translated_html_fixture, success_expected, response: { encoding: 'gzip', status_code: 200 }, compress_data: true)
-      original_html = File.read("test/fixtures/html/#{original_html_fixture}")
-      translated_html = File.read("test/fixtures/html/#{translated_html_fixture}")
-      actual_translated_html = translate(original_html, translated_html, response, compress_data: compress_data)
-
-      if success_expected
-        assert_equal(actual_translated_html, translated_html)
-      else
-        assert_equal(actual_translated_html, original_html)
-      end
-    end
-
-    def translate(original_html, translated_html, response, compress_data: true)
-      api_translator, store, _headers = create_sut
-      translation_request_stub = stub_translation_api_request(store, original_html, translated_html, response, compress_data: compress_data)
-
-      expected_api_timeout = store.settings['api_timeout_seconds']
-      assert_equal(expected_api_timeout, api_translator.send(:api_timeout))
-      actual_translated_html = api_translator.translate(original_html)
-      assert_requested(translation_request_stub, times: 1) if translation_request_stub
-      actual_translated_html
-    end
-
-    def create_sut
-      settings = {
-        'project_token' => '123456',
-        'custom_lang_aliases' => { 'ja' => 'Japanese' },
-        'default_lang' => 'en',
-        'url_pattern' => 'subdomain',
-        'url_pattern_reg' => '^(?<lang>[^.]+)\.',
-        'lang_param_name' => 'lang'
-      }
-      store = Wovnrb::Store.instance
-      store.update_settings(settings)
-      headers = Wovnrb::Headers.new(
-        Wovnrb.get_env('url' => 'http://fr.wovn.io/test'),
-        Wovnrb.get_settings(settings),
-        UrlLanguageSwitcher.new(store)
-      )
-      api_translator = ApiTranslator.new(store, headers, REQUEST_UUID)
-
-      [api_translator, store, headers]
-    end
-
-    def stub_translation_api_request(store, original_html, translated_html, response, compress_data: true)
-      if response
-        cache_key = generate_cache_key(store, original_html)
-        api_host = if store.dev_mode?
-                     'dev-wovn.io:3001'
-                   else
-                     'wovn.global.ssl.fastly.net'
-                   end
-        api_url = "http://#{api_host}/v0/translation?cache_key=#{cache_key}"
-        compressed_data = compress(generate_data(original_html))
-        headers = {
-          'Accept' => '*/*',
-          'Accept-Encoding' => 'gzip',
-          'Content-Length' => compressed_data.bytesize,
-          'Content-Type' => 'application/json',
-          'Content-Encoding' => 'gzip',
-          'User-Agent' => 'Ruby'
-        }
-        stub_response_json = "{\"body\":\"#{translated_html.gsub("\n", '\n')}\"}"
-        stub_response = if compress_data
-                          compress(stub_response_json)
-                        else
-                          stub_response_json
-                        end
-        response_headers = { 'Content-Encoding' => response[:encoding] || 'gzip' }
-        stub_request(:post, api_url)
-          .with(body: compressed_data, headers: headers)
-          .to_return(status: response[:status_code] || 200, body: stub_response, headers: response_headers)
-
-      end
-    end
-
-    def generate_cache_key(store, original_html)
-      settings_hash = Digest::MD5.hexdigest(JSON.dump(store.settings))
-      body_hash = Digest::MD5.hexdigest(original_html)
-      escaped_key = CGI.escape("token=123456&settings_hash=#{settings_hash}&body_hash=#{body_hash}&path=/test&lang=fr&version=wovnrb_#{VERSION}")
-
-      "(#{escaped_key})"
-    end
-
-    def generate_data(original_html)
-      data = {
-        'url' => 'http://wovn.io/test',
-        'token' => '123456',
-        'lang_code' => 'fr',
-        'url_pattern' => 'subdomain',
-        'lang_param_name' => 'lang',
-        'translate_canonical_tag' => true,
-        'product' => 'WOVN.rb',
-        'version' => VERSION,
-        'body' => original_html,
-        'custom_lang_aliases' => '{"ja":"Japanese"}'
-      }
-
-      data.to_json
-    end
-
-    def compress(string)
-      ActiveSupport::Gzip.compress(string)
-    end
-  end
-end
+require 'test_helper'
+
+module Wovnrb
+  REQUEST_UUID = 'ABCD'.freeze
+
+  class ApiTranslatorTest < WovnMiniTest
+    def test_translate
+      assert_translation('test.html', 'test_translated.html', true)
+    end
+
+    def test_translate_falls_back_to_original_body_if_exception
+      Net::HTTP.any_instance.expects(:request).raises
+      assert_translation('test.html', 'test_translated.html', false, response: nil)
+    end
+
+    def test_translate_falls_back_to_original_body_if_api_error
+      assert_translation('test.html', 'test_translated.html', false, response: { encoding: 'text/json', status_code: 500 })
+    end
+
+    def test_translate_continues_if_api_response_is_not_compressed
+      assert_translation('test.html', 'test_translated.html', true, response: { encoding: 'unknown', status: 200 }, compress_data: false)
+    end
+
+    def test_translate_continues_if_api_response_is_compressed
+      assert_translation('test.html', 'test_translated.html', true, response: { encoding: 'unknown', status: 200 })
+    end
+
+    def test_translate_accepts_uncompressed_response_from_api_in_dev_mode
+      Wovnrb::Store.instance.update_settings('wovn_dev_mode' => true)
+      assert_translation('test.html', 'test_translated.html', true, response: { encoding: 'text/json', status: 200 }, compress_data: false)
+    end
+
+    def test_translate_without_api_compression_sends_json
+      Wovnrb::Store.instance.update_settings('compress_api_requests' => false)
+      sut, _store, _headers = create_sut
+      html_body = 'foo'
+
+      stub_request(:post, %r{http://wovn\.global\.ssl\.fastly\.net/v0/translation\?cache_key=.*})
+        .to_return(status: 200, body: { 'body' => 'translated_body' }.to_json)
+
+      sut.translate(html_body)
+
+      assert_requested :post, %r{http://wovn\.global\.ssl\.fastly\.net/v0/translation\?cache_key=.*},
+                       headers: {
+                         'Accept' => '*/*',
+                         'Accept-Encoding' => 'gzip',
+                         'Content-Type' => 'application/json',
+                         'User-Agent' => 'Ruby',
+                         'X-Request-Id' => REQUEST_UUID
+                       },
+                       body: {
+                         'url' => 'http://wovn.io/test',
+                         'token' => '123456',
+                         'lang_code' => 'fr',
+                         'url_pattern' => 'subdomain',
+                         'lang_param_name' => 'lang',
+                         'translate_canonical_tag' => true,
+                         'insert_hreflangs' => true,
+                         'product' => 'WOVN.rb',
+                         'version' => VERSION,
+                         'body' => 'foo',
+                         'custom_lang_aliases' => { 'ja' => 'Japanese' }.to_json
+                       }.to_json,
+                       times: 1
+    end
+
+    def test_api_timeout_is_search_engine_user_higher_default
+      settings = {
+        'project_token' => '123456',
+        'custom_lang_aliases' => { 'ja' => 'Japanese' },
+        'default_lang' => 'en',
+        'url_pattern' => 'subdomain',
+        'url_pattern_reg' => '^(?<lang>[^.]+)\.',
+        'lang_param_name' => 'lang'
+      }
+      store = Wovnrb::Store.instance
+      store.update_settings(settings)
+      headers = Wovnrb::Headers.new(
+        Wovnrb.get_env('url' => 'http://fr.wovn.io/test', 'HTTP_USER_AGENT' => 'Mozilla/5.0 (compatible; Googlebot/2.1; +http://www.google.com/bot.html)'),
+        Wovnrb.get_settings(settings),
+        UrlLanguageSwitcher.new(store)
+      )
+      api_translator = ApiTranslator.new(store, headers, REQUEST_UUID)
+      assert_equal(5.0, api_translator.send(:api_timeout))
+    end
+
+    def test_api_timeout_no_user_agent_use_normal_default
+      settings = {
+        'project_token' => '123456',
+        'custom_lang_aliases' => { 'ja' => 'Japanese' },
+        'default_lang' => 'en',
+        'url_pattern' => 'subdomain',
+        'url_pattern_reg' => '^(?<lang>[^.]+)\.',
+        'lang_param_name' => 'lang'
+      }
+      store = Wovnrb::Store.instance
+      store.update_settings(settings)
+      env = Wovnrb.get_env('url' => 'http://fr.wovn.io/test')
+      env.delete('HTTP_USER_AGENT')
+      headers = Wovnrb::Headers.new(
+        env,
+        Wovnrb.get_settings(settings),
+        UrlLanguageSwitcher.new(store)
+      )
+      api_translator = ApiTranslator.new(store, headers, REQUEST_UUID)
+      assert_equal(1.0, api_translator.send(:api_timeout))
+    end
+
+    private
+
+    def assert_translation(original_html_fixture, translated_html_fixture, success_expected, response: { encoding: 'gzip', status_code: 200 }, compress_data: true)
+      original_html = File.read("test/fixtures/html/#{original_html_fixture}")
+      translated_html = File.read("test/fixtures/html/#{translated_html_fixture}")
+      actual_translated_html = translate(original_html, translated_html, response, compress_data: compress_data)
+
+      if success_expected
+        assert_equal(actual_translated_html, translated_html)
+      else
+        assert_equal(actual_translated_html, original_html)
+      end
+    end
+
+    def translate(original_html, translated_html, response, compress_data: true)
+      api_translator, store, _headers = create_sut
+      translation_request_stub = stub_translation_api_request(store, original_html, translated_html, response, compress_data: compress_data)
+
+      expected_api_timeout = store.settings['api_timeout_seconds']
+      assert_equal(expected_api_timeout, api_translator.send(:api_timeout))
+      actual_translated_html = api_translator.translate(original_html)
+      assert_requested(translation_request_stub, times: 1) if translation_request_stub
+      actual_translated_html
+    end
+
+    def create_sut
+      settings = {
+        'project_token' => '123456',
+        'custom_lang_aliases' => { 'ja' => 'Japanese' },
+        'default_lang' => 'en',
+        'url_pattern' => 'subdomain',
+        'url_pattern_reg' => '^(?<lang>[^.]+)\.',
+        'lang_param_name' => 'lang'
+      }
+      store = Wovnrb::Store.instance
+      store.update_settings(settings)
+      headers = Wovnrb::Headers.new(
+        Wovnrb.get_env('url' => 'http://fr.wovn.io/test'),
+        Wovnrb.get_settings(settings),
+        UrlLanguageSwitcher.new(store)
+      )
+      api_translator = ApiTranslator.new(store, headers, REQUEST_UUID)
+
+      [api_translator, store, headers]
+    end
+
+    def stub_translation_api_request(store, original_html, translated_html, response, compress_data: true)
+      if response
+        cache_key = generate_cache_key(store, original_html)
+        api_host = if store.dev_mode?
+                     'dev-wovn.io:3001'
+                   else
+                     'wovn.global.ssl.fastly.net'
+                   end
+        api_url = "http://#{api_host}/v0/translation?cache_key=#{cache_key}"
+        compressed_data = compress(generate_data(original_html))
+        headers = {
+          'Accept' => '*/*',
+          'Accept-Encoding' => 'gzip',
+          'Content-Length' => compressed_data.bytesize,
+          'Content-Type' => 'application/json',
+          'Content-Encoding' => 'gzip',
+          'User-Agent' => 'Ruby'
+        }
+        stub_response_json = "{\"body\":\"#{translated_html.gsub("\n", '\n')}\"}"
+        stub_response = if compress_data
+                          compress(stub_response_json)
+                        else
+                          stub_response_json
+                        end
+        response_headers = { 'Content-Encoding' => response[:encoding] || 'gzip' }
+        stub_request(:post, api_url)
+          .with(body: compressed_data, headers: headers)
+          .to_return(status: response[:status_code] || 200, body: stub_response, headers: response_headers)
+
+      end
+    end
+
+    def generate_cache_key(store, original_html)
+      settings_hash = Digest::MD5.hexdigest(JSON.dump(store.settings))
+      body_hash = Digest::MD5.hexdigest(original_html)
+      escaped_key = CGI.escape("token=123456&settings_hash=#{settings_hash}&body_hash=#{body_hash}&path=/test&lang=fr&version=wovnrb_#{VERSION}")
+
+      "(#{escaped_key})"
+    end
+
+    def generate_data(original_html)
+      data = {
+        'url' => 'http://wovn.io/test',
+        'token' => '123456',
+        'lang_code' => 'fr',
+        'url_pattern' => 'subdomain',
+        'lang_param_name' => 'lang',
+        'translate_canonical_tag' => true,
+        'insert_hreflangs' => true,
+        'product' => 'WOVN.rb',
+        'version' => VERSION,
+        'body' => original_html,
+        'custom_lang_aliases' => '{"ja":"Japanese"}'
+      }
+
+      data.to_json
+    end
+
+    def compress(string)
+      ActiveSupport::Gzip.compress(string)
+    end
+  end
+end

data/test/lib/custom_domain/custom_domain_lang_test.rb

@@ -0,0 +1,85 @@
+require 'test_helper'
+require 'wovnrb/custom_domain/custom_domain_lang'
+
+module Wovnrb
+  class CustomDomainLangTest < WovnMiniTest
+    def setup
+      @custom_domain_root_path = CustomDomainLang.new('foo.com', '/', 'fr')
+      @custom_domain_with_path_no_trailing_slash = CustomDomainLang.new('foo.com', '/path', 'fr')
+      @custom_domain_with_path_trailing_slash = CustomDomainLang.new('foo.com', '/path/', 'fr')
+      @custom_domain_path_encoded_spaces = CustomDomainLang.new('foo.com', '/dir%20path', 'fr')
+    end
+
+    def test_custom_domain_lang_params
+      assert_equal('foo.com', @custom_domain_root_path.host)
+      assert_equal('/', @custom_domain_root_path.path)
+      assert_equal('fr', @custom_domain_root_path.lang)
+      assert_equal('foo.com', @custom_domain_root_path.host_and_path_without_trailing_slash)
+
+      assert_equal('foo.com', @custom_domain_with_path_no_trailing_slash.host)
+      assert_equal('/path/', @custom_domain_with_path_no_trailing_slash.path)
+      assert_equal('fr', @custom_domain_with_path_no_trailing_slash.lang)
+      assert_equal('foo.com/path', @custom_domain_with_path_no_trailing_slash.host_and_path_without_trailing_slash)
+
+      assert_equal('foo.com', @custom_domain_with_path_trailing_slash.host)
+      assert_equal('/path/', @custom_domain_with_path_trailing_slash.path)
+      assert_equal('fr', @custom_domain_with_path_trailing_slash.lang)
+      assert_equal('foo.com/path', @custom_domain_with_path_trailing_slash.host_and_path_without_trailing_slash)
+
+      assert_equal('foo.com', @custom_domain_path_encoded_spaces.host)
+      assert_equal('/dir%20path/', @custom_domain_path_encoded_spaces.path)
+      assert_equal('fr', @custom_domain_path_encoded_spaces.lang)
+      assert_equal('foo.com/dir%20path', @custom_domain_path_encoded_spaces.host_and_path_without_trailing_slash)
+    end
+
+    def test_is_match_with_different_domain
+      refute(@custom_domain_root_path.match?(Addressable::URI.parse('http://otherdomain.com/other/test.html')))
+    end
+
+    def test_is_match_with_different_port_number_should_be_ignored
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com:3000/other/test.html')))
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com:80/other/test.html')))
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com/other/test.html')))
+    end
+
+    def test_is_match_with_domain_containing_substring_should_be_false
+      refute(@custom_domain_root_path.match?(Addressable::URI.parse('http://en.foo.com/other/test.html')))
+    end
+
+    def test_is_match_with_same_domain_should_be_true
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com/other/test.html')))
+    end
+
+    def test_is_match_with_same_domain_different_casing_should_be_true
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com/other/test.html')))
+    end
+
+    def test_is_match_with_path_starts_with_custom_path_should_be_true
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com')))
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com/')))
+      assert(@custom_domain_root_path.match?(Addressable::URI.parse('http://foo.com/other/test.html?foo=bar')))
+
+      assert(@custom_domain_with_path_no_trailing_slash.match?(Addressable::URI.parse('http://foo.com/path')))
+      assert(@custom_domain_with_path_no_trailing_slash.match?(Addressable::URI.parse('http://foo.com/path/')))
+      assert(@custom_domain_with_path_no_trailing_slash.match?(Addressable::URI.parse('http://foo.com/path/other/test.html?foo=bar')))
+
+      assert(@custom_domain_with_path_trailing_slash.match?(Addressable::URI.parse('http://foo.com/path')))
+      assert(@custom_domain_with_path_trailing_slash.match?(Addressable::URI.parse('http://foo.com/path/')))
+      assert(@custom_domain_with_path_trailing_slash.match?(Addressable::URI.parse('http://foo.com/path/other/test.html?foo=bar')))
+
+      assert(@custom_domain_path_encoded_spaces.match?(Addressable::URI.parse('http://foo.com/dir%20path')))
+      assert(@custom_domain_path_encoded_spaces.match?(Addressable::URI.parse('http://foo.com/dir%20path?foo=bar')))
+    end
+
+    def test_is_match_with_path_matches_substring_should_be_false
+      refute(@custom_domain_with_path_no_trailing_slash.match?(Addressable::URI.parse('http://foo.com/pathsuffix/other/test.html')))
+      refute(@custom_domain_with_path_trailing_slash.match?(Addressable::URI.parse('http://foo.com/pathsuffix/other/test.html')))
+      refute(@custom_domain_path_encoded_spaces.match?(Addressable::URI.parse('http://foo.com/dir%20pathsuffix/other/test.html')))
+    end
+
+    def test_is_match_with_path_matches_custom_path_as_suffix_should_be_false
+      refute(@custom_domain_with_path_no_trailing_slash.match?(Addressable::URI.parse('http://foo.com/images/path/foo.png')))
+      refute(@custom_domain_with_path_trailing_slash.match?(Addressable::URI.parse('http://foo.com/images/path/foo.png')))
+    end
+  end
+end

data/test/lib/custom_domain/custom_domain_lang_url_handler_test.rb

@@ -0,0 +1,75 @@
+require 'test_helper'
+require 'wovnrb/custom_domain/custom_domain_langs'
+require 'wovnrb/custom_domain/custom_domain_lang_url_handler'
+
+module Wovnrb
+  class CustomDomainLangUrlHandlerTest < WovnMiniTest
+    def setup
+      custom_domain_langs_setting = {
+        'fr' => { 'url' => 'foo.com/' },
+        'ja' => { 'url' => 'foo.com/path' },
+        'zh-CHS' => { 'url' => 'foo.com/dir/path' },
+        'en' => { 'url' => 'english.foo.com/' },
+        'zh-Hant-HK' => { 'url' => 'zh-hant-hk.foo.com/zh' },
+        'pt' => { 'url' => '17797-trial2.foo.com/' }
+      }
+      @custom_domain_langs = CustomDomainLangs.new(custom_domain_langs_setting)
+    end
+
+    def test_add_custom_domain_lang_to_absolute_url
+      # apply to original lang
+      assert_equal('foo.com', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com', 'fr', @custom_domain_langs))
+      assert_equal('foo.com/path', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com', 'ja', @custom_domain_langs))
+      assert_equal('foo.com/dir/path', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com', 'zh-CHS', @custom_domain_langs))
+      assert_equal('english.foo.com', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com', 'en', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com', 'zh-Hant-HK', @custom_domain_langs))
+
+      # apply to target lang
+      assert_equal('foo.com', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('zh-hant-hk.foo.com/zh', 'fr', @custom_domain_langs))
+      assert_equal('foo.com/path', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('zh-hant-hk.foo.com/zh', 'ja', @custom_domain_langs))
+      assert_equal('foo.com/dir/path', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('zh-hant-hk.foo.com/zh', 'zh-CHS', @custom_domain_langs))
+      assert_equal('english.foo.com', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('zh-hant-hk.foo.com/zh', 'en', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('zh-hant-hk.foo.com/zh', 'zh-Hant-HK', @custom_domain_langs))
+
+      assert_equal('zh-hant-hk.foo.com/zh', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com/path', 'zh-Hant-HK', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh/', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com/path/', 'zh-Hant-HK', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh/index.html', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com/path/index.html', 'zh-Hant-HK', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh/path2/index.html', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com/path/path2/index.html', 'zh-Hant-HK', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh/path2/index.html?test=1', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com/path/path2/index.html?test=1', 'zh-Hant-HK', @custom_domain_langs))
+      assert_equal('zh-hant-hk.foo.com/zh/path2/index.html#hash', CustomDomainLangUrlHandler.add_custom_domain_lang_to_absolute_url('foo.com/path/path2/index.html#hash', 'zh-Hant-HK', @custom_domain_langs))
+    end
+
+    def test_change_to_new_custom_domain_lang
+      fr = @custom_domain_langs.custom_domain_lang_by_lang('fr')
+      ja = @custom_domain_langs.custom_domain_lang_by_lang('ja')
+      zh_chs = @custom_domain_langs.custom_domain_lang_by_lang('zh-CHS')
+      en = @custom_domain_langs.custom_domain_lang_by_lang('en')
+      zh_hant_hk = @custom_domain_langs.custom_domain_lang_by_lang('zh-Hant-HK')
+      pt = @custom_domain_langs.custom_domain_lang_by_lang('pt')
+
+      assert_equal('foo.com', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('foo.com', fr, fr))
+      assert_equal('foo.com/path', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('foo.com', fr, ja))
+      assert_equal('foo.com/dir/path', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('foo.com', fr, zh_chs))
+      assert_equal('english.foo.com', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('foo.com', fr, en))
+      assert_equal('zh-hant-hk.foo.com/zh', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('foo.com', fr, zh_hant_hk))
+
+      assert_equal('foo.com', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh', zh_hant_hk, fr))
+      assert_equal('foo.com/path', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh', zh_hant_hk, ja))
+      assert_equal('foo.com/dir/path', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh', zh_hant_hk, zh_chs))
+      assert_equal('english.foo.com', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh', zh_hant_hk, en))
+      assert_equal('zh-hant-hk.foo.com/zh', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh', zh_hant_hk, zh_hant_hk))
+
+      assert_equal('foo.com/path', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh', zh_hant_hk, ja))
+      assert_equal('foo.com/path/', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh/', zh_hant_hk, ja))
+      assert_equal('foo.com/path/index.html', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh/index.html', zh_hant_hk, ja))
+      assert_equal('foo.com/path/path', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh/path', zh_hant_hk, ja))
+      assert_equal('foo.com/path/path/index.html', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh/path/index.html', zh_hant_hk, ja))
+      assert_equal('foo.com/path/path/index.html?test=1', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh/path/index.html?test=1', zh_hant_hk, ja))
+      assert_equal('foo.com/path/path/index.html#hash', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zh/path/index.html#hash', zh_hant_hk, ja))
+
+      assert_equal('zh-hant-hk.foo.com/zhtrap', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('zh-hant-hk.foo.com/zhtrap', zh_hant_hk, ja))
+      assert_equal('english.foo.com/zhtrap', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('17797-trial2.foo.com/zhtrap', pt, en))
+      assert_equal('17797-trial2.foo.com/zhtrap', CustomDomainLangUrlHandler.change_to_new_custom_domain_lang('17797-trial2.foo.com/zhtrap', pt, pt))
+    end
+  end
+end