spec/invoker/power/url_rewriter_spec.rb in invoker-1.4.1 vs spec/invoker/power/url_rewriter_spec.rb in invoker-1.5.1

@@ -2,11 +2,22 @@
 
 describe Invoker::Power::UrlRewriter do
   let(:rewriter) { Invoker::Power::UrlRewriter.new }
 
   context "matching domain part of incoming request" do
-    it "should do foo.dev match" do
+    before(:all) do
+      @original_invoker_config = Invoker.config
+
+      Invoker.config = mock
+      Invoker.config.stubs(:tld).returns("dev")
+    end
+
+    after(:all) do
+      Invoker.config = @original_invoker_config
+    end
+
+    it "should match foo.dev" do
       match = rewriter.extract_host_from_domain("foo.dev")
       expect(match).to_not be_empty
 
       matching_string = match[0]
       expect(matching_string).to eq("foo")
@@ -31,8 +42,29 @@
     it "should match hello-world.dev" do
       match = rewriter.extract_host_from_domain("hello-world.dev")
       expect(match).to_not be_nil
 
       expect(match[0]).to eq("hello-world")
+    end
+
+    context 'user sets up a custom top level domain' do
+      before(:all) do
+        @original_invoker_config = Invoker.config
+
+        Invoker.config = mock
+        Invoker.config.stubs(:tld).returns("local")
+      end
+
+      it 'should match domain part of incoming request correctly' do
+        match = rewriter.extract_host_from_domain("foo.local")
+        expect(match).to_not be_empty
+
+        matching_string = match[0]
+        expect(matching_string).to eq("foo")
+      end
+
+      after(:all) do
+        Invoker.config = @original_invoker_config
+      end
     end
   end
 end