Fix intermittent tests by cleaning up after each

alphagov · Oct 22, 2024 · fe34f32 · fe34f32
1 parent 11ed1fd
commit fe34f32
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/spec/javascripts/table-of-contents-spec.js b/spec/javascripts/table-of-contents-spec.js
@@ -7,6 +7,7 @@ describe('Table of contents', function () {
   var $toc
   var $closeButton
   var $openButton
+  var $tocStickyHeader
   var module
 
   beforeAll(function () {
@@ -64,13 +65,18 @@ describe('Table of contents', function () {
 
     $closeButton = $toc.find('.js-toc-close')
     $openButton = $html.find('.js-toc-show')
+
+    $tocStickyHeader = $html.find('.toc-show')
   })
 
   afterEach(function () {
     // clear up any classes left on <html>
-    $html.removeClass('.toc-open')
+    $html.removeClass('toc-open')
     $html.find('body #toc-heading').remove()
     $html.find('body .toc').remove()
+    if ($tocStickyHeader && $tocStickyHeader.length) {
+      $tocStickyHeader.remove()
+    }
   })
 
   describe('when the module is started', function () {