@rancher/shell 1.2.3 → 2.0.1

package/edit/provisioning.cattle.io.cluster/__tests__/DirectoryConfig.test.ts

@@ -0,0 +1,228 @@
+/* eslint-disable jest/no-hooks */
+import { mount, Wrapper } from '@vue/test-utils';
+import DirectoryConfig from '@shell/edit/provisioning.cattle.io.cluster/tabs/DirectoryConfig.vue';
+import { _EDIT, _CREATE } from '@shell/config/query-params';
+import { clone } from '@shell/utils/object';
+
+describe('component: DirectoryConfig', () => {
+  let wrapper: Wrapper<InstanceType<typeof DirectoryConfig>>;
+
+  const mountOptions = {
+    propsData: {
+      value: {
+        systemAgent:  '',
+        provisioning: '',
+        k8sDistro:    '',
+      },
+      mode: _CREATE,
+    },
+    mocks: {
+      $store: {
+        getters: {
+          'i18n/t':      jest.fn(),
+          'i18n/exists': jest.fn(),
+        }
+      },
+    }
+  };
+
+  it('should render the component with its default configuration (create scenario)', () => {
+    wrapper = mount(
+      DirectoryConfig,
+      mountOptions
+    );
+
+    const title = wrapper.find('h3');
+    const checkbox = wrapper.find('[data-testid="rke2-directory-config-individual-config-checkbox"]');
+    const commonInput = wrapper.find('[data-testid="rke2-directory-config-common-data-dir"]');
+    const systemAgentInput = wrapper.find('[data-testid="rke2-directory-config-systemAgent-data-dir"]');
+    const provisioningInput = wrapper.find('[data-testid="rke2-directory-config-provisioning-data-dir"]');
+    const k8sDistroInput = wrapper.find('[data-testid="rke2-directory-config-k8sDistro-data-dir"]');
+
+    expect(title.exists()).toBe(true);
+    expect(checkbox.exists()).toBe(true);
+    // for the default config, checkbox should be checked
+    expect(wrapper.vm.isSettingCommonConfig).toBe(true);
+    expect(commonInput.exists()).toBe(true);
+
+    // since we have all of the vars empty, then the individual inputs should not be there
+    expect(systemAgentInput.exists()).toBe(false);
+    expect(provisioningInput.exists()).toBe(false);
+    expect(k8sDistroInput.exists()).toBe(false);
+  });
+
+  it('updating common config path should set the correct values on each data dir variable', async() => {
+    wrapper = mount(
+      DirectoryConfig,
+      mountOptions
+    );
+
+    const inputPath = 'some-data-dir';
+    const commonInput = wrapper.find('[data-testid="rke2-directory-config-common-data-dir"]');
+
+    commonInput.setValue(inputPath);
+    await wrapper.vm.$nextTick();
+
+    expect(wrapper.vm.value.systemAgent).toStrictEqual(inputPath);
+    expect(wrapper.vm.value.provisioning).toStrictEqual(inputPath);
+    expect(wrapper.vm.value.k8sDistro).toStrictEqual(inputPath);
+  });
+
+  it('updating each individual data dir should set the correct values on each data dir variable', async() => {
+    wrapper = mount(
+      DirectoryConfig,
+      mountOptions
+    );
+
+    const inputPath = 'some-data-dir';
+    const checkbox = wrapper.find('[data-testid="rke2-directory-config-individual-config-checkbox"]');
+
+    await checkbox.find('label').trigger('click');
+    await checkbox.vm.$nextTick();
+    await wrapper.vm.$nextTick();
+
+    expect(wrapper.vm.isSettingCommonConfig).toBe(false);
+
+    const systemAgentInput = wrapper.find('[data-testid="rke2-directory-config-systemAgent-data-dir"]');
+    const provisioningInput = wrapper.find('[data-testid="rke2-directory-config-provisioning-data-dir"]');
+    const k8sDistroInput = wrapper.find('[data-testid="rke2-directory-config-k8sDistro-data-dir"]');
+
+    systemAgentInput.setValue(inputPath);
+    provisioningInput.setValue(inputPath);
+    k8sDistroInput.setValue(inputPath);
+    await wrapper.vm.$nextTick();
+
+    expect(wrapper.vm.value.systemAgent).toStrictEqual(inputPath);
+    expect(wrapper.vm.value.provisioning).toStrictEqual(inputPath);
+    expect(wrapper.vm.value.k8sDistro).toStrictEqual(inputPath);
+  });
+
+  it('checkbox should be checked if all data dir values are the same (with all data dir values filled)', () => {
+    const newMountOptions = clone(mountOptions);
+    const inputPath = 'some-data-dir';
+
+    newMountOptions.propsData.value.systemAgent = inputPath;
+    newMountOptions.propsData.value.provisioning = inputPath;
+    newMountOptions.propsData.value.k8sDistro = inputPath;
+
+    wrapper = mount(
+      DirectoryConfig,
+      newMountOptions
+    );
+
+    const checkbox = wrapper.find('[data-testid="rke2-directory-config-individual-config-checkbox"]');
+
+    expect(checkbox.exists()).toBe(true);
+    expect(wrapper.vm.isSettingCommonConfig).toBe(true);
+
+    const commonInput = wrapper.find('[data-testid="rke2-directory-config-common-data-dir"]');
+    const systemAgentInput = wrapper.find('[data-testid="rke2-directory-config-systemAgent-data-dir"]');
+    const provisioningInput = wrapper.find('[data-testid="rke2-directory-config-provisioning-data-dir"]');
+    const k8sDistroInput = wrapper.find('[data-testid="rke2-directory-config-k8sDistro-data-dir"]');
+
+    expect(commonInput.exists()).toBe(true);
+    expect(systemAgentInput.exists()).toBe(false);
+    expect(provisioningInput.exists()).toBe(false);
+    expect(k8sDistroInput.exists()).toBe(false);
+
+    expect(wrapper.vm.value.systemAgent).toStrictEqual(inputPath);
+    expect(wrapper.vm.value.provisioning).toStrictEqual(inputPath);
+    expect(wrapper.vm.value.k8sDistro).toStrictEqual(inputPath);
+  });
+
+  it('checkbox should NOT be checked if some data dir values are the different (with all data dir values filled)', () => {
+    const newMountOptions = clone(mountOptions);
+    const inputPath1 = 'some-data-dir1';
+    const inputPath2 = 'some-data-dir2';
+    const inputPath3 = 'some-data-dir3';
+
+    newMountOptions.propsData.value.systemAgent = inputPath1;
+    newMountOptions.propsData.value.provisioning = inputPath2;
+    newMountOptions.propsData.value.k8sDistro = inputPath3;
+
+    wrapper = mount(
+      DirectoryConfig,
+      newMountOptions
+    );
+
+    expect(wrapper.vm.isSettingCommonConfig).toBe(false);
+
+    const commonInput = wrapper.find('[data-testid="rke2-directory-config-common-data-dir"]');
+    const systemAgentInput = wrapper.find('[data-testid="rke2-directory-config-systemAgent-data-dir"]');
+    const provisioningInput = wrapper.find('[data-testid="rke2-directory-config-provisioning-data-dir"]');
+    const k8sDistroInput = wrapper.find('[data-testid="rke2-directory-config-k8sDistro-data-dir"]');
+
+    expect(commonInput.exists()).toBe(false);
+    expect(systemAgentInput.exists()).toBe(true);
+    expect(provisioningInput.exists()).toBe(true);
+    expect(k8sDistroInput.exists()).toBe(true);
+
+    expect(wrapper.vm.value.systemAgent).toStrictEqual(inputPath1);
+    expect(wrapper.vm.value.provisioning).toStrictEqual(inputPath2);
+    expect(wrapper.vm.value.k8sDistro).toStrictEqual(inputPath3);
+  });
+
+  it('on a mode different than _CREATE all visible inputs should be disabled (with common config)', () => {
+    const newMountOptions = clone(mountOptions);
+    const inputPath = 'some-data-dir';
+
+    newMountOptions.propsData.value.systemAgent = inputPath;
+    newMountOptions.propsData.value.provisioning = inputPath;
+    newMountOptions.propsData.value.k8sDistro = inputPath;
+    newMountOptions.propsData.mode = _EDIT;
+
+    wrapper = mount(
+      DirectoryConfig,
+      newMountOptions
+    );
+
+    const checkbox = wrapper.find('[data-testid="rke2-directory-config-individual-config-checkbox"]');
+    const commonInput = wrapper.find('[data-testid="rke2-directory-config-common-data-dir"]');
+    const systemAgentInput = wrapper.find('[data-testid="rke2-directory-config-systemAgent-data-dir"]');
+    const provisioningInput = wrapper.find('[data-testid="rke2-directory-config-provisioning-data-dir"]');
+    const k8sDistroInput = wrapper.find('[data-testid="rke2-directory-config-k8sDistro-data-dir"]');
+
+    expect(checkbox.exists()).toBe(true);
+    expect(commonInput.exists()).toBe(true);
+    expect(systemAgentInput.exists()).toBe(false);
+    expect(provisioningInput.exists()).toBe(false);
+    expect(k8sDistroInput.exists()).toBe(false);
+
+    expect(checkbox.find('label').classes('disabled')).toBe(true);
+    expect(commonInput.attributes('disabled')).toBe('disabled');
+  });
+
+  it('on a mode different than _CREATE all visible inputs should be disabled (with different values)', () => {
+    const newMountOptions = clone(mountOptions);
+    const inputPath1 = 'some-data-dir1';
+    const inputPath2 = 'some-data-dir2';
+    const inputPath3 = 'some-data-dir3';
+
+    newMountOptions.propsData.value.systemAgent = inputPath1;
+    newMountOptions.propsData.value.provisioning = inputPath2;
+    newMountOptions.propsData.value.k8sDistro = inputPath3;
+    newMountOptions.propsData.mode = _EDIT;
+
+    wrapper = mount(
+      DirectoryConfig,
+      newMountOptions
+    );
+
+    const checkbox = wrapper.find('[data-testid="rke2-directory-config-individual-config-checkbox"]');
+    const commonInput = wrapper.find('[data-testid="rke2-directory-config-common-data-dir"]');
+    const systemAgentInput = wrapper.find('[data-testid="rke2-directory-config-systemAgent-data-dir"]');
+    const provisioningInput = wrapper.find('[data-testid="rke2-directory-config-provisioning-data-dir"]');
+    const k8sDistroInput = wrapper.find('[data-testid="rke2-directory-config-k8sDistro-data-dir"]');
+
+    expect(checkbox.exists()).toBe(true);
+    expect(commonInput.exists()).toBe(false);
+    expect(systemAgentInput.exists()).toBe(true);
+    expect(provisioningInput.exists()).toBe(true);
+    expect(k8sDistroInput.exists()).toBe(true);
+
+    expect(checkbox.find('label').classes('disabled')).toBe(true);
+    expect(systemAgentInput.attributes('disabled')).toBe('disabled');
+    expect(provisioningInput.attributes('disabled')).toBe('disabled');
+    expect(k8sDistroInput.attributes('disabled')).toBe('disabled');
+  });
+});

