blob: e1a2f73e215c3207a6fc4805fcf62b8395a436b5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
#!/bin/bash
# Test script to verify the retry logic works correctly
# Simulates Schema Registry eventual consistency behavior
set -euo pipefail
# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
NC='\033[0m'
log_info() {
echo -e "${BLUE}[TEST]${NC} $1"
}
log_success() {
echo -e "${GREEN}[PASS]${NC} $1"
}
log_error() {
echo -e "${RED}[FAIL]${NC} $1"
}
# Mock function that simulates Schema Registry eventual consistency
# First N attempts fail, then succeeds
mock_schema_registry_query() {
local subject=$1
local min_attempts_to_succeed=$2
local current_attempt=$3
if [[ $current_attempt -ge $min_attempts_to_succeed ]]; then
# Simulate successful response
echo '{"id":1,"version":1,"schema":"test"}'
return 0
else
# Simulate 404 Not Found
echo '{"error_code":40401,"message":"Subject not found"}'
return 1
fi
}
# Simulate verify_schema_with_retry logic
test_verify_with_retry() {
local subject=$1
local min_attempts_to_succeed=$2
local max_attempts=5
local attempt=1
log_info "Testing $subject (should succeed after $min_attempts_to_succeed attempts)"
while [[ $attempt -le $max_attempts ]]; do
local response
if response=$(mock_schema_registry_query "$subject" "$min_attempts_to_succeed" "$attempt"); then
if echo "$response" | grep -q '"id"'; then
if [[ $attempt -gt 1 ]]; then
log_success "$subject verified after $attempt attempts"
else
log_success "$subject verified on first attempt"
fi
return 0
fi
fi
# Schema not found, wait and retry
if [[ $attempt -lt $max_attempts ]]; then
# Exponential backoff: 0.1s, 0.2s, 0.4s, 0.8s
local wait_time=$(echo "scale=3; 0.1 * (2 ^ ($attempt - 1))" | bc)
log_info " Attempt $attempt failed, waiting ${wait_time}s before retry..."
sleep "$wait_time"
attempt=$((attempt + 1))
else
log_error "$subject verification failed after $max_attempts attempts"
return 1
fi
done
return 1
}
# Run tests
log_info "=========================================="
log_info "Testing Schema Registry Retry Logic"
log_info "=========================================="
echo ""
# Test 1: Schema available immediately
log_info "Test 1: Schema available immediately"
if test_verify_with_retry "immediate-schema" 1; then
log_success "✓ Test 1 passed"
else
log_error "✗ Test 1 failed"
exit 1
fi
echo ""
# Test 2: Schema available after 2 attempts (200ms delay)
log_info "Test 2: Schema available after 2 attempts"
if test_verify_with_retry "delayed-schema-2" 2; then
log_success "✓ Test 2 passed"
else
log_error "✗ Test 2 failed"
exit 1
fi
echo ""
# Test 3: Schema available after 3 attempts (600ms delay)
log_info "Test 3: Schema available after 3 attempts"
if test_verify_with_retry "delayed-schema-3" 3; then
log_success "✓ Test 3 passed"
else
log_error "✗ Test 3 failed"
exit 1
fi
echo ""
# Test 4: Schema available after 4 attempts (1400ms delay)
log_info "Test 4: Schema available after 4 attempts"
if test_verify_with_retry "delayed-schema-4" 4; then
log_success "✓ Test 4 passed"
else
log_error "✗ Test 4 failed"
exit 1
fi
echo ""
# Test 5: Schema never available (should fail)
log_info "Test 5: Schema never available (should fail gracefully)"
if test_verify_with_retry "missing-schema" 10; then
log_error "✗ Test 5 failed (should have failed but passed)"
exit 1
else
log_success "✓ Test 5 passed (correctly failed after max attempts)"
fi
echo ""
log_success "=========================================="
log_success "All tests passed! ✓"
log_success "=========================================="
log_info ""
log_info "Summary:"
log_info "- Immediate availability: works ✓"
log_info "- 2-4 retry attempts: works ✓"
log_info "- Max attempts handling: works ✓"
log_info "- Exponential backoff: works ✓"
log_info ""
log_info "Total retry time budget: ~1.5 seconds (0.1+0.2+0.4+0.8)"
log_info "This should handle Schema Registry consumer lag gracefully."
|
