X-Git-Url: https://adrianiainlam.tk/git/?a=blobdiff_plain;f=auto_tests%2Ftests%2Fstacked.js;h=706f56fd2da6a51517a41bc5a18333b8d7336103;hb=abc8c570a4aac8be7ff4fb266e6e410b768277be;hp=f7ca4428128550066b99e10e9e2c37c2255d1ed1;hpb=9b3d945989b5cfc8b4c498f7c001f93b22ad73ac;p=dygraphs.git

diff --git a/auto_tests/tests/stacked.js b/auto_tests/tests/stacked.js
index f7ca442..706f56f 100644
--- a/auto_tests/tests/stacked.js
+++ b/auto_tests/tests/stacked.js
@@ -78,11 +78,7 @@ stackedTestCase.prototype.testSelectionValues = function() {
 
   g.setSelection(0);
 
-  var legend = document.getElementsByClassName("dygraph-legend");
-  assertEquals(1, legend.length);
-  legend = legend[0];
-
-  assertEquals("0: Y1:1 Y2:1", legend.textContent);
+  assertEquals("0: Y1: 1 Y2: 1", Util.getLegend());
 
   // Verify that the behavior is correct with highlightSeriesOpts as well.
   g.updateOptions({
@@ -93,8 +89,8 @@ stackedTestCase.prototype.testSelectionValues = function() {
   // NOTE: calling g.setSelection(0) here makes the test fail, due to an
   // unrelated bug.
   g.setSelection(1);
-  assertEquals("1: Y1:1 Y2:1", legend.textContent);
+  assertEquals("1: Y1: 1 Y2: 1", Util.getLegend());
 
   g.setSelection(0, 'Y2');
-  assertEquals("0: Y1:1 Y2:1", legend.textContent);
+  assertEquals("0: Y1: 1 Y2: 1", Util.getLegend());
 };