package/edit/provisioning.cattle.io.cluster/__tests__/DrainOptions.test.ts

@@ -1,5 +1,5 @@
 import { mount } from '@vue/test-utils';
-import DrainOptions from '@shell/edit/provisioning.cattle.io.cluster/DrainOptions.vue';
+import DrainOptions from '@shell/edit/provisioning.cattle.io.cluster/tabs/upgrade/DrainOptions';
 import { DefaultProps } from 'vue/types/options';
 import { ExtendedVue, Vue } from 'vue/types/vue';
 

package/edit/provisioning.cattle.io.cluster/__tests__/index.test.ts

@@ -0,0 +1,73 @@
+import { shallowMount, createLocalVue } from '@vue/test-utils';
+import Vuex from 'vuex';
+import ClusterCreate from '@shell/edit/provisioning.cattle.io.cluster/index.vue';
+
+describe('component: Cluster: Create', () => {
+  it('should hide RKE1 and RKE2 toggle button if RKE1 ui feature flag is NOT set', () => {
+    const localVue = createLocalVue();
+
+    localVue.use(Vuex);
+
+    const store = new Vuex.Store({
+      modules: {
+        i18n: {
+          namespaced: true,
+          getters:    {
+            t:            jest.fn(),
+            withFallback: () => jest.fn(),
+          }
+        },
+        features: {
+          namespaced: true,
+          getters:    {
+            get:          () => jest.fn(),
+            withFallback: jest.fn(),
+          }
+        },
+        prefs: {
+          namespaced: true,
+          getters:    { get: jest.fn() }
+        },
+        mapFeature: {
+          namespaced: true,
+          getters:    { get: jest.fn() }
+        },
+        'type-map': {
+          namespaced: true,
+          getters:    { activeProducts: () => [] }
+        },
+        catalog: {
+          namespaced: true,
+          getters:    { charts: () => [] }
+        }
+      },
+      getters: {
+        defaultClusterId: jest.fn(),
+        clusterId:        jest.fn()
+      }
+    });
+
+    const wrapper = shallowMount(ClusterCreate, {
+      computed:  { rke1UiEnabled: () => false },
+      propsData: {
+        value:           { metadata: {}, spec: { template: {} } },
+        realMode:        '',
+        mode:            'edit',
+        componentTestid: 'cluster-manager-create',
+      },
+      mixins: [],
+      store,
+      localVue,
+      mocks:  {
+        $route:      { params: {}, query: {} },
+        $router:     { applyQuery: jest.fn() },
+        $fetchState: { pending: false },
+      },
+      stubs: { CruResource: { template: '<div><slot name="subtypes"></slot></div>' } }
+    });
+
+    const element = wrapper.find('[data-testid="cluster-manager-create-rke-switch"]').element;
+
+    expect(element).not.toBeDefined();
+  });
+});

