Move cluster agent GraphQL types and resolvers to core

Changelog: added

ee/app/graphql/resolvers/clusters/agent_tokens_resolver.rb → app/graphql/resolvers/clusters/agent_tokens_resolver.rb

@@ -18,7 +18,7 @@ module Resolvers
       private
 
       def can_read_agent_tokens?
-        project.feature_available?(:cluster_agents) && current_user.can?(:admin_cluster, project)
+        current_user.can?(:admin_cluster, project)
       end
     end
   end

ee/app/graphql/resolvers/kas/agent_configurations_resolver.rb → app/graphql/resolvers/kas/agent_configurations_resolver.rb

@@ -21,7 +21,7 @@ module Resolvers
       private
 
       def can_read_agent_configuration?
-        project.licensed_feature_available?(:cluster_agents) && current_user.can?(:admin_cluster, project)
+        current_user.can?(:admin_cluster, project)
       end
 
       def kas_client

ee/app/graphql/resolvers/kas/agent_connections_resolver.rb → app/graphql/resolvers/kas/agent_connections_resolver.rb

@@ -24,7 +24,7 @@ module Resolvers
       private
 
       def can_read_connected_agents?
-        project.licensed_feature_available?(:cluster_agents) && current_user.can?(:admin_cluster, project)
+        current_user.can?(:admin_cluster, project)
       end
 
       def get_connected_agents

app/graphql/types/project_type.rb

@@ -361,6 +361,25 @@ module Types
           complexity: 5,
           resolver: ::Resolvers::TimelogResolver
 
+    field :agent_configurations,
+          ::Types::Kas::AgentConfigurationType.connection_type,
+          null: true,
+          description: 'Agent configurations defined by the project',
+          resolver: ::Resolvers::Kas::AgentConfigurationsResolver
+
+    field :cluster_agent,
+          ::Types::Clusters::AgentType,
+          null: true,
+          description: 'Find a single cluster agent by name.',
+          resolver: ::Resolvers::Clusters::AgentsResolver.single
+
+    field :cluster_agents,
+          ::Types::Clusters::AgentType.connection_type,
+          extras: [:lookahead],
+          null: true,
+          description: 'Cluster agents associated with the project.',
+          resolver: ::Resolvers::Clusters::AgentsResolver
+
     def label(title:)
       BatchLoader::GraphQL.for(title).batch(key: project) do |titles, loader, args|
         LabelsFinder

ee/app/graphql/ee/types/project_type.rb

@@ -95,25 +95,6 @@ module EE
               resolver: ::Resolvers::DastSiteValidationResolver,
               description: 'DAST Site Validations associated with the project.'
 
-        field :agent_configurations,
-              ::Types::Kas::AgentConfigurationType.connection_type,
-              null: true,
-              description: 'Agent configurations defined by the project',
-              resolver: ::Resolvers::Kas::AgentConfigurationsResolver
-
-        field :cluster_agent,
-              ::Types::Clusters::AgentType,
-              null: true,
-              description: 'Find a single cluster agent by name.',
-              resolver: ::Resolvers::Clusters::AgentsResolver.single
-
-        field :cluster_agents,
-              ::Types::Clusters::AgentType.connection_type,
-              extras: [:lookahead],
-              null: true,
-              description: 'Cluster agents associated with the project.',
-              resolver: ::Resolvers::Clusters::AgentsResolver
-
         field :repository_size_excess,
               GraphQL::FLOAT_TYPE,
               null: true,

ee/spec/graphql/types/project_type_spec.rb

@@ -19,7 +19,7 @@ RSpec.describe GitlabSchema.types['Project'] do
     expected_fields = %w[
       vulnerabilities vulnerability_scanners requirement_states_count
       vulnerability_severities_count packages compliance_frameworks vulnerabilities_count_by_day
-      security_dashboard_path iterations iteration_cadences cluster_agents repository_size_excess actual_repository_size_limit
+      security_dashboard_path iterations iteration_cadences repository_size_excess actual_repository_size_limit
       code_coverage_summary api_fuzzing_ci_configuration path_locks incident_management_escalation_policies
       incident_management_escalation_policy scan_execution_policies network_policies
     ]
@@ -101,139 +101,6 @@ RSpec.describe GitlabSchema.types['Project'] do
     end
   end
 
