tastytrade 0.2.0

data/spec/tastytrade/cli_helpers_spec.rb

@@ -0,0 +1,248 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+require "tastytrade/cli_helpers"
+require "tastytrade/cli_config"
+
+RSpec.describe Tastytrade::CLIHelpers do
+  # Create a test class that includes the module
+  let(:test_class) do
+    Class.new do
+      include Tastytrade::CLIHelpers
+    end
+  end
+
+  let(:instance) { test_class.new }
+
+  describe "output helpers" do
+    describe "#error" do
+      it "outputs red error message to stderr" do
+        expect { instance.error("Something went wrong") }
+          .to output(/Error: Something went wrong/).to_stderr
+      end
+    end
+
+    describe "#warning" do
+      it "outputs yellow warning message to stderr" do
+        expect { instance.warning("Be careful") }
+          .to output(/Warning: Be careful/).to_stderr
+      end
+    end
+
+    describe "#success" do
+      it "outputs green success message with checkmark" do
+        expect { instance.success("Operation completed") }
+          .to output(/✓ Operation completed/).to_stdout
+      end
+    end
+
+    describe "#info" do
+      it "outputs cyan info message with arrow" do
+        expect { instance.info("Processing...") }
+          .to output(/→ Processing.../).to_stdout
+      end
+    end
+  end
+
+  describe "#format_currency" do
+    it "formats nil as $0.00" do
+      expect(instance.format_currency(nil)).to eq("$0.00")
+    end
+
+    it "formats zero as $0.00" do
+      expect(instance.format_currency(0)).to eq("$0.00")
+    end
+
+    it "formats positive values" do
+      expect(instance.format_currency(1234.56)).to eq("$1,234.56")
+    end
+
+    it "formats negative values" do
+      expect(instance.format_currency(-1234.56)).to eq("-$1,234.56")
+    end
+
+    it "formats small values" do
+      expect(instance.format_currency(0.99)).to eq("$0.99")
+    end
+
+    it "formats large values with commas" do
+      expect(instance.format_currency(1_234_567.89)).to eq("$1,234,567.89")
+    end
+  end
+
+  describe "#color_value" do
+    it "colors positive values green" do
+      result = instance.color_value(100)
+      expect(result).to include("$100.00")
+      # Pastel adds ANSI codes, so we check for presence of the value
+      expect(instance.pastel.strip(result)).to eq("$100.00")
+    end
+
+    it "colors negative values red" do
+      result = instance.color_value(-100)
+      expect(result).to include("$100.00")
+      expect(instance.pastel.strip(result)).to eq("-$100.00")
+    end
+
+    it "colors zero as dim" do
+      result = instance.color_value(0)
+      expect(instance.pastel.strip(result)).to eq("$0.00")
+    end
+
+    it "can format without currency" do
+      result = instance.color_value(42, format_as_currency: false)
+      expect(instance.pastel.strip(result)).to eq("42")
+    end
+  end
+
+  describe "#pastel" do
+    it "returns a Pastel instance" do
+      expect(instance.pastel).to be_a(Pastel::Delegator)
+    end
+
+    it "memoizes the instance" do
+      pastel1 = instance.pastel
+      pastel2 = instance.pastel
+      expect(pastel1).to be(pastel2)
+    end
+  end
+
+  describe "#prompt" do
+    it "returns a TTY::Prompt instance" do
+      expect(instance.prompt).to be_a(TTY::Prompt)
+    end
+
+    it "memoizes the instance" do
+      prompt1 = instance.prompt
+      prompt2 = instance.prompt
+      expect(prompt1).to be(prompt2)
+    end
+  end
+
+  describe "#config" do
+    it "returns a Config instance" do
+      expect(instance.config).to be_a(Tastytrade::CLIConfig)
+    end
+
+    it "memoizes the instance" do
+      config1 = instance.config
+      config2 = instance.config
+      expect(config1).to be(config2)
+    end
+  end
+
+  describe "authentication helpers" do
+    describe "#authenticated?" do
+      context "when no session exists" do
+        it "returns false" do
+          expect(instance.authenticated?).to be false
+        end
+      end
+    end
+
+    describe "#require_authentication!" do
+      context "when not authenticated" do
+        it "exits with error message" do
+          expect(instance).to receive(:exit).with(1)
+          expect { instance.require_authentication! }
+            .to output(/You must be logged in/).to_stderr
+        end
+
+        it "suggests login command" do
+          expect(instance).to receive(:exit).with(1)
+          expect { instance.require_authentication! }
+            .to output(/Run 'tastytrade login'/).to_stdout
+        end
+      end
+    end
+  end
+
+  describe "class methods" do
+    it "sets exit_on_failure? to true" do
+      expect(test_class.exit_on_failure?).to be true
+    end
+  end
+
+  describe "#current_account" do
+    let(:session) { instance_double(Tastytrade::Session) }
+    let(:config) { instance_double(Tastytrade::CLIConfig) }
+    let(:account) do
+      instance_double(Tastytrade::Models::Account,
+                      account_number: "5WX12345",
+                      nickname: "Test Account")
+    end
+
+    before do
+      allow(instance).to receive(:current_session).and_return(session)
+      allow(instance).to receive(:config).and_return(config)
+    end
+
+    context "when account number is saved" do
+      before do
+        allow(config).to receive(:get).with("current_account_number").and_return("5WX12345")
+        allow(Tastytrade::Models::Account).to receive(:get)
+          .with(session, "5WX12345").and_return(account)
+      end
+
+      it "returns the account" do
+        expect(instance.current_account).to eq(account)
+      end
+
+      it "caches the account" do
+        expect(Tastytrade::Models::Account).to receive(:get).once
+        2.times { instance.current_account }
+      end
+    end
+
+    context "when no account number is saved" do
+      before do
+        allow(config).to receive(:get).with("current_account_number").and_return(nil)
+      end
+
+      it "returns nil" do
+        expect(instance.current_account).to be_nil
+      end
+    end
+
+    context "when account fetch fails" do
+      before do
+        allow(config).to receive(:get).with("current_account_number").and_return("5WX12345")
+        allow(Tastytrade::Models::Account).to receive(:get)
+          .and_raise(StandardError, "Network error")
+      end
+
+      it "returns nil" do
+        expect { instance.current_account }.to output(/Failed to load current account/).to_stderr
+        expect(instance.current_account).to be_nil
+      end
+    end
+  end
+
+  describe "#current_account_number" do
+    let(:config) { instance_double(Tastytrade::CLIConfig) }
+
+    before do
+      allow(instance).to receive(:config).and_return(config)
+    end
+
+    context "when account number is saved" do
+      before do
+        allow(config).to receive(:get).with("current_account_number").and_return("5WX12345")
+      end
+
+      it "returns the account number" do
+        expect(instance.current_account_number).to eq("5WX12345")
+      end
+    end
+
+    context "when no account number is saved" do
+      before do
+        allow(config).to receive(:get).with("current_account_number").and_return(nil)
+      end
+
+      it "returns nil" do
+        expect(instance.current_account_number).to be_nil
+      end
+    end
+  end
+end

