Add unit test for switch cli

Emyrk · Emyrk · commit 3bd5f2f930b2 · 2024-02-23T11:12:52.000-06:00
diff --git a/cli/organization_test.go b/cli/organization_test.go
@@ -43,3 +43,37 @@ func TestCurrentOrganization(t *testing.T) {
 		pty.ExpectMatch(first.OrganizationID.String())
 	})
 }
+
+func TestOrganizationSwitch(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Switch", func(t *testing.T) {
+		t.Parallel()
+		ownerClient := coderdtest.New(t, nil)
+		first := coderdtest.CreateFirstUser(t, ownerClient)
+		// Owner is required to make orgs
+		client, _ := coderdtest.CreateAnotherUser(t, ownerClient, first.OrganizationID, rbac.RoleOwner())
+
+		ctx := testutil.Context(t, testutil.WaitMedium)
+		orgs := []string{"foo", "bar"}
+		for _, orgName := range orgs {
+			_, err := client.CreateOrganization(ctx, codersdk.CreateOrganizationRequest{
+				Name: orgName,
+			})
+			require.NoError(t, err)
+		}
+
+		exp, err := client.OrganizationByName(ctx, codersdk.Me, "foo")
+		require.NoError(t, err)
+
+		inv, root := clitest.New(t, "organizations", "switch", "foo")
+		clitest.SetupConfig(t, client, root)
+		pty := ptytest.New(t).Attach(inv)
+		errC := make(chan error)
+		go func() {
+			errC <- inv.Run()
+		}()
+		require.NoError(t, <-errC)
+		pty.ExpectMatch(exp.ID.String())
+	})
+}