-  describe 'agent_configurations' do
-    let_it_be(:query) do
-      %(
-        query {
-          project(fullPath: "#{project.full_path}") {
-            agentConfigurations {
-              nodes {
-                agentName
-              }
-            }
-          }
-        }
-      )
-    end
-
-    let(:agent_name) { 'example-agent-name' }
-    let(:kas_client) { instance_double(Gitlab::Kas::Client, list_agent_config_files: [double(agent_name: agent_name)]) }
-
-    subject { GitlabSchema.execute(query, context: { current_user: user }).as_json }
-
-    before do
-      stub_licensed_features(cluster_agents: true)
-
-      project.add_maintainer(user)
-      allow(Gitlab::Kas::Client).to receive(:new).and_return(kas_client)
-    end
-
-    it 'returns configured agents' do
-      agents = subject.dig('data', 'project', 'agentConfigurations', 'nodes')
-
-      expect(agents.count).to eq(1)
-      expect(agents.first['agentName']).to eq(agent_name)
-    end
-  end
-
-  describe 'cluster_agents' do
-    let_it_be(:cluster_agent) { create(:cluster_agent, project: project, name: 'agent-name') }
-    let_it_be(:query) do
-      %(
-        query {
-          project(fullPath: "#{project.full_path}") {
-            clusterAgents {
-              count
-              nodes {
-                id
-                name
-                createdAt
-                updatedAt
-
-                project {
-                  id
-                }
-              }
-            }
-          }
-        }
-      )
-    end
-
-    subject { GitlabSchema.execute(query, context: { current_user: user }).as_json }
-
-    before do
-      stub_licensed_features(cluster_agents: true)
-
-      project.add_maintainer(user)
-    end
-
-    it 'returns associated cluster agents' do
-      agents = subject.dig('data', 'project', 'clusterAgents', 'nodes')
-
-      expect(agents.count).to be(1)
-      expect(agents.first['id']).to eq(cluster_agent.to_global_id.to_s)
-      expect(agents.first['name']).to eq('agent-name')
-      expect(agents.first['createdAt']).to be_present
-      expect(agents.first['updatedAt']).to be_present
-      expect(agents.first['project']['id']).to eq(project.to_global_id.to_s)
-    end
-
-    it 'returns count of cluster agents' do
-      count = subject.dig('data', 'project', 'clusterAgents', 'count')
-
-      expect(count).to be(project.cluster_agents.size)
-    end
-  end
-
-  describe 'cluster_agent' do
-    let_it_be(:cluster_agent) { create(:cluster_agent, project: project, name: 'agent-name') }
-    let_it_be(:agent_token) { create(:cluster_agent_token, agent: cluster_agent) }
-    let_it_be(:query) do
-      %(
-        query {
-          project(fullPath: "#{project.full_path}") {
-            clusterAgent(name: "#{cluster_agent.name}") {
-              id
-
-              tokens {
-                count
-                nodes {
-                  id
-                }
-              }
-            }
-          }
-        }
-      )
-    end
-
-    subject { GitlabSchema.execute(query, context: { current_user: user }).as_json }
-
-    before do
-      stub_licensed_features(cluster_agents: true)
-
-      project.add_maintainer(user)
-    end
-
-    it 'returns associated cluster agents' do
-      agent = subject.dig('data', 'project', 'clusterAgent')
-      tokens = agent.dig('tokens', 'nodes')
-
-      expect(agent['id']).to eq(cluster_agent.to_global_id.to_s)
-
-      expect(tokens.count).to be(1)
-      expect(tokens.first['id']).to eq(agent_token.to_global_id.to_s)
-    end
-
-    it 'returns count of agent tokens' do
-      agent = subject.dig('data', 'project', 'clusterAgent')
-      count = agent.dig('tokens', 'count')
-
-      expect(cluster_agent.agent_tokens.size).to be(count)
-    end
-  end
-
   describe 'code coverage summary field' do
     subject { described_class.fields['codeCoverageSummary'] }
 

ee/spec/graphql/resolvers/clusters/agent_tokens_resolver_spec.rb → spec/graphql/resolvers/clusters/agent_tokens_resolver_spec.rb

@@ -11,7 +11,6 @@ RSpec.describe Resolvers::Clusters::AgentTokensResolver do
   describe '#resolve' do
     let(:agent) { create(:cluster_agent) }
     let(:user) { create(:user, maintainer_projects: [agent.project]) }