data/spec/tastytrade/cli_interactive_spec.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+require "tastytrade/cli"
+
+RSpec.describe "Tastytrade::CLI interactive command" do
+  let(:cli) { Tastytrade::CLI.new }
+  let(:session) { instance_double(Tastytrade::Session) }
+  let(:prompt) { instance_double(TTY::Prompt) }
+  let(:config) { instance_double(Tastytrade::CLIConfig) }
+
+  before do
+    allow(cli).to receive(:prompt).and_return(prompt)
+    allow(cli).to receive(:config).and_return(config)
+    allow(cli).to receive(:exit)
+    allow(config).to receive(:get).with("current_account_number").and_return(nil)
+  end
+
+  describe "#interactive" do
+    context "when authenticated" do
+      before do
+        allow(cli).to receive(:current_session).and_return(session)
+        allow(cli).to receive(:interactive_mode)
+      end
+
+      it "enters interactive mode" do
+        expect(cli).to receive(:interactive_mode)
+        cli.interactive
+      end
+    end
+
+    context "when not authenticated" do
+      before do
+        allow(cli).to receive(:current_session).and_return(nil)
+        allow(cli).to receive(:exit).with(1).and_raise(SystemExit)
+      end
+
+      it "requires authentication" do
+        expect { cli.interactive }.to raise_error(SystemExit)
+          .and output(/You must be logged in/).to_stderr
+      end
+
+      it "suggests login command" do
+        expect { cli.interactive }.to raise_error(SystemExit)
+          .and output(/Run 'tastytrade login'/).to_stdout
+      end
+
+      it "does not enter interactive mode" do
+        expect(cli).not_to receive(:interactive_mode)
+        expect { cli.interactive }.to raise_error(SystemExit)
+      end
+    end
+  end
+end

data/spec/tastytrade/cli_logout_spec.rb

