make test a bit more specific

uc-cdis · Oct 13, 2022 · 14e88b6 · 14e88b6
1 parent bee095b
commit 14e88b6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/UserProfile/UserProfile.test.jsx b/src/UserProfile/UserProfile.test.jsx
@@ -42,7 +42,8 @@ describe('the UserProfile component', () => {
         />
       </StaticRouter>,
     );
-    expect($vdom.find('tbody tr')).toHaveLength(testProps.userProfile.jtis.length);
+    // we expect one delete button per jti:
+    expect($vdom.find(`button[name="${DELETE_BTN}"]`)).toHaveLength(testProps.userProfile.jtis.length);
   });
 
   it('triggers create-key events', (done) => {
@@ -79,7 +80,7 @@ describe('the UserProfile component', () => {
       </StaticRouter>,
     );
     const $deleteBtn = $vdom.find(`button[name="${DELETE_BTN}"]`);
-    expect($deleteBtn).toHaveLength($vdom.find('tbody tr').length);
+    expect($deleteBtn).toHaveLength(testProps.userProfile.jtis.length);
     $deleteBtn.at(0).simulate('click');
     // should invoke onRequestDeleteKey callback  (above - calls done()) ...
   });