1 files changed, 4 insertions, 1 deletions
diff --git a/spec/frontend/clusters_list/components/agent_token_spec.js b/spec/frontend/clusters_list/components/agent_token_spec.js
index 7f6ec2eb3a2..cdd94d33545 100644
--- a/spec/frontend/clusters_list/components/agent_token_spec.js
+++ b/spec/frontend/clusters_list/components/agent_token_spec.js
@@ -8,6 +8,7 @@ import ModalCopyButton from '~/vue_shared/components/modal_copy_button.vue';
 
 const kasAddress = 'kas.example.com';
 const agentToken = 'agent-token';
+const kasVersion = '15.0.0';
 const modalId = INSTALL_AGENT_MODAL_ID;
 
 describe('InstallAgentModal', () => {
@@ -21,6 +22,7 @@ describe('InstallAgentModal', () => {
   const createWrapper = () => {
     const provide = {
       kasAddress,
+      kasVersion,
     };
 
     const propsData = {
@@ -59,7 +61,7 @@ describe('InstallAgentModal', () => {
     it('renders a copy button', () => {
       expect(findCopyButton().props()).toMatchObject({
         title: 'Copy command',
-        text: generateAgentRegistrationCommand(agentToken, kasAddress),
+        text: generateAgentRegistrationCommand(agentToken, kasAddress, kasVersion),
         modalId,
       });
     });
@@ -71,6 +73,7 @@ describe('InstallAgentModal', () => {
     it('shows code block with agent installation command', () => {
       expect(findCodeBlock().props('code')).toContain(`--set config.token=${agentToken}`);
       expect(findCodeBlock().props('code')).toContain(`--set config.kasAddress=${kasAddress}`);
+      expect(findCodeBlock().props('code')).toContain(`--set image.tag=v${kasVersion}`);
     });
   });
 });