Add tests for headless helper modules

Cover TokenSession, ExpirationOptions, and ViewHelpers to bring
branch coverage above the 60% CI threshold.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: rameerez

test/helpers/expiration_options_test.rb

@@ -0,0 +1,101 @@
+# frozen_string_literal: true
+
+require "test_helper"
+
+module ApiKeys
+  module Helpers
+    class ExpirationOptionsTest < ApiKeys::Test
+      test "for_select returns default options" do
+        options = ExpirationOptions.for_select
+
+        assert_equal 6, options.length
+        assert_equal ["No Expiration", "no_expiration"], options.first
+        assert_equal ["7 days", "7_days"], options[1]
+        assert_equal ["1 year", "365_days"], options.last
+      end
+
+      test "for_select without no_expiration option" do
+        options = ExpirationOptions.for_select(include_no_expiration: false)
+
+        assert_equal 5, options.length
+        refute options.any? { |label, _| label == "No Expiration" }
+        assert_equal ["7 days", "7_days"], options.first
+      end
+
+      test "for_select with custom presets" do
+        options = ExpirationOptions.for_select(presets: [7, 30, 365])
+
+        assert_equal 4, options.length # includes "No Expiration"
+        assert_equal ["No Expiration", "no_expiration"], options.first
+        assert_equal ["7 days", "7_days"], options[1]
+        assert_equal ["30 days", "30_days"], options[2]
+        assert_equal ["1 year", "365_days"], options[3]
+      end
+
+      test "for_select with custom presets without no_expiration" do
+        options = ExpirationOptions.for_select(presets: [7, 30], include_no_expiration: false)
+
+        assert_equal 2, options.length
+        assert_equal ["7 days", "7_days"], options.first
+        assert_equal ["30 days", "30_days"], options.last
+      end
+
+      test "for_select generates correct label for 1 day" do
+        options = ExpirationOptions.for_select(presets: [1])
+
+        assert_equal ["1 day", "1_days"], options[1]
+      end
+
+      test "for_select generates correct label for multiple years" do
+        options = ExpirationOptions.for_select(presets: [730])
+
+        assert_equal ["2 years", "730_days"], options[1]
+      end
+
+      test "parse returns nil for no_expiration" do
+        result = ExpirationOptions.parse("no_expiration")
+
+        assert_nil result
+      end
+
+      test "parse returns nil for blank value" do
+        assert_nil ExpirationOptions.parse(nil)
+        assert_nil ExpirationOptions.parse("")
+      end
+
+      test "parse returns correct date for days preset" do
+        freeze_time do
+          result = ExpirationOptions.parse("30_days")
+
+          assert_in_delta 30.days.from_now.to_i, result.to_i, 1
+        end
+      end
+
+      test "parse handles singular day preset" do
+        freeze_time do
+          result = ExpirationOptions.parse("1_day")
+
+          assert_in_delta 1.day.from_now.to_i, result.to_i, 1
+        end
+      end
+
+      test "parse returns nil for invalid preset" do
+        assert_nil ExpirationOptions.parse("invalid")
+        assert_nil ExpirationOptions.parse("abc_days")
+        assert_nil ExpirationOptions.parse("0_days")
+      end
+
+      test "parse works with known presets" do
+        freeze_time do
+          result = ExpirationOptions.parse("365_days")
+
+          assert_in_delta 365.days.from_now.to_i, result.to_i, 1
+        end
+      end
+
+      test "default_value returns no_expiration" do
+        assert_equal "no_expiration", ExpirationOptions.default_value
+      end
+    end
+  end
+end

test/helpers/token_session_test.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+require "test_helper"
+
+module ApiKeys
+  module Helpers
+    class TokenSessionTest < ApiKeys::Test
+      def setup
+        @session = {}
+      end
+
+      test "stores token from api_key object with token method" do
+        api_key = mock("api_key")
+        api_key.stubs(:respond_to?).with(:token).returns(true)
+        api_key.stubs(:token).returns("sk_test_abc123")
+
+        result = TokenSession.store(@session, api_key)
+
+        assert_equal "sk_test_abc123", result
+        assert_equal "sk_test_abc123", @session[:api_keys_new_token]
+      end
+
+      test "stores token from object without token method using to_s" do
+        result = TokenSession.store(@session, "plain_token_string")
+
+        assert_equal "plain_token_string", result
+        assert_equal "plain_token_string", @session[:api_keys_new_token]
+      end
+
+      test "stores token with custom session key" do
+        api_key = mock("api_key")
+        api_key.stubs(:respond_to?).with(:token).returns(true)
+        api_key.stubs(:token).returns("custom_token")
+
+        TokenSession.store(@session, api_key, key: :my_custom_key)
+
+        assert_equal "custom_token", @session[:my_custom_key]
+        assert_nil @session[:api_keys_new_token]
+      end
+
+      test "retrieve_once returns and clears token" do
+        @session[:api_keys_new_token] = "stored_token"
+
+        result = TokenSession.retrieve_once(@session)
+
+        assert_equal "stored_token", result
+        assert_nil @session[:api_keys_new_token]
+      end
+
+      test "retrieve_once returns nil when no token stored" do
+        result = TokenSession.retrieve_once(@session)
+
+        assert_nil result
+      end
+
+      test "retrieve_once with custom key" do
+        @session[:my_key] = "my_token"
+
+        result = TokenSession.retrieve_once(@session, key: :my_key)
+
+        assert_equal "my_token", result
+        assert_nil @session[:my_key]
+      end
+
+      test "available? returns true when token present" do
+        @session[:api_keys_new_token] = "some_token"
+
+        assert TokenSession.available?(@session)
+      end
+
+      test "available? returns false when token not present" do
+        refute TokenSession.available?(@session)
+      end
+
+      test "available? returns false when token is blank" do
+        @session[:api_keys_new_token] = ""
+
+        refute TokenSession.available?(@session)
+      end
+
+      test "available? with custom key" do
+        @session[:custom] = "token"
+
+        assert TokenSession.available?(@session, key: :custom)
+        refute TokenSession.available?(@session, key: :other)
+      end
+    end
+  end
+end