Function TestCompleteContextNames

cli/command/context/completion_test.go:36–80

(t *testing.T)

}

func TestCompleteContextNames(t *testing.T) {
	allNames := []string{"context-b", "context-c", "context-a"}
	cli := &fakeContextProvider{
		contextStore: fakeContextStore{
			names: allNames,
		},
	}

	t.Run("with limit", func(t *testing.T) {
		compFunc := completeContextNames(cli, 1, false)
		values, directives := compFunc(nil, nil, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveNoFileComp))
		assert.Check(t, is.DeepEqual(values, allNames))

		values, directives = compFunc(nil, []string{"context-c"}, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveNoFileComp))
		assert.Check(t, is.Len(values, 0))
	})

	t.Run("with limit and file completion", func(t *testing.T) {
		compFunc := completeContextNames(cli, 1, true)
		values, directives := compFunc(nil, nil, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveNoFileComp))
		assert.Check(t, is.DeepEqual(values, allNames))

		values, directives = compFunc(nil, []string{"context-c"}, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveDefault), "should provide filenames completion after limit")
		assert.Check(t, is.Len(values, 0))
	})

	t.Run("without limits", func(t *testing.T) {
		compFunc := completeContextNames(cli, -1, false)
		values, directives := compFunc(nil, []string{"context-c"}, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveNoFileComp))
		assert.Check(t, is.DeepEqual(values, []string{"context-b", "context-a"}), "should not contain already completed")

		values, directives = compFunc(nil, []string{"context-c", "context-a"}, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveNoFileComp))
		assert.Check(t, is.DeepEqual(values, []string{"context-b"}), "should not contain already completed")

		values, directives = compFunc(nil, []string{"context-c", "context-a", "context-b"}, "")
		assert.Check(t, is.Equal(directives, cobra.ShellCompDirectiveNoFileComp), "should provide filenames completion after limit")
		assert.Check(t, is.Len(values, 0))
	})
}