iframe-proxy | Sunbelt Computer Software

192 lines (158 loc) · 5.22 KB
package cli_test
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/coder/coder/v2/cli/clitest"
	"github.com/coder/coder/v2/cli/config"
	"github.com/coder/coder/v2/coderd/coderdtest"
	"github.com/coder/coder/v2/testutil"
	"github.com/coder/coder/v2/testutil/expecter"
func TestLogout(t *testing.T) {
	t.Parallel()
	t.Run("Logout", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitMedium)
		logger := testutil.Logger(t)
		config := login(ctx, t)
		// Ensure session files exist.
		require.FileExists(t, string(config.URL()))
		require.FileExists(t, string(config.Session()))
		logoutChan := make(chan struct{})
		logout, _ := clitest.New(t, "logout", "--global-config", string(config))
		stdout := expecter.NewAttachedToInvocation(t, logout)
		stdin := testutil.NewWriterAttachedToInvocation(t, logger.Named("stdin"), logout)
		go func() {
			defer close(logoutChan)
			err := logout.Run()
			assert.NoError(t, err)
			assert.NoFileExists(t, string(config.URL()))
			assert.NoFileExists(t, string(config.Session()))
		stdout.ExpectMatch(ctx, "Are you sure you want to log out?")
		stdin.WriteLine("yes")
		stdout.ExpectMatch(ctx, "You are no longer logged in. You can log in using 'coder login <url>'.")
		<-logoutChan
	t.Run("SkipPrompt", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitMedium)
		config := login(ctx, t)
		// Ensure session files exist.
		require.FileExists(t, string(config.URL()))
		require.FileExists(t, string(config.Session()))
		logoutChan := make(chan struct{})
		logout, _ := clitest.New(t, "logout", "--global-config", string(config), "-y")
		stdout := expecter.NewAttachedToInvocation(t, logout)
		go func() {
			defer close(logoutChan)
			err := logout.Run()
			assert.NoError(t, err)
			assert.NoFileExists(t, string(config.URL()))
			assert.NoFileExists(t, string(config.Session()))
		stdout.ExpectMatch(ctx, "You are no longer logged in. You can log in using 'coder login <url>'.")
		<-logoutChan
	t.Run("NoURLFile", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitMedium)
		config := login(ctx, t)
		// Ensure session files exist.
		require.FileExists(t, string(config.URL()))
		require.FileExists(t, string(config.Session()))
		err := os.Remove(string(config.URL()))
		require.NoError(t, err)
		logoutChan := make(chan struct{})
		logout, _ := clitest.New(t, "logout", "--global-config", string(config))
		executable, err := os.Executable()
		require.NoError(t, err)
		require.NotEqual(t, "", executable)
		go func() {
			defer close(logoutChan)
			err = logout.Run()
			assert.Contains(t, err.Error(), fmt.Sprintf("Try logging in using '%s login <url>'.", executable))
		<-logoutChan
	t.Run("CannotDeleteFiles", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitMedium)
		logger := testutil.Logger(t)
		config := login(ctx, t)
		// Ensure session files exist.
		require.FileExists(t, string(config.URL()))
		require.FileExists(t, string(config.Session()))
			err         error
			urlFile     *os.File
			sessionFile *os.File
		if runtime.GOOS == "windows" {
			// Opening the files so Windows does not allow deleting them.
			urlFile, err = os.Open(string(config.URL()))
			require.NoError(t, err)
			sessionFile, err = os.Open(string(config.Session()))
			require.NoError(t, err)
			// Changing the permissions to throw error during deletion.
			err = os.Chmod(string(config), 0o500)
			require.NoError(t, err)
		defer func() {
			if runtime.GOOS == "windows" {
				// Closing the opened files for cleanup.
				err = urlFile.Close()
				assert.NoError(t, err)
				err = sessionFile.Close()
				assert.NoError(t, err)
			} else {
				// Setting the permissions back for cleanup.
				err = os.Chmod(string(config), 0o700)
				assert.NoError(t, err)
		logout, _ := clitest.New(t, "logout", "--global-config", string(config))
		stdout := expecter.NewAttachedToInvocation(t, logout)
		stdin := testutil.NewWriterAttachedToInvocation(t, logger.Named("stdin"), logout)
		go func() {
			stdout.ExpectMatch(ctx, "Are you sure you want to log out?")
			stdin.WriteLine("yes")
		err = logout.Run()
		require.Error(t, err)
		t.Logf("err: %v", err)
		var wantError string
		if runtime.GOOS == "windows" {
			wantError = "The process cannot access the file because it is being used by another process."
			wantError = "permission denied"
		require.ErrorContains(t, err, wantError)
func login(ctx context.Context, t *testing.T) config.Root {
	t.Helper()
	logger := testutil.Logger(t)
	client := coderdtest.New(t, nil)
	coderdtest.CreateFirstUser(t, client)
	doneChan := make(chan struct{})
	root, cfg := clitest.New(t, "login", "--force-tty", client.URL.String(), "--no-open")
	stdout := expecter.NewAttachedToInvocation(t, root)
	stdin := testutil.NewWriterAttachedToInvocation(t, logger.Named("stdin"), root)
	go func() {
		defer close(doneChan)
		err := root.Run()
		assert.NoError(t, err)
	stdout.ExpectMatch(ctx, "Paste your token here:")
	stdin.WriteLine(client.SessionToken())
	stdout.ExpectMatch(ctx, "Welcome to Coder")
	testutil.TryReceive(ctx, t, doneChan)
	return cfg
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

logout_test.go

Latest commit

History

logout_test.go

File metadata and controls
