import test from 'tape'
import nlp from '../_lib.js'
const here = '[two/contract] '

test('contract basic', function (t) {
  let r = nlp(`he is cool.`)
  r.contract()
  t.equal(r.out('text'), `he's cool.`, here + 'expanded-contract')

  r = nlp(`he's cool.`)
  r.contract()
  t.equal(r.out('text'), `he's cool.`, here + 'contracted-contract')

  r = nlp(`please do not eat the marshmellow`)
  r.contract()
  t.equal(r.out('text'), `please don't eat the marshmellow`, here + 'expanded-contract')

  r = nlp(`please don't eat the marshmellow`)
  r.contract()
  t.equal(r.out('text'), `please don't eat the marshmellow`, here + 'contracted-contract')

  r = nlp(`i have stood`)
  r.contract()
  t.equal(r.out('text'), `i've stood`, here + 'expanded-contract')

  r = nlp(`i've stood`)
  r.contract()
  t.equal(r.out('text'), `i've stood`, here + 'contracted-contract')

  r = nlp('i am good')
  r.contract()
  t.equal(r.out('text'), `i'm good`, here + 'contract-1')
  r.contractions().expand()
  t.equal(r.out('text'), `i am good`, here + 'expand-2')
  r.contract()
  t.equal(r.out('text'), `i'm good`, here + 'contract-2')

  r.contractions().contract().contract().contract()
  t.equal(r.out('text'), `i'm good`, here + 'contract-n')

  t.end()
})

test('avoid contraction messes', function (t) {
  let doc = nlp('Tony, is').contract()
  t.equal(doc.has('is'), true, here + 'avoid-contraction 1')

  doc = nlp('(Tony) is').contract()
  t.equal(doc.has('is'), true, here + 'avoid-contraction 2')

  doc = nlp(`'Tony' is`).contract()
  t.equal(doc.has('is'), true, here + 'avoid-contraction 3')

  doc = nlp('Tony-is').contract()
  t.equal(doc.has('is'), true, here + 'avoid-contraction 4')
  const str = `Tony
is`
  doc = nlp(str).contract()
  t.equal(doc.has('is'), true, here + 'avoid-contraction 5')

  t.end()
})

test('multiple contraction in sentence', function (t) {
  let doc = nlp(`he's foo she's`)
  t.equal(doc.terms().length, 5, here + 'multi-contraction-count')
  t.equal(doc.has('he is foo she is'), true, here + 'multi-contraction-order')

  doc = nlp(`he's she's`)
  t.equal(doc.terms().length, 4, here + 'multi-contraction-count-2')
  t.equal(doc.has('he is she is'), true, here + 'multi-contraction-order-2')

  doc = nlp(`he's dead, he's dead`)
  t.equal(doc.match('he is dead').length, 2, here + 'multi-contraction-count')
  t.end()
})

test('contract multiple matches', function (t) {
  let doc = nlp('he is here and she is there')
  doc.contract()
  t.equal(doc.text(), `he's here and she's there`, here + 'two in one sentence')

  doc = nlp('we are tired and they are not')
  doc.contract()
  t.equal(doc.text(), `we're tired and they're not`, here + 'two of the same')

  doc = nlp('he is here. she is there.')
  doc.contract()
  t.equal(doc.text(), `he's here. she's there.`, here + 'two sentences')

  doc = nlp('Tony, is here and John is there')
  doc.contract()
  t.equal(doc.text(), `Tony, is here and John's there`, here + 'skip one with punctuation')
  t.end()
})