Allow session cookie handling to work with localhost

app/models/test_track/session.rb

@@ -95,10 +95,8 @@ def cookie_domain
   end
 
   def _cookie_domain
-    if bare_ip_address?
+    if bare_ip_address? || fully_qualified_cookie_domain_enabled?
       request.host
-    elsif fully_qualified_cookie_domain_enabled?
-      fully_qualified_domain
     else
       wildcard_domain
     end
@@ -108,16 +106,12 @@ def bare_ip_address?
     request.host.match(Resolv::AddressRegex)
   end
 
-  def fully_qualified_domain
-    public_suffix_host.name
-  end
-
   def wildcard_domain
-    "." + public_suffix_host.domain
+    "." + (public_suffix_domain || request.host)
   end
 
-  def public_suffix_host
-    @public_suffix_host ||= PublicSuffix.parse(request.host, default_rule: nil)
+  def public_suffix_domain
+    @public_suffix_domain ||= PublicSuffix.domain(request.host, default_rule: nil)
   end
 
   def manage_cookies!

lib/test_track_rails_client/version.rb

@@ -1,3 +1,3 @@
 module TestTrackRailsClient
-  VERSION = "4.0.0.alpha13" # rubocop:disable Style/MutableConstant
+  VERSION = "4.0.0.alpha14" # rubocop:disable Style/MutableConstant
 end

spec/models/test_track/session_spec.rb

@@ -235,16 +235,10 @@ def current_clown; end
         end
       end
 
-      it "checks for a valid domain" do
-        allow(request).to receive(:host).and_return("a.bad.actor;did-this<luzer>")
-        expect { subject.manage {} }.to raise_error PublicSuffix::DomainInvalid
-      end
-
-      it "checks for a valid domain when fully qualified cookie domains are enabled" do
-        with_env TEST_TRACK_FULLY_QUALIFIED_COOKIE_DOMAIN_ENABLED: 1 do
-          allow(request).to receive(:host).and_return("a.bad.actor;did-this<luzer>")
-          expect { subject.manage {} }.to raise_error PublicSuffix::DomainInvalid
-        end
+      it "works with localhost" do
+        allow(request).to receive(:host).and_return("localhost")
+        subject.manage {}
+        expect(cookies['tt_visitor_id'][:domain]).to eq ".localhost"
       end
 
       it "doesn't munge an IPv4 hostname" do