diff --git a/src/lib/annotations/doc/DocHighlightThread.js b/src/lib/annotations/doc/DocHighlightThread.js index 51ecf1899..114c754e7 100644 --- a/src/lib/annotations/doc/DocHighlightThread.js +++ b/src/lib/annotations/doc/DocHighlightThread.js @@ -47,6 +47,8 @@ class DocHighlightThread extends AnnotationThread { this.type = TYPES.highlight; } else if (!this.isMobile) { this.destroy(); + } else { + this.reset(); } } diff --git a/src/lib/annotations/doc/__tests__/DocHighlightThread-test.js b/src/lib/annotations/doc/__tests__/DocHighlightThread-test.js index 9b6941bb7..7024dcb80 100644 --- a/src/lib/annotations/doc/__tests__/DocHighlightThread-test.js +++ b/src/lib/annotations/doc/__tests__/DocHighlightThread-test.js @@ -77,6 +77,16 @@ describe('lib/annotations/doc/DocHighlightThread', () => { expect(highlightThread.destroy).to.be.called; assert.equal(highlightThread.element, null); }); + + it('should reset the thread if on mobile and a comment-highlight', () => { + sandbox.stub(highlightThread, 'reset'); + highlightThread.annotations = [{}, {}, {}]; + highlightThread.isMobile = true; + + highlightThread.cancelFirstComment(); + + expect(highlightThread.reset).to.be.called; + }); }); describe('destroy()', () => {