1 files changed, 43 insertions, 0 deletions
diff --git a/spec/frontend/batch_comments/components/drafts_count_spec.js b/spec/frontend/batch_comments/components/drafts_count_spec.js
new file mode 100644
index 00000000000..9d9fffce7e7
--- /dev/null
+++ b/spec/frontend/batch_comments/components/drafts_count_spec.js
@@ -0,0 +1,43 @@
+import Vue from 'vue';
+import DraftsCount from '~/batch_comments/components/drafts_count.vue';
+import { mountComponentWithStore } from 'helpers/vue_mount_component_helper';
+import { createStore } from '~/batch_comments/stores';
+
+describe('Batch comments drafts count component', () => {
+  let vm;
+  let Component;
+
+  beforeAll(() => {
+    Component = Vue.extend(DraftsCount);
+  });
+
+  beforeEach(() => {
+    const store = createStore();
+
+    store.state.batchComments.drafts.push('comment');
+
+    vm = mountComponentWithStore(Component, { store });
+  });
+
+  afterEach(() => {
+    vm.$destroy();
+  });
+
+  it('renders count', () => {
+    expect(vm.$el.querySelector('.drafts-count-number').textContent).toBe('1');
+  });
+
+  it('renders screen reader text', done => {
+    const el = vm.$el.querySelector('.sr-only');
+
+    expect(el.textContent).toContain('draft');
+
+    vm.$store.state.batchComments.drafts.push('comment 2');
+
+    vm.$nextTick(() => {
+      expect(el.textContent).toContain('drafts');
+
+      done();
+    });
+  });
+});