fix: tests

mrjosh · Mar 3, 2024 · 8504637 · 8504637
1 parent ec94307
commit 8504637
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/internal/adapter/yamlls/trimTemplate.go b/internal/adapter/yamlls/trimTemplate.go
@@ -1,9 +1,7 @@
 package yamlls
 
 import (
-	// "github.com/mrjosh/helm-ls/internal/lsp"
 	"github.com/mrjosh/helm-ls/internal/lsp"
-	"github.com/mrjosh/helm-ls/internal/tree-sitter/gotemplate"
 	sitter "github.com/smacker/go-tree-sitter"
 )
 

diff --git a/internal/lsp/document.go b/internal/lsp/document.go
@@ -52,6 +52,9 @@ func (s *DocumentStore) DidOpen(params lsp.DidOpenTextDocumentParams, helmlsConf
 func (s *DocumentStore) Get(docuri uri.URI) (*Document, bool) {
 	path := docuri.Filename()
 	d, ok := s.documents.Load(path)
+	if !ok {
+		return nil, false
+	}
 	return d.(*Document), ok
 }