Go로 PostgreSQL 프록시 만들기 (61) - E2E 커버리지 갭 분석과 미커버 기능 테스트

t.Run("RouteSelectToWriter", func(t *testing.T) {
    var result int
    err := db.QueryRowContext(ctx, "/* route:writer */ SELECT 1").Scan(&result)
    if err != nil {
        t.Fatalf("route hint SELECT failed: %v", err)
    }
})

writerBefore := getMetricValue(t, client, "pgmux_queries_routed_total", `target="writer"`)

// /* route:writer */ SELECT 실행

writerAfter := getMetricValue(t, client, "pgmux_queries_routed_total", `target="writer"`)
if writerAfter <= writerBefore {
    t.Error("writer metric did not increase")
}

t.Run("TimeoutCancelsQuery", func(t *testing.T) {
    start := time.Now()
    _, err := db.ExecContext(ctx, "/* timeout:200ms */ SELECT pg_sleep(5)")
    elapsed := time.Since(start)

    if err == nil {
        t.Fatal("expected timeout error, got nil")
    }
    if elapsed > 3*time.Second {
        t.Errorf("query took %v, expected cancellation within ~200ms", elapsed)
    }
})

// 쿼리 실행으로 digest 적재
for i := 0; i < 5; i++ {
    db.QueryRowContext(ctx, "SELECT name FROM users WHERE id = 1")
}

// 초기화 전: entries > 0
// POST /admin/queries/reset
// 초기화 후: entries == 0

-- 이 두 쿼리는 같은 캐시 엔트리를 공유해야 한다
SELECT  name  FROM  users  WHERE  id = 1
select name from users where id = 1

// 캐시 플러시 후 시작
missBefore := getMetricValue(t, client, "pgmux_cache_misses_total", "")

// 첫 번째 쿼리 (여분 공백) → 캐시 미스
db.QueryRowContext(ctx, "SELECT  name  FROM  users  WHERE  id = 1")
missAfterFirst := getMetricValue(t, client, "pgmux_cache_misses_total", "")

// 두 번째 쿼리 (소문자, 공백 정리) → 캐시 히트 (미스 증가 없음)
db.QueryRowContext(ctx, "select name from users where id = 1")
missAfterSecond := getMetricValue(t, client, "pgmux_cache_misses_total", "")

// missAfterSecond == missAfterFirst → 캐시 히트!

slowBefore := getMetricValue(t, client, "pgmux_slow_queries_total", "")

db.ExecContext(ctx, "SELECT pg_sleep(0.2)") // 200ms > 100ms threshold

time.Sleep(200 * time.Millisecond) // 비동기 audit 처리 대기

slowAfter := getMetricValue(t, client, "pgmux_slow_queries_total", "")
// slowAfter > slowBefore

resp, err := client.Post(e2eAdminURL+"/admin/reload", "application/json", nil)
// status == 200

// 리로드 후에도 쿼리가 정상 동작해야 함
db2 := e2eCheckProxy(t)
db2.QueryRowContext(ctx, "SELECT 1")

ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
defer cancel()

start := time.Now()
_, err := db.ExecContext(ctx, "SELECT pg_sleep(10)")
elapsed := time.Since(start)

// err != nil (취소됨)
// elapsed < 3s (10초가 아닌 ~500ms에 완료)

resp, err := client.Get(e2eAdminURL + "/admin/mirror/stats")
// status == 200, JSON 파싱 가능

func getMetricValue(t *testing.T, client *http.Client, metricName, labelFilter string) float64 {
    resp, err := client.Get(e2eMetricURL)
    // ...
    for _, line := range strings.Split(string(body), "\n") {
        if strings.HasPrefix(line, "#") || line == "" {
            continue
        }
        if !strings.Contains(line, metricName) {
            continue
        }
        if labelFilter != "" && !strings.Contains(line, labelFilter) {
            continue
        }
        parts := strings.Fields(line)
        if len(parts) >= 2 {
            var val float64
            fmt.Sscanf(parts[len(parts)-1], "%f", &val)
            return val
        }
    }
    return 0
}