linux-odyssey/packages/file-graph/fileGraph.test.js

import FileGraph from './fileGraph.js'
import FileNode from './fileNode.js'

describe('FileGraph', () => {
  let root

  beforeEach(() => {
    root = new FileNode({
      path: '/',
      type: 'folder',
      discovered: true,
    })
  })

  test('add files to graph', () => {
    const fileGraph = new FileGraph(root)

    const filesToAdd = [
      {
        path: '/home/user/file1.txt',
        type: 'file',
        discovered: true,
      },
      {
        path: '/home/user/folder1',
        type: 'folder',
        discovered: true,
      },
    ]

    fileGraph.add(filesToAdd)

    console.log(fileGraph.root.toString())
    expect(fileGraph.root.children.length).toBe(1)
    expect(fileGraph.root.children[0].children[0].children.length).toBe(2)
  })

  test('remove files from graph', () => {
    const fileGraph = new FileGraph(root)

    const filesToAdd = [
      {
        path: '/home/user/folder1',
        type: 'folder',
        discovered: true,
      },
      {
        path: '/home/user/folder2',
        type: 'folder',
        discovered: true,
      },
      {
        path: '/home/user/folder1/file1.txt',
        type: 'file',
        discovered: true,
      },
      {
        path: '/home/user/folder2/file2.txt',
        type: 'file',
        discovered: true,
      },
    ]

    fileGraph.add(filesToAdd)

    const filesToRemove = [
      {
        path: '/home/user/folder1/file1.txt',
        type: 'folder',
      },
      {
        path: '/home/user/folder2',
        type: 'folder',
      },
    ]

    fileGraph.remove(filesToRemove)

    console.log(fileGraph.root.toString())
    expect(fileGraph.root.children[0].children[0].children.length).toBe(1)
    expect(fileGraph.root.children[0].children[0].children[0].name).toBe(
      'folder1'
    )
    expect(
      fileGraph.root.children[0].children[0].children[0].children.length
    ).toBe(0)
  })

  test('discover new files', () => {
    const fileGraph = new FileGraph(root)

    const filesToAdd = [
      {
        path: '/home/user/folder1',
        type: 'folder',
        discovered: true,
      },
      {
        path: '/home/user/folder2',
        type: 'folder',
        discovered: true,
      },
      {
        path: '/home/user/folder1/file1.txt',
        type: 'file',
        discovered: true,
      },
      {
        path: '/home/user/folder2/file2.txt',
        type: 'file',
        discovered: true,
      },
    ]

    fileGraph.add(filesToAdd)
    console.log(fileGraph.root.toString())

    const newFiles = [
      {
        path: '/home/user/folder1',
        type: 'folder',
        discovered: true,
      },
      {
        path: '/home/user/folder1/file1.txt',
        type: 'file',
        discovered: true,
      },
      {
        path: '/home/user/folder1/file2.txt',
        type: 'file',
        discovered: true,
      },
      {
        path: '/home/user/folder1/file3.txt',
        type: 'file',
        discovered: true,
      },
    ]

    fileGraph.discover(newFiles)

    console.log(fileGraph.root.toString())
    expect(fileGraph.root.children[0].children[0].children.length).toBe(2)
    expect(
      fileGraph.root.children[0].children[0].children[0].children.length
    ).toBe(3)
  })
})