@@ -98,8 +98,7 @@ describe("Tree", () => {
9898 className : "testClassName"
9999 } ) ;
100100
101- expect ( wrapper . hasClass ( "testClassName" ) ) . toBe ( true ) ;
102- expect ( wrapper . hasClass ( "tree" ) ) . toBe ( true ) ;
101+ expect ( wrapper . find ( ".tree" ) . hasClass ( "testClassName" ) ) . toBe ( true ) ;
103102 } ) ;
104103
105104 it ( "renders as expected when passed a style" , ( ) => {
@@ -369,9 +368,9 @@ describe("Tree", () => {
369368
370369 getTreeNodes ( wrapper ) . forEach ( n => {
371370 if ( "ABECDMN" . split ( "" ) . includes ( getSanitizedNodeText ( n ) ) ) {
372- expect ( n . find ( ".arrow.expanded" ) . exists ( ) ) . toBe ( true ) ;
371+ expect ( n . find ( "i .arrow.expanded" ) . exists ( ) ) . toBe ( true ) ;
373372 } else {
374- expect ( n . find ( ".arrow" ) . exists ( ) ) . toBe ( false ) ;
373+ expect ( n . find ( "i .arrow" ) . exists ( ) ) . toBe ( false ) ;
375374 }
376375 } ) ;
377376 } ) ;
@@ -381,7 +380,7 @@ describe("Tree", () => {
381380 expect ( formatTree ( wrapper ) ) . toMatchSnapshot ( ) ;
382381
383382 getTreeNodes ( wrapper ) . forEach ( n => {
384- const arrow = n . find ( ".arrow" ) ;
383+ const arrow = n . find ( "i .arrow" ) ;
385384 expect ( arrow . exists ( ) ) . toBe ( true ) ;
386385 expect ( arrow . hasClass ( "expanded" ) ) . toBe ( false ) ;
387386 } ) ;
@@ -442,7 +441,7 @@ function formatTree(wrapper) {
442441 . map ( ( node ) => {
443442 const level = node . prop ( "aria-level" ) ;
444443 const indentStr = "| " . repeat ( level || 0 ) ;
445- const arrow = node . find ( ".arrow" ) ;
444+ const arrow = node . find ( "i .arrow" ) ;
446445 let arrowStr = " " ;
447446 if ( arrow . exists ( ) ) {
448447 arrowStr = arrow . hasClass ( "expanded" ) ? "▼ " : "▶︎ " ;
0 commit comments