@@ -0,0 +1,121 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+require "tastytrade/cli"
+
+RSpec.describe "Tastytrade::CLI logout command" do
+  let(:cli) { Tastytrade::CLI.new }
+  let(:config) { instance_double(Tastytrade::CLIConfig) }
+  let(:session_manager) { instance_double(Tastytrade::SessionManager) }
+
+  before do
+    allow(cli).to receive(:config).and_return(config)
+    allow(cli).to receive(:exit) # Prevent actual exit during tests
+  end
+
+  describe "#logout" do
+    context "with active session" do
+      before do
+        allow(config).to receive(:get).with("current_username").and_return("test@example.com")
+        allow(config).to receive(:get).with("environment").and_return("production")
+        allow(Tastytrade::SessionManager).to receive(:new).with(
+          username: "test@example.com",
+          environment: "production"
+        ).and_return(session_manager)
+      end
+
+      context "when logout succeeds" do
+        before do
+          allow(session_manager).to receive(:clear_session!).and_return(true)
+          allow(config).to receive(:delete)
+        end
+
+        it "clears session credentials" do
+          expect(session_manager).to receive(:clear_session!)
+          cli.logout
+        end
+
+        it "removes config entries" do
+          expect(config).to receive(:delete).with("current_username")
+          expect(config).to receive(:delete).with("environment")
+          expect(config).to receive(:delete).with("last_login")
+          cli.logout
+        end
+
+        it "displays success message" do
+          expect { cli.logout }.to output(/Successfully logged out/).to_stdout
+        end
+      end
+
+      context "when logout fails" do
+        before do
+          allow(session_manager).to receive(:clear_session!).and_return(false)
+        end
+
+        it "displays error message" do
+          expect(cli).to receive(:exit).with(1)
+          expect { cli.logout }.to output(/Failed to logout completely/).to_stderr
+        end
+
+        it "exits with status 1" do
+          expect(cli).to receive(:exit).with(1)
+          cli.logout
+        end
+      end
+    end
+
+    context "with sandbox environment" do
+      before do
+        allow(config).to receive(:get).with("current_username").and_return("test@example.com")
+        allow(config).to receive(:get).with("environment").and_return("sandbox")
+        allow(session_manager).to receive(:clear_session!).and_return(true)
+        allow(config).to receive(:delete)
+      end
+
+      it "creates session manager with sandbox environment" do
+        expect(Tastytrade::SessionManager).to receive(:new).with(
+          username: "test@example.com",
+          environment: "sandbox"
+        ).and_return(session_manager)
+        cli.logout
+      end
+    end
+
+    context "without active session" do
+      before do
+        allow(config).to receive(:get).with("current_username").and_return(nil)
+      end
+
+      it "displays warning message" do
+        expect { cli.logout }.to output(/No active session found/).to_stderr
+      end
+
+      it "doesn't attempt to clear session" do
+        expect(Tastytrade::SessionManager).not_to receive(:new)
+        cli.logout
+      end
+
+      it "doesn't exit with error" do
+        expect(cli).not_to receive(:exit)
+        cli.logout
+      end
+    end
+
+    context "with missing environment" do
+      before do
+        allow(config).to receive(:get).with("current_username").and_return("test@example.com")
+        allow(config).to receive(:get).with("environment").and_return(nil)
+        allow(session_manager).to receive(:clear_session!).and_return(true)
+        allow(config).to receive(:delete)
+      end
+
+      it "defaults to production environment" do
+        expect(Tastytrade::SessionManager).to receive(:new).with(
+          username: "test@example.com",
+          environment: "production"
+        ).and_return(session_manager)
+        cli.logout
+      end
+    end
+  end
+end

data/spec/tastytrade/cli_select_spec.rb

