Refactor test to make it even more generic. Remove extra line in stru…

…ct def Signed-off-by: Rafael Chacon <[email protected]>
tinyspeck · Dec 17, 2018 · a51c547 · a51c547
1 parent 0befc7e
commit a51c547
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/go/vt/discovery/tablet_stats_cache.go b/go/vt/discovery/tablet_stats_cache.go
@@ -70,7 +70,6 @@ type tabletStatsCacheEntry struct {
 	healthy []*TabletStats
 	// aggregates has the per-cell aggregates.
 	aggregates map[string]*querypb.AggregateStats
-
 	// aggregatesPerRegion has the per-region aggregates.
 	aggregatesPerRegion map[string]*querypb.AggregateStats
 }

diff --git a/go/vt/vtgate/gateway/discoverygateway_test.go b/go/vt/vtgate/gateway/discoverygateway_test.go
@@ -347,12 +347,7 @@ func benchmarkCellsGetAggregateStats(i int, b *testing.B) {
 	cellsToregions := make(map[string]string)
 	for j := 0; j < i; j++ {
 		cell := fmt.Sprintf("cell%v", j)
-		// let's assume that there are few local cells
-		if j <= 4 {
-			cellsToregions[cell] = "local"
-		} else {
-			cellsToregions[cell] = "remote"
-		}
+		cellsToregions[cell] = "local"
 	}
 
 	topo.UpdateCellsToRegionsForTests(cellsToregions)