From 50dbc67af57159c01b37dad084b2d387e3ec5e1a Mon Sep 17 00:00:00 2001
From: Mononaut <github@monospace.live>
Date: Thu, 26 Mar 2026 06:56:59 +0000
Subject: [PATCH] fix cypress test for address poisoning highlighting

---
 frontend/cypress/e2e/mainnet/mainnet.spec.ts | 46 ++++++++------------
 1 file changed, 17 insertions(+), 29 deletions(-)

diff --git a/frontend/cypress/e2e/mainnet/mainnet.spec.ts b/frontend/cypress/e2e/mainnet/mainnet.spec.ts
index a04803ca0..4c58e157d 100644
--- a/frontend/cypress/e2e/mainnet/mainnet.spec.ts
+++ b/frontend/cypress/e2e/mainnet/mainnet.spec.ts
@@ -229,18 +229,15 @@ describe('Mainnet', () => {
           cy.get('.alert-mempool').should('exist');
           cy.get('.poison-alert').its('length').should('equal', 2);
 
-          cy.get('.prefix')
-            .should('have.length', 2)
-            .each(($el) => {
-              cy.wrap($el).should('have.text', prefix);
-            });
+          cy.get('.poison-alert').parent().find('.address-content').should('have.length', 2).then(($containers) => {
+            const prefixes = $containers.map((i, el) => Cypress.$(el).find('.prefix').text()).get();
+            const infixes = $containers.map((i, el) => Cypress.$(el).find('.infix').text()).get().sort();
 
-          cy.get('.infix')
-            .should('have.length', 2)
-            .then(($infixes) => {
-              cy.wrap($infixes[0]).should('have.text', infix1);
-              cy.wrap($infixes[1]).should('have.text', infix2);
-            });
+            expect(prefixes).to.have.length(2);
+            prefixes.forEach((p) => expect(p).to.equal(prefix));
+
+            expect(infixes).to.deep.equal([infix1, infix2].sort());
+          });
         });
 
         it('highlights potential address poisoning attacks on inputs and outputs, prefix, infix and postfix', () => {
@@ -256,26 +253,17 @@ describe('Mainnet', () => {
           cy.get('.alert-mempool').should('exist');
           cy.get('.poison-alert').its('length').should('equal', 2);
 
-          cy.get('.prefix')
-            .should('have.length', 2)
-            .each(($el) => {
-              cy.wrap($el).should('have.text', prefix);
-            });
+          cy.get('.poison-alert').parent().find('.address-content').should('have.length', 2).then(($containers) => {
+            const prefixes = $containers.map((i, el) => Cypress.$(el).find('.prefix').text()).get();
+            const infixes = $containers.map((i, el) => Cypress.$(el).find('.infix').text()).get().sort();
+            const postfixes = $containers.map((i, el) => Cypress.$(el).find('.postfix').text()).get().sort();
 
-          cy.get('.infix')
-            .should('have.length', 2)
-            .then(($infixes) => {
-              cy.wrap($infixes[0]).should('have.text', infix1);
-              cy.wrap($infixes[1]).should('have.text', infix2);
-            });
-
-            cy.get('.postfix')
-            .should('have.length', 2)
-            .then(($postfixes) => {
-              cy.wrap($postfixes[0]).should('include.text', postfix1);
-              cy.wrap($postfixes[1]).should('include.text', postfix2);
-            });
+            expect(prefixes).to.have.length(2);
+            prefixes.forEach((p) => expect(p).to.equal(prefix));
 
+            expect(infixes).to.deep.equal([infix1, infix2].sort());
+            expect(postfixes).to.deep.equal([postfix1, postfix2].sort());
+          });
         });
       });