call Fini in find

ktr0731 · Dec 13, 2020 · e37529d · e37529d
1 parent a19c30d
commit e37529d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/fuzzyfinder.go b/fuzzyfinder.go
@@ -619,6 +619,10 @@ func (f *finder) find(slice interface{}, itemFunc func(i int) string, opts []Opt
 		return nil, errors.Wrap(err, "failed to initialize the fuzzy finder")
 	}
 
+	if !isInTesting() {
+		defer f.term.Fini()
+	}
+
 	close(inited)
 
 	go func() {
@@ -704,7 +708,6 @@ func Find(slice interface{}, itemFunc func(i int) string, opts ...Option) (int,
 func (f *finder) Find(slice interface{}, itemFunc func(i int) string, opts ...Option) (int, error) {
 	res, err := f.find(slice, itemFunc, opts)
 	if !isInTesting() {
-		f.term.Fini()
 		if err := sendExtraEventFix(); err != nil {
 			return 0, err
 		}
@@ -726,7 +729,6 @@ func (f *finder) FindMulti(slice interface{}, itemFunc func(i int) string, opts
 	opts = append(opts, withMulti())
 	res, err := f.find(slice, itemFunc, opts)
 	if !isInTesting() {
-		f.term.Fini()
 		if err := sendExtraEventFix(); err != nil {
 			return nil, err
 		}

diff --git a/fuzzyfinder_test.go b/fuzzyfinder_test.go
@@ -206,7 +206,6 @@ func TestFind(t *testing.T) {
 				}
 
 				res := term.GetResult()
-				term.Fini()
 				return res
 			})
 		})
@@ -245,7 +244,6 @@ func TestFind_hotReload(t *testing.T) {
 		}
 
 		res := term.GetResult()
-		term.Fini()
 		return res
 	})
 }