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 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)
}
}