Merge pull request #54 from posener/split-last-equal

Add logic to complete when last flag uses the equal sign

1 files changed, 20 insertions, 0 deletions

diff --git a/complete_test.go b/complete_test.go
index 19bc688..1611ad4 100644
--- a/complete_test.go
+++ b/complete_test.go
@@ -147,14 +147,30 @@ func TestCompleter_Complete(t *testing.T) {
 			want: []string{"./a.txt", "./b.txt", "./c.txt", "./.dot.txt", "./", "./dir/", "./outer/"},
 		},
 		{
+			args: "-o=./",
+			want: []string{"./a.txt", "./b.txt", "./c.txt", "./.dot.txt", "./", "./dir/", "./outer/"},
+		},
+		{
 			args: "-o .",
 			want: []string{"./a.txt", "./b.txt", "./c.txt", "./.dot.txt", "./", "./dir/", "./outer/"},
 		},
 		{
+			args: "-o ./b",
+			want: []string{"./b.txt"},
+		},
+		{
+			args: "-o=./b",
+			want: []string{"./b.txt"},
+		},
+		{
 			args: "-o ./read",
 			want: []string{},
 		},
 		{
+			args: "-o=./read",
+			want: []string{},
+		},
+		{
 			args: "-o ./readme.md",
 			want: []string{},
 		},
@@ -163,6 +179,10 @@ func TestCompleter_Complete(t *testing.T) {
 			want: []string{"sub1", "sub2"},
 		},
 		{
+			args: "-o=./readme.md ",
+			want: []string{"sub1", "sub2"},
+		},
+		{
 			args: "-o sub2 -flag3 ",
 			want: []string{"opt1", "opt2", "opt12"},
 		},