Passing unit tests: VPCSubnetsTable, SubnetLinodeRow, and LinodeEntityDetailBody

Author: DevDW

packages/manager/src/features/Linodes/LinodeEntityDetailBody.test.tsx

@@ -6,12 +6,26 @@ import {
   subnetFactory,
   vpcFactory,
 } from 'src/factories';
-import { renderWithTheme } from 'src/utilities/testHelpers';
+import { mockMatchMedia, renderWithTheme } from 'src/utilities/testHelpers';
 
 import { LinodeEntityDetailBody } from './LinodeEntityDetailBody';
 
 import type { BodyProps } from './LinodeEntityDetailBody';
 
+beforeAll(() => mockMatchMedia());
+
+const queryMocks = vi.hoisted(() => ({
+  useAccount: vi.fn().mockReturnValue({}),
+}));
+
+vi.mock('@linode/queries', async () => {
+  const actual = await vi.importActual('@linode/queries');
+  return {
+    ...actual,
+    useAccount: queryMocks.useAccount,
+  };
+});
+
 describe('LinodeEntityDetailBody', () => {
   const baseProps: BodyProps = {
     encryptionStatus: 'enabled',
@@ -82,6 +96,12 @@ describe('LinodeEntityDetailBody', () => {
   });
 
   it('renders VPC IPv4 and VPC IPv6 for a linode assigned to a Dual Stack VPC', () => {
+    queryMocks.useAccount.mockReturnValue({
+      data: {
+        capabilities: ['VPC Dual Stack'],
+      },
+    });
+
     const subnet = subnetFactory.build({
       id: 1,
       linodes: [subnetAssignedLinodeDataFactory.build({ id: 1 })],

packages/manager/src/features/VPCs/VPCDetail/SubnetLinodeRow.test.tsx

@@ -32,6 +32,7 @@ const queryMocks = vi.hoisted(() => ({
   useLinodeConfigQuery: vi.fn().mockReturnValue({}),
   useLinodeInterfaceQuery: vi.fn().mockReturnValue({}),
   useLinodeInterfaceFirewallsQuery: vi.fn().mockReturnValue({}),
+  useAccount: vi.fn().mockReturnValue({}),
 }));
 
 vi.mock('@linode/queries', async () => {
@@ -44,6 +45,7 @@ vi.mock('@linode/queries', async () => {
     useLinodeInterfaceQuery: queryMocks.useLinodeInterfaceQuery,
     useLinodeInterfaceFirewallsQuery:
       queryMocks.useLinodeInterfaceFirewallsQuery,
+    useAccount: queryMocks.useAccount,
   };
 });
 
@@ -206,7 +208,13 @@ describe('SubnetLinodeRow', () => {
     expect(firewall).toBeVisible();
   });
 
-  it('should display the VPC IPv6 and VPC IPv6 Ranges when vpcIpv6 feature flag is enabled (config/legacy interface)', async () => {
+  it('should display the VPC IPv6 and VPC IPv6 Ranges when vpcIpv6 feature flag is enabled and user has VPC Dual Stack account capability (config/legacy interface)', async () => {
+    queryMocks.useAccount.mockReturnValue({
+      data: {
+        capabilities: ['VPC Dual Stack'],
+      },
+    });
+
     const linodeFactory1 = linodeFactory.build({ id: 1, label: 'linode-1' });
     const subnetFactory1 = subnetFactory.build({ id: 1, label: 'subnet-1' });
     const config = linodeConfigFactory.build({
@@ -242,7 +250,13 @@ describe('SubnetLinodeRow', () => {
     await findByText('2001:db8::/64');
   });
 
-  it('should display the VPC IPv6 and VPC IPv6 Ranges when vpcIpv6 feature flag is enabled (Linode Interface)', async () => {
+  it('should display the VPC IPv6 and VPC IPv6 Ranges when vpcIpv6 feature flag is enabled and user has VPC Dual Stack account capability (Linode Interface)', async () => {
+    queryMocks.useAccount.mockReturnValue({
+      data: {
+        capabilities: ['VPC Dual Stack'],
+      },
+    });
+
     const linodeFactory1 = linodeFactory.build({ id: 1, label: 'linode-1' });
     const vpcLinodeInterface = linodeInterfaceFactoryDualStackVPC.build({
       id: 1,

packages/manager/src/features/VPCs/VPCDetail/VPCSubnetsTable.test.tsx

@@ -2,11 +2,12 @@ import { screen, waitForElementToBeRemoved } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
 import * as React from 'react';
 
-import { firewallSettingsFactory } from 'src/factories';
+import { accountFactory, firewallSettingsFactory } from 'src/factories';
 import {
   subnetAssignedLinodeDataFactory,
   subnetFactory,
 } from 'src/factories/subnets';
+import { http, HttpResponse, server } from 'src/mocks/testServer';
 import { mockMatchMedia, renderWithTheme } from 'src/utilities/testHelpers';
 
 import { VPCSubnetsTable } from './VPCSubnetsTable';
@@ -199,6 +200,10 @@ describe('VPC Subnets table', () => {
 
   // @TODO VPC IPv6: Remove this assertion once VPC IPv6 is fully rolled out
   it('renders VPC IPv6 and VPC IPv6 Ranges columns in Linode table when vpcIpv6 feature flag is enabled', async () => {
+    const account = accountFactory.build({
+      capabilities: ['VPC Dual Stack'],
+    });
+
     const subnet = subnetFactory.build({
       linodes: [subnetAssignedLinodeDataFactory.build({ id: 1 })],
     });
@@ -209,6 +214,12 @@ describe('VPC Subnets table', () => {
       },
     });
 
+    server.use(
+      http.get('*/account', () => {
+        return HttpResponse.json(account);
+      })
+    );
+
     renderWithTheme(
       <VPCSubnetsTable
         isVPCLKEEnterpriseCluster={false}