Lots more changes

2026-06-01 01:29:47 -05:00
parent 8bec7f0dda
commit 050ec138f3
20 changed files with 1773 additions and 689 deletions
@@ -162,7 +162,7 @@ func TestReadRoutesAreRateLimitedWhenEnabled(t *testing.T) {

 	createReq := httptest.NewRequest(http.MethodPost, "/api/scratch", bytes.NewBufferString("hello scratch"))
 	createReq.Header.Set("Content-Type", "text/plain; charset=utf-8")
-	createReq.RemoteAddr = "127.0.0.1:9111"
+	createReq.RemoteAddr = "127.0.0.2:9111"
 	createRec := httptest.NewRecorder()
 	handler.ServeHTTP(createRec, createReq)
 	if createRec.Code != http.StatusCreated {
@@ -1041,7 +1041,7 @@ func newTestHandlerAndStore(t *testing.T, opts testServerOptions) (http.Handler,
 	t.Helper()

 	dataDir := filepath.Join(t.TempDir(), "data")
-	store, err := storage.NewFilesystemStore(dataDir)
+	store, err := storage.NewFilesystemStore(dataDir, testStorageKey)
 	if err != nil {
 		t.Fatalf("NewFilesystemStore() error = %v", err)
 	}
@@ -1060,7 +1060,17 @@ func newTestHandlerAndStore(t *testing.T, opts testServerOptions) (http.Handler,
 			DataDir: dataDir,
 		},
 		Security: config.SecurityConfig{
-			RateLimit: config.RateLimitConfig{
+			RateLimitUI: config.RateLimitConfig{
+				Enabled:           opts.rateLimitEnable,
+				RequestsPerMinute: 120,
+				Burst:             1,
+			},
+			RateLimitAPIRead: config.RateLimitConfig{
+				Enabled:           opts.rateLimitEnable,
+				RequestsPerMinute: 120,
+				Burst:             1,
+			},
+			RateLimitAPIWrite: config.RateLimitConfig{
 				Enabled:           opts.rateLimitEnable,
 				RequestsPerMinute: 120,
 				Burst:             1,