2 files changed, 106 insertions, 5 deletions
diff --git a/test/unit/specs/modules/statuses.spec.js b/test/unit/specs/modules/statuses.spec.js
index 891423ca..d25cc108 100644
--- a/test/unit/specs/modules/statuses.spec.js
+++ b/test/unit/specs/modules/statuses.spec.js
@@ -125,18 +125,19 @@ describe('The Statuses module', () => {
   it('removes statuses by tag on deletion', () => {
     const state = cloneDeep(defaultState)
     const status = makeMockStatus({id: 1})
+    const otherStatus = makeMockStatus({id: 3})
     status.uri = 'xxx'
     const deletion = makeMockStatus({id: 2, is_post_verb: false})
     deletion.text = 'Dolus deleted notice {{tag:gs.smuglo.li,2016-11-18:noticeId=1038007:objectType=note}}.'
     deletion.uri = 'xxx'
 
-    mutations.addNewStatuses(state, { statuses: [status], showImmediately: true, timeline: 'public' })
+    mutations.addNewStatuses(state, { statuses: [status, otherStatus], showImmediately: true, timeline: 'public' })
     mutations.addNewStatuses(state, { statuses: [deletion], showImmediately: true, timeline: 'public' })
 
-    expect(state.allStatuses).to.eql([])
-    expect(state.timelines.public.statuses).to.eql([])
-    expect(state.timelines.public.visibleStatuses).to.eql([])
-    expect(state.timelines.public.maxId).to.eql(2)
+    expect(state.allStatuses).to.eql([otherStatus])
+    expect(state.timelines.public.statuses).to.eql([otherStatus])
+    expect(state.timelines.public.visibleStatuses).to.eql([otherStatus])
+    expect(state.timelines.public.maxId).to.eql(3)
   })
 
   it('does not update the maxId when the noIdUpdate flag is set', () => {
@@ -319,6 +320,36 @@ describe('The Statuses module', () => {
       expect(state.notifications[0].type).to.eql('mention')
     })
 
+    it('removes a notification when the notice gets removed', () => {
+      const user = { id: 1 }
+      const state = cloneDeep(defaultState)
+      const status = makeMockStatus({id: 1})
+      const otherStatus = makeMockStatus({id: 3})
+      const mentionedStatus = makeMockStatus({id: 2})
+      mentionedStatus.attentions = [user]
+      mentionedStatus.uri = 'xxx'
+      otherStatus.attentions = [user]
+
+      const deletion = makeMockStatus({id: 4, is_post_verb: false})
+      deletion.text = 'Dolus deleted notice {{tag:gs.smuglo.li,2016-11-18:noticeId=1038007:objectType=note}}.'
+      deletion.uri = 'xxx'
+
+      mutations.addNewStatuses(state, { statuses: [status, otherStatus], user })
+
+      expect(state.notifications.length).to.eql(1)
+
+      mutations.addNewStatuses(state, { statuses: [mentionedStatus], user })
+      expect(state.allStatuses.length).to.eql(3)
+      expect(state.notifications.length).to.eql(2)
+      expect(state.notifications[1].status).to.eql(mentionedStatus)
+      expect(state.notifications[1].action).to.eql(mentionedStatus)
+      expect(state.notifications[1].type).to.eql('mention')
+
+      mutations.addNewStatuses(state, { statuses: [deletion], user })
+      expect(state.allStatuses.length).to.eql(2)
+      expect(state.notifications.length).to.eql(1)
+    })
+
     it('adds the message to mentions when you are mentioned', () => {
       const user = { id: 1 }
       const state = cloneDeep(defaultState)
diff --git a/test/unit/specs/services/completion/completion.spec.js b/test/unit/specs/services/completion/completion.spec.js
new file mode 100644
index 00000000..8a41c653
--- /dev/null
+++ b/test/unit/specs/services/completion/completion.spec.js
@@ -0,0 +1,70 @@
+import { replaceWord, addPositionToWords, wordAtPosition, splitIntoWords } from '../../../../../src/services/completion/completion.js'
+
+describe('addPositiontoWords', () => {
+  it('adds the position to a word list', () => {
+    const words = ['hey', 'this', 'is', 'fun']
+
+    const expected = [
+      {
+        word: 'hey',
+        start: 0,
+        end: 3
+      },
+      {
+        word: 'this',
+        start: 3,
+        end: 7
+      },
+      {
+        word: 'is',
+        start: 7,
+        end: 9
+      },
+      {
+        word: 'fun',
+        start: 9,
+        end: 12
+      }
+    ]
+
+    const res = addPositionToWords(words)
+
+    expect(res).to.eql(expected)
+  })
+})
+
+describe('splitIntoWords', () => {
+  it('splits at whitespace boundaries', () => {
+    const str = 'This is a #nice @test for you, @idiot.'
+    const expected = ['This', ' ', 'is', ' ', 'a', ' ', '#nice', ' ', '@test', ' ', 'for', ' ', 'you', ', ', '@idiot', '.']
+    const res = splitIntoWords(str)
+
+    expect(res).to.eql(expected)
+  })
+})
+
+describe('wordAtPosition', () => {
+  it('returns the word for a given string and postion, plus the start and end position of that word', () => {
+    const str = 'Hey this is fun'
+
+    const { word, start, end } = wordAtPosition(str, 4)
+
+    expect(word).to.eql('this')
+    expect(start).to.eql(4)
+    expect(end).to.eql(8)
+  })
+})
+
+describe('replaceWord', () => {
+  it('replaces a word (with start and end) with another word in a given string', () => {
+    const str = 'hey @take, how are you'
+    const wordsWithPosition = addPositionToWords(splitIntoWords(str))
+    const toReplace = wordsWithPosition[2]
+
+    expect(toReplace.word).to.eql('@take')
+
+    const expected = 'hey @takeshitakenji, how are you'
+    const res = replaceWord(str, toReplace, '@takeshitakenji')
+    expect(res).to.eql(expected)
+  })
+})