pzp-db/test/tangle.test.js

const test = require('tape')
const path = require('path')
const os = require('os')
const rimraf = require('rimraf')
const SecretStack = require('secret-stack')
const caps = require('ssb-caps')
const p = require('util').promisify
const { generateKeypair } = require('./util')
const Tangle = require('../lib/tangle')

const DIR = path.join(os.tmpdir(), 'ppppp-db-tangle')
rimraf.sync(DIR)

let peer
let rootPost, reply1Lo, reply1Hi, reply2A, reply3Lo, reply3Hi
test('setup', async (t) => {
  const keysA = generateKeypair('alice')
  const keysB = generateKeypair('bob')
  const keysC = generateKeypair('carol')

  peer = SecretStack({ appKey: caps.shs })
    .use(require('../'))
    .use(require('ssb-box'))
    .call(null, { keys: keysA, path: DIR })

  await peer.db.loaded()

  // Slow down append so that we can create msgs in parallel
  const originalAppend = peer.db._getLog().append
  peer.db._getLog().append = function (...args) {
    setTimeout(originalAppend, 20, ...args)
  }

  rootPost = (
    await p(peer.db.create)({
      keys: keysA,
      type: 'comment',
      content: { text: 'root' },
    })
  ).hash

  const [{ hash: reply1B }, { hash: reply1C }] = await Promise.all([
    p(peer.db.create)({
      keys: keysB,
      type: 'comment',
      content: { text: 'reply 1' },
      tangles: [rootPost],
    }),
    p(peer.db.create)({
      keys: keysC,
      type: 'comment',
      content: { text: 'reply 1' },
      tangles: [rootPost],
    }),
  ])
  reply1Lo = reply1B.localeCompare(reply1C) < 0 ? reply1B : reply1C
  reply1Hi = reply1B.localeCompare(reply1C) < 0 ? reply1C : reply1B

  reply2A = (
    await p(peer.db.create)({
      keys: keysA,
      type: 'comment',
      content: { text: 'reply 2' },
      tangles: [rootPost],
    })
  ).hash

  const [{ hash: reply3B }, { hash: reply3C }] = await Promise.all([
    p(peer.db.create)({
      keys: keysB,
      type: 'comment',
      content: { text: 'reply 3' },
      tangles: [rootPost],
    }),
    p(peer.db.create)({
      keys: keysC,
      type: 'comment',
      content: { text: 'reply 3' },
      tangles: [rootPost],
    }),
  ])
  reply3Lo = reply3B.localeCompare(reply3C) < 0 ? reply3B : reply3C
  reply3Hi = reply3B.localeCompare(reply3C) < 0 ? reply3C : reply3B
})

test('Tangle.has', (t) => {
  const tangle = new Tangle(rootPost, peer.db.records())

  t.true(tangle.has(rootPost), 'has rootPost')
  t.true(tangle.has(reply1Lo), 'has reply1Lo')
  t.true(tangle.has(reply1Hi), 'has reply1Hi')
  t.true(tangle.has(reply2A), 'has reply2A')
  t.true(tangle.has(reply3Lo), 'has reply3Lo')
  t.true(tangle.has(reply3Hi), 'has reply3Hi')
  t.false(tangle.has('nonsense'), 'does not have nonsense')
  t.end()
})

test('Tangle.getDepth', t=> {
  const tangle = new Tangle(rootPost, peer.db.records())
  t.equals(tangle.getDepth(rootPost), 0, 'depth of rootPost is 0')
  t.equals(tangle.getDepth(reply1Lo), 1, 'depth of reply1Lo is 1')
  t.equals(tangle.getDepth(reply1Hi), 1, 'depth of reply1Hi is 1')
  t.equals(tangle.getDepth(reply2A), 2, 'depth of reply2A is 2')
  t.equals(tangle.getDepth(reply3Lo), 3, 'depth of reply3Lo is 3')
  t.equals(tangle.getDepth(reply3Hi), 3, 'depth of reply3Hi is 3')
  t.end()
})

test('Tangle.getMaxDepth', t => {
  const tangle = new Tangle(rootPost, peer.db.records())
  t.equals(tangle.getMaxDepth(), 3, 'max depth is 3')
  t.end()
})

test('Tangle.topoSort', (t) => {
  const tangle = new Tangle(rootPost, peer.db.records())
  const sorted = tangle.topoSort()

  t.deepEquals(sorted, [
    rootPost,
    reply1Lo,
    reply1Hi,
    reply2A,
    reply3Lo,
    reply3Hi,
  ])
  t.end()
})

test('Tangle.getTips', (t) => {
  const tangle = new Tangle(rootPost, peer.db.records())
  const tips = tangle.getTips()

  t.equals(tips.size, 2, 'there are 2 tips')
  t.true(tips.has(reply3Lo), 'tips contains reply3Lo')
  t.true(tips.has(reply3Hi), 'tips contains reply3Hi')
  t.end()
})

test('Tangle.getLipmaaSet', (t) => {
  const tangle = new Tangle(rootPost, peer.db.records())
  t.equals(tangle.getLipmaaSet(0).size, 0, 'lipmaa 0 (empty)')

  t.equals(tangle.getLipmaaSet(1).size, 1, 'lipmaa 1 (-1)')
  t.true(tangle.getLipmaaSet(1).has(rootPost), 'lipmaa 1 (-1)')

  t.equals(tangle.getLipmaaSet(2).size, 2, 'lipmaa 2 (-1)')
  t.true(tangle.getLipmaaSet(2).has(reply1Lo), 'lipmaa 2 (-1)')
  t.true(tangle.getLipmaaSet(2).has(reply1Hi), 'lipmaa 2 (-1)')

  t.equals(tangle.getLipmaaSet(3).size, 1, 'lipmaa 3 (leap!)')
  t.true(tangle.getLipmaaSet(3).has(rootPost), 'lipmaa 3 (leap!)')

  t.equals(tangle.getLipmaaSet(4).size, 2, 'lipmaa 4 (-1)')
  t.true(tangle.getLipmaaSet(4).has(reply3Lo), 'lipmaa 4 (-1)')
  t.true(tangle.getLipmaaSet(4).has(reply3Hi), 'lipmaa 4 (-1)')

  t.equals(tangle.getLipmaaSet(5).size, 0, 'lipmaa 5 (empty)')

  t.end()
})

test('Tangle.getDeletablesAndErasables basic', (t) => {
  const tangle = new Tangle(rootPost, peer.db.records())
  const { deletables, emptyables } = tangle.getDeletablesAndErasables(reply2A)

  t.deepEquals(deletables, [reply1Hi], 'deletables')
  t.deepEquals(emptyables, [reply1Lo, rootPost], 'emptyables')
  t.end()
})

test('Tangle.getDeletablesAndErasables with lipmaa', (t) => {
  const tangle = new Tangle(rootPost, peer.db.records())
  const { deletables, emptyables } = tangle.getDeletablesAndErasables(reply3Lo)

  t.deepEquals(deletables, [reply1Lo, reply1Hi, reply2A], 'deletables')
  t.deepEquals(emptyables, [rootPost], 'emptyables')
  t.end()
})

test('teardown', async (t) => {
  await p(peer.close)(true)
})