diff options
Diffstat (limited to 'spec/models/clusters')
-rw-r--r-- | spec/models/clusters/agent_token_spec.rb | 39 |
1 files changed, 35 insertions, 4 deletions
diff --git a/spec/models/clusters/agent_token_spec.rb b/spec/models/clusters/agent_token_spec.rb index 41f8215b713..bc158fc9117 100644 --- a/spec/models/clusters/agent_token_spec.rb +++ b/spec/models/clusters/agent_token_spec.rb @@ -2,7 +2,7 @@ require 'spec_helper' -RSpec.describe Clusters::AgentToken do +RSpec.describe Clusters::AgentToken, feature_category: :deployment_management do it { is_expected.to belong_to(:agent).class_name('Clusters::Agent').required } it { is_expected.to belong_to(:created_by_user).class_name('User').optional } it { is_expected.to validate_length_of(:description).is_at_most(1024) } @@ -12,8 +12,9 @@ RSpec.describe Clusters::AgentToken do it_behaves_like 'having unique enum values' describe 'scopes' do + let_it_be(:agent) { create(:cluster_agent) } + describe '.order_last_used_at_desc' do - let_it_be(:agent) { create(:cluster_agent) } let_it_be(:token_1) { create(:cluster_agent_token, agent: agent, last_used_at: 7.days.ago) } let_it_be(:token_2) { create(:cluster_agent_token, agent: agent, last_used_at: nil) } let_it_be(:token_3) { create(:cluster_agent_token, agent: agent, last_used_at: 2.days.ago) } @@ -25,8 +26,8 @@ RSpec.describe Clusters::AgentToken do end describe 'status-related scopes' do - let!(:active_token) { create(:cluster_agent_token) } - let!(:revoked_token) { create(:cluster_agent_token, :revoked) } + let!(:active_token) { create(:cluster_agent_token, agent: agent) } + let!(:revoked_token) { create(:cluster_agent_token, :revoked, agent: agent) } describe '.with_status' do context 'when filtering by active status' do @@ -48,6 +49,29 @@ RSpec.describe Clusters::AgentToken do it { is_expected.to contain_exactly(active_token) } end end + + describe '.connected' do + let!(:token) { create(:cluster_agent_token, agent: agent, status: status, last_used_at: last_used_at) } + + let(:status) { :active } + let(:last_used_at) { 2.minutes.ago } + + subject { described_class.connected } + + it { is_expected.to contain_exactly(token) } + + context 'when the token has not been used recently' do + let(:last_used_at) { 2.hours.ago } + + it { is_expected.to be_empty } + end + + context 'when the token is not active' do + let(:status) { :revoked } + + it { is_expected.to be_empty } + end + end end describe '#token' do @@ -64,6 +88,13 @@ RSpec.describe Clusters::AgentToken do agent_token = create(:cluster_agent_token) expect(agent_token.token.length).to be >= 50 end + + it 'has a prefix' do + agent_token = build(:cluster_agent_token, token_encrypted: nil) + agent_token.save! + + expect(agent_token.token).to start_with described_class::TOKEN_PREFIX + end end describe '#to_ability_name' do |