1 files changed, 54 insertions, 0 deletions
diff --git a/workhorse/internal/lsif_transformer/parser/docs_test.go b/workhorse/internal/lsif_transformer/parser/docs_test.go
new file mode 100644
index 00000000000..57dca8e773d
--- /dev/null
+++ b/workhorse/internal/lsif_transformer/parser/docs_test.go
@@ -0,0 +1,54 @@
+package parser
+
+import (
+	"bytes"
+	"fmt"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func createLine(id, label, uri string) []byte {
+	return []byte(fmt.Sprintf(`{"id":"%s","label":"%s","uri":"%s"}`+"\n", id, label, uri))
+}
+
+func TestParse(t *testing.T) {
+	d, err := NewDocs(Config{})
+	require.NoError(t, err)
+	defer d.Close()
+
+	data := []byte(`{"id":"1","label":"metaData","projectRoot":"file:///Users/nested"}` + "\n")
+	data = append(data, createLine("2", "document", "file:///Users/nested/file.rb")...)
+	data = append(data, createLine("3", "document", "file:///Users/nested/folder/file.rb")...)
+	data = append(data, createLine("4", "document", "file:///Users/wrong/file.rb")...)
+
+	require.NoError(t, d.Parse(bytes.NewReader(data)))
+
+	require.Equal(t, d.Entries[2], "file.rb")
+	require.Equal(t, d.Entries[3], "folder/file.rb")
+	require.Equal(t, d.Entries[4], "file:///Users/wrong/file.rb")
+}
+
+func TestParseContainsLine(t *testing.T) {
+	d, err := NewDocs(Config{})
+	require.NoError(t, err)
+	defer d.Close()
+
+	data := []byte(`{"id":"5","label":"contains","outV":"1", "inVs": ["2", "3"]}` + "\n")
+	data = append(data, []byte(`{"id":"6","label":"contains","outV":"1", "inVs": [4]}`+"\n")...)
+
+	require.NoError(t, d.Parse(bytes.NewReader(data)))
+
+	require.Equal(t, []Id{2, 3, 4}, d.DocRanges[1])
+}
+
+func TestParsingVeryLongLine(t *testing.T) {
+	d, err := NewDocs(Config{})
+	require.NoError(t, err)
+	defer d.Close()
+
+	line := []byte(`{"id": "` + strings.Repeat("a", 64*1024) + `"}`)
+
+	require.NoError(t, d.Parse(bytes.NewReader(line)))
+}