chore: using viem address comparison

Author: hanzel98

packages/delegation-toolkit/src/actions/isValid7702Implementation.ts

@@ -1,4 +1,5 @@
 import type { Client, Address, Hex } from 'viem';
+import { isAddressEqual } from 'viem';
 import { getCode } from 'viem/actions';
 
 import type { DeleGatorEnvironment } from '../types';
@@ -92,9 +93,7 @@ export async function isValid7702Implementation({
       return false;
     }
 
-    return (
-      delegatedAddress.toLowerCase() === expectedImplementation.toLowerCase()
-    );
+    return isAddressEqual(delegatedAddress, expectedImplementation);
   } catch (error) {
     // If the call fails (e.g., no code at address, network error),
     // then it's not properly delegated to our implementation

packages/delegation-toolkit/test/actions/isValid7702Implementation.test.ts

@@ -60,12 +60,12 @@ describe('isValid7702Implementation', () => {
 
     it('should return true when delegation code uses different case but matches implementation', async () => {
       const testAddress = '0xabcdefabcdefabcdefabcdefabcdefabcdefabcd';
-      const delegationCode = '0xef01004000000000000000000000000000000000000000';
+      const delegationCode = '0xef0100abcdef1234567890abcdef1234567890abcdef00';
       mockGetCode.mockResolvedValue(delegationCode);
 
-      // Update environment with uppercase address
+      // Update environment with uppercase address to test case insensitivity
       mockEnvironment.implementations.EIP7702StatelessDeleGatorImpl =
-        '0x4000000000000000000000000000000000000000';
+        '0xABCDEF1234567890ABCDEF1234567890ABCDEF00';
 
       const result = await isValid7702Implementation({
         client: publicClient,
@@ -75,6 +75,21 @@ describe('isValid7702Implementation', () => {
 
       expect(result).toBe(true);
     });
+
+    it('should return false when delegation prefix uses different case (case-sensitive prefix)', async () => {
+      const testAddress = '0xdddddddddddddddddddddddddddddddddddddddd';
+      // Use uppercase delegation prefix - this should fail because startsWith is case-sensitive
+      const delegationCode = '0xEF01004000000000000000000000000000000000000000';
+      mockGetCode.mockResolvedValue(delegationCode);
+
+      const result = await isValid7702Implementation({
+        client: publicClient,
+        accountAddress: testAddress,
+        environment: mockEnvironment,
+      });
+
+      expect(result).toBe(false);
+    });
   });
 
   describe('Failure Cases', () => {