From 5b255031ebd142a1ef0ed5af8fd7f2db2de8c9f9 Mon Sep 17 00:00:00 2001
From: Sven <38101365+svenvoskamp@users.noreply.github.com>
Date: Sun, 16 Feb 2025 18:25:48 +0100
Subject: [PATCH] test: add test for `w3m-connect-injected-widget` (#3651)

Co-authored-by: Enes <enesozturk.d@gmail.com>
---
 .../w3m-connect-injected-widget.test.ts       | 221 ++++++++++++++++++
 1 file changed, 221 insertions(+)
 create mode 100644 packages/scaffold-ui/test/partials/w3m-connect-injected-widget.test.ts

diff --git a/packages/scaffold-ui/test/partials/w3m-connect-injected-widget.test.ts b/packages/scaffold-ui/test/partials/w3m-connect-injected-widget.test.ts
new file mode 100644
index 0000000000..99bbebef7e
--- /dev/null
+++ b/packages/scaffold-ui/test/partials/w3m-connect-injected-widget.test.ts
@@ -0,0 +1,221 @@
+import { elementUpdated, fixture } from '@open-wc/testing'
+import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest'
+
+import { html } from 'lit'
+
+import { ConstantsUtil } from '@reown/appkit-common'
+import type { ConnectorType, ConnectorWithProviders } from '@reown/appkit-core'
+import {
+  ApiController,
+  ConnectionController,
+  ConnectorController,
+  CoreHelperUtil,
+  RouterController
+} from '@reown/appkit-core'
+
+import { W3mConnectInjectedWidget } from '../../src/partials/w3m-connect-injected-widget'
+import { HelpersUtil } from '../utils/HelpersUtil'
+
+// --- Constants ---------------------------------------------------- //
+const MOCK_INJECTED_CONNECTOR: ConnectorWithProviders = {
+  id: 'mockInjected',
+  name: 'Mock Injected Wallet',
+  type: 'INJECTED' as ConnectorType,
+  info: {
+    rdns: 'mock.injected.wallet'
+  },
+  provider: undefined,
+  chain: ConstantsUtil.CHAIN.EVM
+}
+
+const BROWSER_WALLET_CONNECTOR: ConnectorWithProviders = {
+  id: 'browserWallet',
+  name: 'Browser Wallet',
+  type: 'INJECTED' as ConnectorType,
+  provider: undefined,
+  chain: ConstantsUtil.CHAIN.EVM
+}
+
+describe('W3mConnectInjectedWidget', () => {
+  beforeAll(() => {
+    vi.spyOn(CoreHelperUtil, 'isMobile').mockReturnValue(false)
+  })
+
+  beforeEach(() => {
+    vi.spyOn(ApiController, 'state', 'get').mockReturnValue({
+      ...ApiController.state,
+      excludedRDNS: []
+    })
+    vi.spyOn(ConnectionController, 'checkInstalled').mockReturnValue(true)
+  })
+
+  afterEach(() => {
+    vi.resetAllMocks()
+  })
+
+  it('should not render anything if there are no injected connectors', async () => {
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: []
+    })
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    expect(element.style.display).toBe('none')
+  })
+
+  it('should render injected connectors', async () => {
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [MOCK_INJECTED_CONNECTOR]
+    })
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    element.requestUpdate()
+    await elementUpdated(element)
+
+    const walletSelector = HelpersUtil.getByTestId(
+      element,
+      `wallet-selector-${MOCK_INJECTED_CONNECTOR.id}`
+    )
+    expect(walletSelector).not.toBeNull()
+    expect(walletSelector.getAttribute('name')).toBe(MOCK_INJECTED_CONNECTOR.name)
+    expect(walletSelector.getAttribute('tagLabel')).toBe('installed')
+    expect(walletSelector.getAttribute('tagVariant')).toBe('success')
+  })
+
+  it('should not render excluded RDNS wallets', async () => {
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [MOCK_INJECTED_CONNECTOR]
+    })
+
+    vi.spyOn(ApiController, 'state', 'get').mockReturnValue({
+      ...ApiController.state,
+      excludedRDNS: ['mock.injected.wallet']
+    })
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    element.requestUpdate()
+    await elementUpdated(element)
+
+    const walletSelector = HelpersUtil.getByTestId(
+      element,
+      `wallet-selector-${MOCK_INJECTED_CONNECTOR.id}`
+    )
+    expect(walletSelector).toBeNull()
+  })
+
+  it('should not render Browser Wallet on desktop', async () => {
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [BROWSER_WALLET_CONNECTOR]
+    })
+    vi.spyOn(CoreHelperUtil, 'isMobile').mockReturnValue(false)
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    element.requestUpdate()
+    await elementUpdated(element)
+
+    const walletSelector = HelpersUtil.getByTestId(
+      element,
+      `wallet-selector-${BROWSER_WALLET_CONNECTOR.id}`
+    )
+    expect(walletSelector).toBeNull()
+  })
+
+  it('should render Browser Wallet on mobile', async () => {
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [BROWSER_WALLET_CONNECTOR]
+    })
+    vi.spyOn(CoreHelperUtil, 'isMobile').mockReturnValue(true)
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    element.requestUpdate()
+    await elementUpdated(element)
+
+    const walletSelector = HelpersUtil.getByTestId(
+      element,
+      `wallet-selector-${BROWSER_WALLET_CONNECTOR.id}`
+    )
+    expect(walletSelector).not.toBeNull()
+  })
+
+  it('should route to ConnectingExternal on connector click', async () => {
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [MOCK_INJECTED_CONNECTOR]
+    })
+    const setActiveConnectorSpy = vi.spyOn(ConnectorController, 'setActiveConnector')
+    const pushSpy = vi.spyOn(RouterController, 'push')
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    const walletSelector = HelpersUtil.getByTestId(
+      element,
+      `wallet-selector-${MOCK_INJECTED_CONNECTOR.id}`
+    )
+    walletSelector.click()
+
+    expect(setActiveConnectorSpy).toHaveBeenCalledWith(MOCK_INJECTED_CONNECTOR)
+    expect(pushSpy).toHaveBeenCalledWith('ConnectingExternal', {
+      connector: MOCK_INJECTED_CONNECTOR
+    })
+  })
+
+  it('should handle unknown wallet names', async () => {
+    const unknownConnector: ConnectorWithProviders = {
+      ...MOCK_INJECTED_CONNECTOR,
+      name: undefined
+    } as unknown as ConnectorWithProviders
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [unknownConnector]
+    })
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    const walletSelector = HelpersUtil.getByTestId(
+      element,
+      `wallet-selector-${unknownConnector.id}`
+    )
+    expect(walletSelector.getAttribute('name')).toBe('Unknown')
+  })
+
+  it('should not render when no RDNS and not installed', async () => {
+    const noRdnsConnector: ConnectorWithProviders = {
+      ...MOCK_INJECTED_CONNECTOR,
+      info: undefined
+    }
+    vi.spyOn(ConnectorController, 'state', 'get').mockReturnValue({
+      ...ConnectorController.state,
+      connectors: [noRdnsConnector]
+    })
+    vi.spyOn(ConnectionController, 'checkInstalled').mockReturnValue(false)
+
+    const element: W3mConnectInjectedWidget = await fixture(
+      html`<w3m-connect-injected-widget></w3m-connect-injected-widget>`
+    )
+
+    expect(element.style.display).toBe('none')
+  })
+})