Use getAnnotationElements utility where not used in the test cases (#757

)
chartjs · Jun 8, 2022 · 8352c68 · 8352c68
1 parent 990e148
commit 8352c68
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/test/specs/annotation.spec.js b/test/specs/annotation.spec.js
@@ -161,8 +161,7 @@ describe('Annotation plugin', function() {
           }
         }
       });
-      const state = window['chartjs-plugin-annotation']._getState(chart);
-      const element = state.elements[0];
+      const element = window.getAnnotationElements(chart)[0];
       expect(element.options.drawTime).toBe('fallback');
     });
 
@@ -186,8 +185,7 @@ describe('Annotation plugin', function() {
           }
         }
       });
-      const state = window['chartjs-plugin-annotation']._getState(chart);
-      const element = state.elements[0];
+      const element = window.getAnnotationElements(chart)[0];
       expect(element.options.drawTime).toBe('afterDatasetsDraw');
     });
 
@@ -209,8 +207,7 @@ describe('Annotation plugin', function() {
           }
         }
       });
-      const state = window['chartjs-plugin-annotation']._getState(chart);
-      const element = state.elements[0];
+      const element = window.getAnnotationElements(chart)[0];
       expect(element.options.drawTime).toBe(chart.options.plugins.annotation.annotations.label.drawTime);
     });
   });