package/edit/provisioning.cattle.io.cluster/__tests__/rke2.test.ts

@@ -1,4 +1,6 @@
 import { mount, shallowMount } from '@vue/test-utils';
+import { SECRET } from '@shell/config/types';
+import { _CREATE } from '@shell/config/query-params';
 import rke2 from '@shell/edit/provisioning.cattle.io.cluster/rke2.vue';
 
 /**
@@ -39,7 +41,13 @@ const defaultStubs = {
   UnitInput:                true,
   YamlEditor:               true,
   MemberRoles:              true,
-  Basics:                   true
+  Basics:                   true,
+  Etcd:                     true,
+  Networking:               true,
+  Upgrade:                  true,
+  Registries:               true,
+  AddOnConfig:              true,
+  Advanced:                 true
 };
 
 const mockAgentArgs = { 'cloud-provider-name': { options: [], profile: { options: [{ anything: 'yes' }] } } };
@@ -70,12 +78,13 @@ const defaultComputed = {
 };
 
 const defaultGetters = {
-  currentStore:                     () => 'current_store',
-  'management/schemaFor':           jest.fn(),
-  'current_store/all':              jest.fn(),
-  'i18n/t':                         jest.fn(),
-  'i18n/withFallback':              jest.fn(),
-  'plugins/cloudProviderForDriver': jest.fn()
+  currentStore:                      () => 'current_store',
+  'management/schemaFor':            jest.fn(),
+  'current_store/all':               jest.fn(),
+  'i18n/t':                          jest.fn(),
+  'i18n/withFallback':               jest.fn(),
+  'plugins/cloudProviderForDriver':  jest.fn(),
+  'customization/getPreviewCluster': jest.fn(),
 };
 
 const defaultMocks = {
@@ -109,7 +118,7 @@ describe('component: rke2', () => {
     const k8s = 'v1.25.0+rke2r1';
     const wrapper = mount(rke2, {
       propsData: {
-        mode:  'create',
+        mode:  _CREATE,
         value: {
           spec: {
             ...defaultSpec,
@@ -124,7 +133,7 @@ describe('component: rke2', () => {
       computed: defaultComputed,
       mocks:    {
         ...defaultMocks,
-        $store: { getters: defaultGetters },
+        $store: { dispatch: () => jest.fn(), getters: defaultGetters },
       },
       stubs: defaultStubs
     });
@@ -136,7 +145,7 @@ describe('component: rke2', () => {
     const k8s = 'v1.25.0+k3s1';
     const wrapper = mount(rke2, {
       propsData: {
-        mode:  'create',
+        mode:  _CREATE,
         value: {
           spec: {
             ...defaultSpec,
@@ -150,7 +159,7 @@ describe('component: rke2', () => {
       computed: defaultComputed,
       mocks:    {
         ...defaultMocks,
-        $store: { getters: defaultGetters },
+        $store: { dispatch: () => jest.fn(), getters: defaultGetters },
       },
       stubs: defaultStubs
     });
@@ -162,7 +171,7 @@ describe('component: rke2', () => {
     const k8s = 'v1.25.0+k3s1';
     const wrapper = mount(rke2, {
       propsData: {
-        mode:  'create',
+        mode:  _CREATE,
         value: {
           spec: {
             ...defaultSpec,
@@ -176,7 +185,7 @@ describe('component: rke2', () => {
       computed: defaultComputed,
       mocks:    {
         ...defaultMocks,
-        $store: { getters: defaultGetters },
+        $store: { dispatch: () => jest.fn(), getters: defaultGetters },
       },
       stubs: defaultStubs
     });
@@ -186,6 +195,61 @@ describe('component: rke2', () => {
     wrapper.vm.machinePools.forEach((p: any) => expect(p.drainBeforeDelete).toBe(true));
   });
 
+  it('should set distro root directory from k8sDistro on a Harvester cluster creation on save override (_doSaveOverride)', async() => {
+    const k8s = 'v1.25.0+k3s1';
+
+    const HARVESTER = 'harvester';
+    const HARVESTER_CLOUD_PROVIDER = 'harvester-cloud-provider';
+
+    const newSpec = Object.assign({}, defaultSpec);
+
+    newSpec.rkeConfig.dataDirectories = { k8sDistro: 'my-k8s-distro-path' };
+
+    const wrapper = mount(rke2, {
+      propsData: {
+        mode:  _CREATE,
+        value: {
+          spec: {
+            ...newSpec,
+            kubernetesVersion: k8s
+          },
+          metadata:    { name: 'cluster-name' },
+          agentConfig: { 'cloud-provider-name': HARVESTER }
+        },
+        provider: 'custom'
+      },
+      data: () => ({
+        credentialId: 'I am authenticated',
+        credential:   { decodedData: { clusterId: 'some-cluster-id' } },
+        machinePools: [],
+      }),
+      computed: defaultComputed,
+      mocks:    {
+        ...defaultMocks,
+        $store: {
+          // mock secret creation on "createKubeconfigSecret"
+          dispatch: (action: any, opts: any) => {
+            if (action === 'management/create' && opts.type === SECRET) {
+              return { save: () => jest.fn };
+            } else {
+              return jest.fn();
+            }
+          },
+          getters: defaultGetters
+        },
+      },
+      stubs: defaultStubs
+    });
+
+    // we need to mock the "save" method from the create-edit-view-mixin
+    // otherwise we get console errors
+    jest.spyOn(wrapper.vm, 'save').mockImplementation();
+
+    await wrapper.vm._doSaveOverride(jest.fn());
+
+    expect(wrapper.vm.chartValues[HARVESTER_CLOUD_PROVIDER].cloudConfigPath).toStrictEqual('my-k8s-distro-path/etc/config-files/cloud-provider-config');
+  });
+
   // TODO: Complete test after implementing fetch https://github.com/rancher/dashboard/issues/9322
   // eslint-disable-next-line jest/no-disabled-tests
   describe.skip('should initialize agent configuration values', () => {