@@ -0,0 +1,174 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+require "tastytrade/cli"
+
+RSpec.describe "Tastytrade::CLI select command" do
+  let(:cli) { Tastytrade::CLI.new }
+  let(:config) { instance_double(Tastytrade::CLIConfig) }
+  let(:prompt) { instance_double(TTY::Prompt) }
+  let(:session) { instance_double(Tastytrade::Session) }
+  let(:account1) do
+    instance_double(Tastytrade::Models::Account,
+                    account_number: "5WX12345",
+                    nickname: "Main Account",
+                    account_type_name: "Margin")
+  end
+  let(:account2) do
+    instance_double(Tastytrade::Models::Account,
+                    account_number: "5WX67890",
+                    nickname: nil,
+                    account_type_name: "Cash")
+  end
+
+  before do
+    allow(cli).to receive(:config).and_return(config)
+    allow(cli).to receive(:prompt).and_return(prompt)
+    allow(cli).to receive(:exit) # Prevent actual exit during tests
+    allow(cli).to receive(:current_session).and_return(session)
+    allow(cli).to receive(:authenticated?).and_return(true)
+    allow(config).to receive(:get).with("current_account_number").and_return(nil)
+    allow(config).to receive(:set)
+  end
+
+  describe "#select" do
+    context "when not authenticated" do
+      before do
+        allow(cli).to receive(:authenticated?).and_return(false)
+      end
+
+      it "requires authentication" do
+        expect(cli).to receive(:require_authentication!)
+        allow(Tastytrade::Models::Account).to receive(:get_all).with(session).and_return([])
+        cli.select
+      end
+    end
+
+    context "with no accounts" do
+      before do
+        allow(Tastytrade::Models::Account).to receive(:get_all).with(session).and_return([])
+      end
+
+      it "returns early with warning" do
+        expect { cli.select }.to output(/No accounts found/).to_stderr
+      end
+
+      it "doesn't prompt for selection" do
+        expect(prompt).not_to receive(:select)
+        cli.select
+      end
+    end
+
+    context "with single account" do
+      before do
+        allow(Tastytrade::Models::Account).to receive(:get_all).with(session).and_return([account1])
+      end
+
+      it "auto-selects the account" do
+        expect(config).to receive(:set).with("current_account_number", "5WX12345")
+        cli.select
+      end
+
+      it "displays success message" do
+        expect { cli.select }.to output(/Using account: 5WX12345/).to_stdout
+      end
+
+      it "doesn't prompt for selection" do
+        expect(prompt).not_to receive(:select)
+        cli.select
+      end
+    end
+
+    context "with multiple accounts" do
+      before do
+        allow(Tastytrade::Models::Account).to receive(:get_all).with(session).and_return([account1, account2])
+      end
+
+      it "prompts for account selection" do
+        expect(prompt).to receive(:select).with(
+          "Choose an account:",
+          [
+            { name: "5WX12345 - Main Account (Margin)", value: "5WX12345" },
+            { name: "5WX67890 (Cash)", value: "5WX67890" }
+          ]
+        ).and_return("5WX12345")
+
+        cli.select
+      end
+
+      it "saves selected account" do
+        allow(prompt).to receive(:select).and_return("5WX67890")
+
+        expect(config).to receive(:set).with("current_account_number", "5WX67890")
+        cli.select
+      end
+
+      it "displays success message" do
+        allow(prompt).to receive(:select).and_return("5WX12345")
+
+        expect { cli.select }.to output(/Selected account: 5WX12345/).to_stdout
+      end
+
+      context "with current account selected" do
+        before do
+          allow(config).to receive(:get).with("current_account_number").and_return("5WX12345")
+        end
+
+        it "marks current account in choices" do
+          expect(prompt).to receive(:select).with(
+            "Choose an account:",
+            [
+              { name: "5WX12345 - Main Account (Margin) [current]", value: "5WX12345" },
+              { name: "5WX67890 (Cash)", value: "5WX67890" }
+            ]
+          ).and_return("5WX12345")
+
+          cli.select
+        end
+      end
+
+      context "with account without nickname" do
+        it "handles nil nickname properly" do
+          expect(prompt).to receive(:select).with(
+            "Choose an account:",
+            [
+              { name: "5WX12345 - Main Account (Margin)", value: "5WX12345" },
+              { name: "5WX67890 (Cash)", value: "5WX67890" }
+            ]
+          ).and_return("5WX67890")
+
+          cli.select
+        end
+      end
+    end
+
+    context "on API error" do
+      before do
+        allow(Tastytrade::Models::Account).to receive(:get_all).with(session)
+                                                               .and_raise(Tastytrade::Error, "Network error")
+      end
+
+      it "displays error message" do
+        expect(cli).to receive(:exit).with(1)
+        expect { cli.select }.to output(/Failed to fetch accounts: Network error/).to_stderr
+      end
+
+      it "exits with status 1" do
+        expect(cli).to receive(:exit).with(1)
+        cli.select
+      end
+    end
+
+    context "on unexpected error" do
+      before do
+        allow(Tastytrade::Models::Account).to receive(:get_all).with(session)
+                                                               .and_raise(StandardError, "Unexpected issue")
+      end
+
+      it "displays generic error message" do
+        expect(cli).to receive(:exit).with(1)
+        expect { cli.select }.to output(/Unexpected error: Unexpected issue/).to_stderr
+      end
+    end
+  end
+end