add test coverage

src/identity/hooks/useAvatar.test.tsx

@@ -1,5 +1,5 @@
 import { renderHook, waitFor } from '@testing-library/react';
-import { base, baseSepolia, mainnet } from 'viem/chains';
+import { base, baseSepolia, mainnet, optimism } from 'viem/chains';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 import { publicClient } from '../../network/client';
 import { getChainPublicClient } from '../../network/getChainPublicClient';
@@ -107,4 +107,31 @@ describe('useAvatar', () => {
 
     expect(getChainPublicClient).toHaveBeenCalledWith(baseSepolia);
   });
+
+  it('returns error for unsupported chain ', async () => {
+    const testEnsName = 'shrek.basetest.eth';
+    const testEnsAvatar = 'shrektestface';
+
+    // Mock the getEnsAvatar method of the publicClient
+    mockGetEnsAvatar.mockResolvedValue(testEnsAvatar);
+
+    // Use the renderHook function to create a test harness for the useAvatar hook
+    const { result } = renderHook(
+      () => useAvatar({ ensName: testEnsName, chain: optimism }),
+      {
+        wrapper: getNewReactQueryTestProvider(),
+      },
+    );
+
+    // Wait for the hook to finish fetching the ENS name
+    await waitFor(() => {
+      // Check that the ENS name and loading state are correct
+      expect(result.current.data).toBe(undefined);
+      expect(result.current.isLoading).toBe(false);
+      expect(result.current.isError).toBe(true);
+      expect(result.current.error).toBe(
+        'ChainId not supported, avatar resolution is only supported on Ethereum and Base.',
+      );
+    });
+  });
 });

src/identity/utils/getAvatar.test.tsx

@@ -1,4 +1,4 @@
-import { base, baseSepolia, mainnet } from 'viem/chains';
+import { base, baseSepolia, mainnet, optimism } from 'viem/chains';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 import { publicClient } from '../../network/client';
 import { getChainPublicClient } from '../../network/getChainPublicClient';
@@ -108,4 +108,11 @@ describe('getAvatar', () => {
     // getAvatar defaulted to mainnet
     expect(getChainPublicClient).toHaveBeenCalledWith(mainnet);
   });
+
+  it('should throw an error on unsupported chain', async () => {
+    const ensName = 'shrek.basetest.eth';
+    await expect(getAvatar({ ensName, chain: optimism })).rejects.toBe(
+      'ChainId not supported, avatar resolution is only supported on Ethereum and Base.',
+    );
+  });
 });

src/identity/utils/getAvatar.ts

@@ -2,6 +2,7 @@ import { mainnet } from 'viem/chains';
 import { normalize } from 'viem/ens';
 import type { GetEnsAvatarReturnType } from 'wagmi/actions';
 import { isBase } from '../../isBase';
+import { isEthereum } from '../../isEthereum';
 import { getChainPublicClient } from '../../network/getChainPublicClient';
 import { RESOLVER_ADDRESSES_BY_CHAIN_ID } from '../constants';
 import type { GetAvatar, GetAvatarReturnType } from '../types';
@@ -10,10 +11,19 @@ export const getAvatar = async ({
   ensName,
   chain = mainnet,
 }: GetAvatar): Promise<GetAvatarReturnType> => {
+  const chainIsBase = isBase({ chainId: chain.id });
+  const chainIsEthereum = isEthereum({ chainId: chain.id });
+  const chainSupportsUniversalResolver = chainIsEthereum || chainIsBase;
+
+  if (!chainSupportsUniversalResolver) {
+    return Promise.reject(
+      'ChainId not supported, avatar resolution is only supported on Ethereum and Base.',
+    );
+  }
+
   let client = getChainPublicClient(chain);
 
-  if (isBase({ chainId: chain.id })) {
-    client = getChainPublicClient(chain);
+  if (chainIsBase) {
     try {
       const baseEnsAvatar = await client.getEnsAvatar({
         name: normalize(ensName),