Set kubectl.kubernetes.io/restartedAt annotation on workload redeploy (#18496)

Fixes #16445

Co-authored-by: bender <bender@local>

Author: marcelofukumoto

shell/config/labels-annotations.js

@@ -2,6 +2,7 @@ export const NORMAN_NAME = 'field.cattle.io/name';
 export const DESCRIPTION = 'field.cattle.io/description';
 export const HOSTNAME = 'kubernetes.io/hostname';
 export const TIMESTAMP = 'cattle.io/timestamp';
+export const KUBECTL_RESTARTED_AT = 'kubectl.kubernetes.io/restartedAt';
 export const SYSTEM_NAMESPACE = 'management.cattle.io/system-namespace';
 export const PROJECT = 'field.cattle.io/projectId';
 export const DEFAULT_PROJECT = 'authz.management.cattle.io/default-project';

shell/dialog/RedeployWorkloadDialog.vue

@@ -1,5 +1,5 @@
 <script lang="ts">
-import { TIMESTAMP } from '@shell/config/labels-annotations';
+import { TIMESTAMP, KUBECTL_RESTARTED_AT } from '@shell/config/labels-annotations';
 import AsyncButton from '@shell/components/AsyncButton';
 import { Card } from '@components/Card';
 import { Banner } from '@components/Banner';
@@ -83,6 +83,7 @@ export default {
           const annotations = metadata.annotations ??= {};
 
           annotations[TIMESTAMP] = now;
+          annotations[KUBECTL_RESTARTED_AT] = now;
 
           await workload.save();
         }

shell/dialog/__tests__/RedeployWorkloadDialog.test.ts

@@ -0,0 +1,124 @@
+import { shallowMount } from '@vue/test-utils';
+import RedeployWorkloadDialog from '@shell/dialog/RedeployWorkloadDialog.vue';
+import { TIMESTAMP, KUBECTL_RESTARTED_AT } from '@shell/config/labels-annotations';
+
+const defaultStubs = {
+  Card:        true,
+  AsyncButton: true,
+  Banner:      true,
+};
+
+const defaultMocks = {
+  $store: {
+    getters: {
+      'i18n/t':            jest.fn((key: string) => key),
+      'type-map/labelFor': jest.fn(() => 'Deployment'),
+    },
+  },
+  t: jest.fn((key: string) => key),
+};
+
+function createWorkload(overrides = {}) {
+  return {
+    nameDisplay: 'my-workload',
+    type:        'apps.deployment',
+    schema:      { id: 'apps.deployment' },
+    spec:        { template: { metadata: { annotations: {} } } },
+    save:        jest.fn().mockResolvedValue(undefined),
+    ...overrides,
+  };
+}
+
+describe('component: RedeployWorkloadDialog', () => {
+  const createWrapper = (propsData = {}, mocks = {}) => {
+    return shallowMount(RedeployWorkloadDialog, {
+      propsData: {
+        workloads: [createWorkload()],
+        ...propsData,
+      },
+      global: {
+        mocks: {
+          ...defaultMocks,
+          ...mocks,
+        },
+        stubs:      defaultStubs,
+        directives: { 'clean-html': true },
+      },
+    });
+  };
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
+  it('should render correctly', () => {
+    const wrapper = createWrapper();
+
+    expect(wrapper.exists()).toBe(true);
+  });
+
+  describe('apply', () => {
+    it('should set both TIMESTAMP and kubectl restartedAt annotations', async() => {
+      const workload = createWorkload();
+      const wrapper = createWrapper({ workloads: [workload] });
+      const buttonDone = jest.fn();
+
+      await (wrapper.vm as any).apply(buttonDone);
+
+      const annotations = workload.spec.template.metadata.annotations;
+
+      expect(annotations[TIMESTAMP]).toBeDefined();
+      expect(annotations[KUBECTL_RESTARTED_AT]).toBeDefined();
+      expect(annotations[TIMESTAMP]).toStrictEqual(annotations[KUBECTL_RESTARTED_AT]);
+      expect(workload.save).toHaveBeenCalledWith();
+      expect(buttonDone).toHaveBeenCalledWith(true);
+    });
+
+    it('should set annotations as ISO timestamps without milliseconds', async() => {
+      const workload = createWorkload();
+      const wrapper = createWrapper({ workloads: [workload] });
+
+      await (wrapper.vm as any).apply();
+
+      const ts = workload.spec.template.metadata.annotations[KUBECTL_RESTARTED_AT];
+
+      expect(ts).toMatch(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}Z$/);
+    });
+
+    it('should initialize metadata and annotations when missing', async() => {
+      const workload = createWorkload({ spec: { template: {} } });
+      const wrapper = createWrapper({ workloads: [workload] });
+
+      await (wrapper.vm as any).apply();
+
+      const annotations = workload.spec.template.metadata.annotations;
+
+      expect(annotations[TIMESTAMP]).toBeDefined();
+      expect(annotations[KUBECTL_RESTARTED_AT]).toBeDefined();
+    });
+
+    it('should redeploy all workloads', async() => {
+      const workloads = [createWorkload(), createWorkload(), createWorkload()];
+      const wrapper = createWrapper({ workloads });
+
+      await (wrapper.vm as any).apply();
+
+      for (const w of workloads) {
+        expect(w.save).toHaveBeenCalledWith();
+        expect(w.spec.template.metadata.annotations[TIMESTAMP]).toBeDefined();
+        expect(w.spec.template.metadata.annotations[KUBECTL_RESTARTED_AT]).toBeDefined();
+      }
+    });
+
+    it('should report errors and call buttonDone(false) on failure', async() => {
+      const workload = createWorkload({ save: jest.fn().mockRejectedValue(new Error('save failed')) });
+      const wrapper = createWrapper({ workloads: [workload] });
+      const buttonDone = jest.fn();
+
+      await (wrapper.vm as any).apply(buttonDone);
+
+      expect(buttonDone).toHaveBeenCalledWith(false);
+      expect((wrapper.vm as any).errors.length).toBeGreaterThan(0);
+    });
+  });
+});