-    let(:feature_available) { true }
     let(:ctx) { Hash(current_user: user) }
 
     let!(:matching_token1) { create(:cluster_agent_token, agent: agent, last_used_at: 5.days.ago) }
@@ -20,20 +19,10 @@ RSpec.describe Resolvers::Clusters::AgentTokensResolver do
 
     subject { resolve(described_class, obj: agent, ctx: ctx) }
 
-    before do
-      stub_licensed_features(cluster_agents: feature_available)
-    end
-
     it 'returns tokens associated with the agent, ordered by last_used_at' do
       expect(subject).to eq([matching_token2, matching_token1])
     end
 
-    context 'feature is not available' do
-      let(:feature_available) { false }
-
-      it { is_expected.to be_empty }
-    end
-
     context 'user does not have permission' do
       let(:user) { create(:user, developer_projects: [agent.project]) }
 

ee/spec/graphql/resolvers/clusters/agents_resolver_spec.rb → spec/graphql/resolvers/clusters/agents_resolver_spec.rb

@@ -21,10 +21,6 @@ RSpec.describe Resolvers::Clusters::AgentsResolver do
 
     let(:ctx) { { current_user: current_user } }
 
-    before do
-      stub_licensed_features(cluster_agents: true)
-    end
-
     subject { resolve_agents }
 
     context 'the current user has access to clusters' do

ee/spec/graphql/resolvers/kas/agent_configurations_resolver_spec.rb → spec/graphql/resolvers/kas/agent_configurations_resolver_spec.rb

@@ -14,7 +14,6 @@ RSpec.describe Resolvers::Kas::AgentConfigurationsResolver do
 
     let(:user) { create(:user, maintainer_projects: [project]) }
     let(:ctx) { Hash(current_user: user) }
-    let(:feature_available) { true }
 
     let(:agent1) { double }
     let(:agent2) { double }
@@ -23,8 +22,6 @@ RSpec.describe Resolvers::Kas::AgentConfigurationsResolver do
     subject { resolve(described_class, obj: project, ctx: ctx) }
 
     before do
-      stub_licensed_features(cluster_agents: feature_available)
-
       allow(Gitlab::Kas::Client).to receive(:new).and_return(kas_client)
     end
 
@@ -42,12 +39,6 @@ RSpec.describe Resolvers::Kas::AgentConfigurationsResolver do
       end
     end
 
-    context 'feature is not available' do
-      let(:feature_available) { false }
-
-      it { is_expected.to be_empty }
-    end
-
     context 'user does not have permission' do
       let(:user) { create(:user) }
 

ee/spec/graphql/resolvers/kas/agent_connections_resolver_spec.rb → spec/graphql/resolvers/kas/agent_connections_resolver_spec.rb

@@ -15,7 +15,6 @@ RSpec.describe Resolvers::Kas::AgentConnectionsResolver do
 
     let(:user) { create(:user, maintainer_projects: [project]) }
     let(:ctx) { Hash(current_user: user) }
-    let(:feature_available) { true }
 
     let(:connection1) { double(agent_id: agent1.id) }
     let(:connection2) { double(agent_id: agent1.id) }
@@ -30,8 +29,6 @@ RSpec.describe Resolvers::Kas::AgentConnectionsResolver do
     end
 
     before do
-      stub_licensed_features(cluster_agents: feature_available)
-
       allow(Gitlab::Kas::Client).to receive(:new).and_return(kas_client)
     end
 
@@ -60,12 +57,6 @@ RSpec.describe Resolvers::Kas::AgentConnectionsResolver do
       end
     end
 
-    context 'feature is not available' do
-      let(:feature_available) { false }
-
-      it { is_expected.to be_empty }
-    end
-
     context 'user does not have permission' do
       let(:user) { create(:user) }
 

spec/graphql/types/project_type_spec.rb

@@ -33,6 +33,7 @@ RSpec.describe GitlabSchema.types['Project'] do
       issue_status_counts terraform_states alert_management_integrations
       container_repositories container_repositories_count
       pipeline_analytics squash_read_only sast_ci_configuration
+      cluster_agent cluster_agents agent_configurations
       ci_template timelogs
     ]
 
