File: whoami_test.rb

package info (click to toggle)
vagrant 2.2.14%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 9,800 kB
  • sloc: ruby: 97,301; sh: 375; makefile: 16; lisp: 1
file content (110 lines) | stat: -rw-r--r-- 3,091 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
require File.expand_path("../../../../../base", __FILE__)

require Vagrant.source_root.join("plugins/commands/cloud/auth/whoami")

describe VagrantPlugins::CloudCommand::AuthCommand::Command::Whoami do
  include_context "unit"

  let(:argv)     { [] }
  let(:env) do
    # We have to create a Vagrantfile so there is a root path
    env = isolated_environment
    env.vagrantfile("")
    env.create_vagrant_env
  end
  let(:client) { double("client", token: token) }
  let(:token) { double("token") }
  let(:account_username) { "account-username" }
  let(:account) { double("account", username: account_username) }
  let(:action_runner) { double("action_runner") }

  subject { described_class.new(argv, env) }

  before do
    allow(env).to receive(:action_runner).and_return(action_runner)
    allow(VagrantPlugins::CloudCommand::Client).to receive(:new).and_return(client)
    allow(VagrantCloud::Account).to receive(:new).and_return(account)
  end

  describe "whoami" do
    context "when token is unset" do
      let(:token) { "" }

      it "should output an error" do
        expect(env.ui).to receive(:error)
        subject.whoami(token)
      end

      it "should return non-zero" do
        r = subject.whoami(token)
        expect(r).not_to eq(0)
        expect(r).to be_a(Integer)
      end
    end

    context "when token is set" do
      let(:token) { "my-token" }

      it "should load an account to validate" do
        expect(VagrantCloud::Account).to receive(:new).
          with(hash_including(access_token: token)).and_return(account)
        subject.whoami(token)
      end

      it "should output the account username" do
        expect(env.ui).to receive(:success).with(/#{account_username}/)
        subject.whoami(token)
      end

      it "should return zero value" do
        expect(subject.whoami(token)).to eq(0)
      end

      context "when error is encountered" do
        before { allow(VagrantCloud::Account).to receive(:new).and_raise(VagrantCloud::Error::ClientError) }

        it "should output an error" do
          expect(env.ui).to receive(:error).twice
          subject.execute
        end

        it "should return a non-zero value" do
          r = subject.execute
          expect(r).not_to eq(0)
          expect(r).to be_a(Integer)
        end
      end
    end
  end

  describe "#execute" do
    before do
      allow(subject).to receive(:whoami)
    end

    context "with too many arguments" do
      let(:argv) { ["token", "token", "token"] }
      it "shows help" do
        expect { subject.execute }.
          to raise_error(Vagrant::Errors::CLIInvalidUsage)
      end
    end

    context "with no argument" do
      it "should use stored token via client" do
        expect(subject).to receive(:whoami).with(token)
        subject.execute
      end
    end

    context "with token argument" do
      let(:token_arg) { "TOKEN_ARG" }
      let(:argv) { [token_arg] }

      it "should use the passed token" do
        expect(subject).to receive(:whoami).with(token_arg)
        subject.execute
      end
    end
  end
end