func (tf *testFixture) configuration(t *testing.T, client throttlerclient.Client) {
initialConfigs, err := client.GetConfiguration(context.Background(), "" /* all */)
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
// Test UpdateConfiguration.
config := &throttlerdata.Configuration{
TargetReplicationLagSec: 1,
MaxReplicationLagSec: 2,
InitialRate: 3,
MaxIncrease: 0.4,
EmergencyDecrease: 0.5,
MinDurationBetweenIncreasesSec: 6,
MaxDurationBetweenIncreasesSec: 7,
MinDurationBetweenDecreasesSec: 8,
SpreadBacklogAcrossSec: 9,
IgnoreNSlowestReplicas: 10,
IgnoreNSlowestRdonlys: 11,
AgeBadRateAfterSec: 12,
BadRateIncrease: 0.13,
}
names, err := client.UpdateConfiguration(context.Background(), "t2", config /* false */, true /* copyZeroValues */)
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
if got, want := names, []string{"t2"}; !reflect.DeepEqual(got, want) {
t.Fatalf("returned names of updated throttlers is wrong. got = %v, want = %v", got, want)
}
// Test GetConfiguration.
configs, err := client.GetConfiguration(context.Background(), "t2")
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
if len(configs) != 1 || configs["t2"] == nil {
t.Fatalf("wrong named configuration returned. got = %v, want configuration for t2", configs)
}
if got, want := configs["t2"], config; !proto.Equal(got, want) {
t.Fatalf("did not read updated config. got = %v, want = %v", got, want)
}
// Reset should return the initial configs.
namesForReset, err := client.ResetConfiguration(context.Background(), "" /* all */)
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
if got, want := namesForReset, throttlerNames; !reflect.DeepEqual(got, want) {
t.Fatalf("returned names of reset throttlers is wrong. got = %v, want = %v", got, want)
}
// Verify that it was correctly set.
configsAfterReset, err := client.GetConfiguration(context.Background(), "" /* all */)
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
if got, want := configsAfterReset, initialConfigs; !reflect.DeepEqual(got, want) {
t.Fatalf("wrong configurations after reset. got = %v, want = %v", got, want)
}
}
func (tf *testFixture) setMaxRate(t *testing.T, client throttlerclient.Client) {
got, err := client.SetMaxRate(context.Background(), 23)
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
if !reflect.DeepEqual(got, throttlerNames) {
t.Fatalf("rate was not updated on all registered throttlers. got = %v, want = %v", got, throttlerNames)
}
}
func (tf *testFixture) maxRates(t *testing.T, client throttlerclient.Client) {
_, err := client.SetMaxRate(context.Background(), 23)
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
got, err := client.MaxRates(context.Background())
if err != nil {
t.Fatalf("Cannot execute remote command: %v", err)
}
want := map[string]int64{
"t1": 23,
"t2": 23,
}
if !reflect.DeepEqual(got, want) {
t.Fatalf("rate was not updated on all registered throttlers. got = %v, want = %v", got, throttlerNames)
}
}
func setMaxRatePanics(t *testing.T, client throttlerclient.Client) {
_, err := client.SetMaxRate(context.Background(), 23)
if !errorFromPanicHandler(err) {
t.Fatalf("SetMaxRate RPC implementation does not catch panics properly: %v", err)
}
}
func resetConfigurationPanics(t *testing.T, client throttlerclient.Client) {
_, err := client.ResetConfiguration(context.Background(), "")
if !errorFromPanicHandler(err) {
t.Fatalf("ResetConfiguration RPC implementation does not catch panics properly: %v", err)
}
}