@@ -458,4 +459,137 @@ RSpec.describe GitlabSchema.types['Project'] do
     it { is_expected.to have_graphql_type(Types::Ci::JobTokenScopeType) }
     it { is_expected.to have_graphql_resolver(Resolvers::Ci::JobTokenScopeResolver) }
   end
+
+  describe 'agent_configurations' do
+    let_it_be(:project) { create(:project) }
+    let_it_be(:user) { create(:user) }
+    let_it_be(:query) do
+      %(
+        query {
+          project(fullPath: "#{project.full_path}") {
+            agentConfigurations {
+              nodes {
+                agentName
+              }
+            }
+          }
+        }
+      )
+    end
+
+    let(:agent_name) { 'example-agent-name' }
+    let(:kas_client) { instance_double(Gitlab::Kas::Client, list_agent_config_files: [double(agent_name: agent_name)]) }
+
+    subject { GitlabSchema.execute(query, context: { current_user: user }).as_json }
+
+    before do
+      project.add_maintainer(user)
+      allow(Gitlab::Kas::Client).to receive(:new).and_return(kas_client)
+    end
+
+    it 'returns configured agents' do
+      agents = subject.dig('data', 'project', 'agentConfigurations', 'nodes')
+
+      expect(agents.count).to eq(1)
+      expect(agents.first['agentName']).to eq(agent_name)
+    end
+  end
+
+  describe 'cluster_agents' do
+    let_it_be(:project) { create(:project) }
+    let_it_be(:user) { create(:user) }
+    let_it_be(:cluster_agent) { create(:cluster_agent, project: project, name: 'agent-name') }
+    let_it_be(:query) do
+      %(
+        query {
+          project(fullPath: "#{project.full_path}") {
+            clusterAgents {
+              count
+              nodes {
+                id
+                name
+                createdAt
+                updatedAt
+
+                project {
+                  id
+                }
+              }
+            }
+          }
+        }
+      )
+    end
+
+    subject { GitlabSchema.execute(query, context: { current_user: user }).as_json }
+
+    before do
+      project.add_maintainer(user)
+    end
+
+    it 'returns associated cluster agents' do
+      agents = subject.dig('data', 'project', 'clusterAgents', 'nodes')
+
+      expect(agents.count).to be(1)
+      expect(agents.first['id']).to eq(cluster_agent.to_global_id.to_s)
+      expect(agents.first['name']).to eq('agent-name')
+      expect(agents.first['createdAt']).to be_present
+      expect(agents.first['updatedAt']).to be_present
+      expect(agents.first['project']['id']).to eq(project.to_global_id.to_s)
+    end
+
+    it 'returns count of cluster agents' do
+      count = subject.dig('data', 'project', 'clusterAgents', 'count')
+
+      expect(count).to be(project.cluster_agents.size)
+    end
+  end
+
+  describe 'cluster_agent' do
+    let_it_be(:project) { create(:project) }
+    let_it_be(:user) { create(:user) }
+    let_it_be(:cluster_agent) { create(:cluster_agent, project: project, name: 'agent-name') }
+    let_it_be(:agent_token) { create(:cluster_agent_token, agent: cluster_agent) }
+    let_it_be(:query) do
+      %(
+        query {
+          project(fullPath: "#{project.full_path}") {
+            clusterAgent(name: "#{cluster_agent.name}") {
+              id
+
+              tokens {
+                count
+                nodes {
+                  id
+                }
+              }
+            }
+          }
+        }
+      )
+    end
+
+    subject { GitlabSchema.execute(query, context: { current_user: user }).as_json }
+
+    before do
+      project.add_maintainer(user)
+    end
+
+    it 'returns associated cluster agents' do
+      agent = subject.dig('data', 'project', 'clusterAgent')
+      tokens = agent.dig('tokens', 'nodes')
+
+      expect(agent['id']).to eq(cluster_agent.to_global_id.to_s)
+
+      expect(tokens.count).to be(1)
+      expect(tokens.first['id']).to eq(agent_token.to_global_id.to_s)
+    end
+
+    it 'returns count of agent tokens' do
+      agent = subject.dig('data', 'project', 'clusterAgent')
+      count = agent.dig('tokens', 'count')
+
+      expect(cluster_agent.agent_tokens.size).to be(count)
+    end
+  end
 end