add unit tests: limits configuration should be reloadable

hashicorp · Apr 8, 2018 · 255492b · 255492b
1 parent 464bad0
commit 255492b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/agent/agent_endpoint_test.go b/agent/agent_endpoint_test.go
@@ -281,6 +281,10 @@ func TestAgent_Reload(t *testing.T) {
 				handler = "true"
 			}
 		]
+    limits = {
+      rpc_rate=1
+      rpc_max_burst=100
+    }
 	`)
 	defer a.Shutdown()
 
@@ -302,6 +306,10 @@ func TestAgent_Reload(t *testing.T) {
 					name = "redis-reloaded"
 				}
 			]
+      limits = {
+        rpc_rate=2
+        rpc_max_burst=200
+      }
 		`,
 	})
 
@@ -312,6 +320,14 @@ func TestAgent_Reload(t *testing.T) {
 		t.Fatal("missing redis-reloaded service")
 	}
 
+	if a.config.RPCRateLimit != 2 {
+		t.Fatalf("RPC rate not set correctly.  Got %v. Want 2", a.config.RPCRateLimit)
+	}
+
+	if a.config.RPCMaxBurst != 200 {
+		t.Fatalf("RPC max burst not set correctly.  Got %v. Want 200", a.config.RPCMaxBurst)
+	}
+
 	for _, wp := range a.watchPlans {
 		if !wp.IsStopped() {
 			t.Fatalf("Reloading configs should stop watch plans of the previous configuration")