misc cleanup

Signed-off-by: Andrew Mason <[email protected]>
vitessio · Jun 1, 2021 · 8536013 · 8536013
1 parent 88b1ad4
commit 8536013
Show file tree

Hide file tree

Showing 3 changed files with 56 additions and 53 deletions.
diff --git a/go/cmd/vtctldclient/internal/command/routing_rules.go b/go/cmd/vtctldclient/internal/command/routing_rules.go
@@ -34,17 +34,19 @@ import (
 var (
 	// ApplyRoutingRules makes an ApplyRoutingRules gRPC call to a vtctld.
 	ApplyRoutingRules = &cobra.Command{
-		Use:   "ApplyRoutingRules {--rules RULES | --rules-file RULES_FILE} [--cells=c1,c2,...] [--skip-rebuild] [--dry-run]",
-		Short: "Applies the VSchema routing rules.",
-		Args:  cobra.NoArgs,
-		RunE:  commandApplyRoutingRules,
+		Use:                   "ApplyRoutingRules {--rules RULES | --rules-file RULES_FILE} [--cells=c1,c2,...] [--skip-rebuild] [--dry-run]",
+		Short:                 "Applies the VSchema routing rules.",
+		DisableFlagsInUseLine: true,
+		Args:                  cobra.NoArgs,
+		RunE:                  commandApplyRoutingRules,
 	}
 	// GetRoutingRules makes a GetRoutingRules gRPC call to a vtctld.
 	GetRoutingRules = &cobra.Command{
-		Use:   "GetRoutingRules",
-		Short: "Displays the VSchema routing rules.",
-		Args:  cobra.NoArgs,
-		RunE:  commandGetRoutingRules,
+		Use:                   "GetRoutingRules",
+		Short:                 "Displays the VSchema routing rules.",
+		DisableFlagsInUseLine: true,
+		Args:                  cobra.NoArgs,
+		RunE:                  commandGetRoutingRules,
 	}
 )
 

diff --git a/go/cmd/vtctldclient/internal/command/serving_graph.go b/go/cmd/vtctldclient/internal/command/serving_graph.go
@@ -53,10 +53,11 @@ var (
 	}
 	// RebuildVSchemaGraph makes a RebuildVSchemaGraph gRPC call to a vtctld.
 	RebuildVSchemaGraph = &cobra.Command{
-		Use:   "RebuildVSchemaGraph [--cells=c1,c2,...]",
-		Short: "Rebuilds the cell-specific SrvVSchema from the global VSchema objects in the provided cells (or all cells if none provided).",
-		Args:  cobra.NoArgs,
-		RunE:  commandRebuildVSchemaGraph,
+		Use:                   "RebuildVSchemaGraph [--cells=c1,c2,...]",
+		Short:                 "Rebuilds the cell-specific SrvVSchema from the global VSchema objects in the provided cells (or all cells if none provided).",
+		DisableFlagsInUseLine: true,
+		Args:                  cobra.NoArgs,
+		RunE:                  commandRebuildVSchemaGraph,
 	}
 )
 

diff --git a/go/vt/vtctl/grpcvtctldserver/server_test.go b/go/vt/vtctl/grpcvtctldserver/server_test.go
@@ -2859,47 +2859,6 @@ func TestGetRoutingRules(t *testing.T) {
 	}
 }
 
-func TestGetTablet(t *testing.T) {
-	t.Parallel()
-
-	ctx := context.Background()
-	ts := memorytopo.NewServer("cell1")
-	vtctld := testutil.NewVtctldServerWithTabletManagerClient(t, ts, nil, func(ts *topo.Server) vtctlservicepb.VtctldServer {
-		return NewVtctldServer(ts)
-	})
-
-	tablet := &topodatapb.Tablet{
-		Alias: &topodatapb.TabletAlias{
-			Cell: "cell1",
-			Uid:  100,
-		},
-		Hostname: "localhost",
-		Keyspace: "testkeyspace",
-		Shard:    "-",
-		Type:     topodatapb.TabletType_REPLICA,
-	}
-
-	testutil.AddTablet(ctx, t, ts, tablet, nil)
-
-	resp, err := vtctld.GetTablet(ctx, &vtctldatapb.GetTabletRequest{
-		TabletAlias: &topodatapb.TabletAlias{
-			Cell: "cell1",
-			Uid:  100,
-		},
-	})
-	assert.NoError(t, err)
-	utils.MustMatch(t, resp.Tablet, tablet)
-
-	// not found
-	_, err = vtctld.GetTablet(ctx, &vtctldatapb.GetTabletRequest{
-		TabletAlias: &topodatapb.TabletAlias{
-			Cell: "cell1",
-			Uid:  101,
-		},
-	})
-	assert.Error(t, err)
-}
-
 func TestGetSchema(t *testing.T) {
 	ctx := context.Background()
 	ts := memorytopo.NewServer("zone1")
@@ -3631,6 +3590,47 @@ func TestGetSrvVSchemas(t *testing.T) {
 	}
 }
 
+func TestGetTablet(t *testing.T) {
+	t.Parallel()
+
+	ctx := context.Background()
+	ts := memorytopo.NewServer("cell1")
+	vtctld := testutil.NewVtctldServerWithTabletManagerClient(t, ts, nil, func(ts *topo.Server) vtctlservicepb.VtctldServer {
+		return NewVtctldServer(ts)
+	})
+
+	tablet := &topodatapb.Tablet{
+		Alias: &topodatapb.TabletAlias{
+			Cell: "cell1",
+			Uid:  100,
+		},
+		Hostname: "localhost",
+		Keyspace: "testkeyspace",
+		Shard:    "-",
+		Type:     topodatapb.TabletType_REPLICA,
+	}
+
+	testutil.AddTablet(ctx, t, ts, tablet, nil)
+
+	resp, err := vtctld.GetTablet(ctx, &vtctldatapb.GetTabletRequest{
+		TabletAlias: &topodatapb.TabletAlias{
+			Cell: "cell1",
+			Uid:  100,
+		},
+	})
+	assert.NoError(t, err)
+	utils.MustMatch(t, resp.Tablet, tablet)
+
+	// not found
+	_, err = vtctld.GetTablet(ctx, &vtctldatapb.GetTabletRequest{
+		TabletAlias: &topodatapb.TabletAlias{
+			Cell: "cell1",
+			Uid:  101,
+		},
+	})
+	assert.Error(t, err)
+}
+
 func TestGetTablets(t *testing.T) {
 	t.Parallel()