synapsecns/sanguine

package tree_test

import (
    . "github.com/stretchr/testify/assert"
    "github.com/synapsecns/sanguine/contrib/git-changes-action/detector/tree"
    "testing"
)

func TestTree(t *testing.T) {
    testPaths := []string{
        "a/b/c/d.go",
        "a/b/g",
        "a/d",
    }

    testTree := tree.NewTree()
    testTree.Add(testPaths...)

    // add again to make sure it doesn't add duplicates
    testTree.Add(testPaths...)

    for _, path := range testPaths {
        if !testTree.HasPath(path) {
            t.Errorf("tree should have path %s", path)
        }
    }

    if testTree.HasPath("a/b/c/d/e.go") {
        t.Errorf("tree should not have path a/b/c/d/e.go")
    }

    if !testTree.HasPath("a/b/c") {
        t.Errorf("tree should have path a/b/c")
    }

    expectedOutput := `.
└── a
    ├── b
    │   ├── c
    │   │   └── d.go
    │   └── g
    └── d
`

    Equal(t, expectedOutput, testTree.ToString())
    Equal(t, 6, len(testTree.AllPaths()))

    for _, path := range testPaths {
        Contains(t, testTree.AllPaths(), path)
    }
}