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
|
#include "test_framework.h"
#include <stdbool.h>
#include <string.h>
#include <stdlib.h>
typedef struct {
char url[128];
char name[64];
uint32_t latency_ms;
bool supports_nip77;
bool alive;
int consecutive_failures;
} test_relay_t;
static int compare_relays(const void *a, const void *b)
{
const test_relay_t *ra = (const test_relay_t *)a;
const test_relay_t *rb = (const test_relay_t *)b;
if (ra->alive && !rb->alive) return -1;
if (!ra->alive && rb->alive) return 1;
int score_a = (ra->supports_nip77 ? 1000 : 0) - ra->consecutive_failures * 100;
int score_b = (rb->supports_nip77 ? 1000 : 0) - rb->consecutive_failures * 100;
if (score_a != score_b) return score_b - score_a;
return (int)ra->latency_ms - (int)rb->latency_ms;
}
int main(void)
{
printf("=== test_relay_selector ===\n");
printf("\n--- NIP-77 relay preferred over non-NIP-77 ---\n");
test_relay_t relays[4] = {
{ .url = "relay_a", .latency_ms = 50, .supports_nip77 = false, .alive = true, .consecutive_failures = 0 },
{ .url = "relay_b", .latency_ms = 200, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
{ .url = "relay_c", .latency_ms = 30, .supports_nip77 = false, .alive = true, .consecutive_failures = 0 },
{ .url = "relay_d", .latency_ms = 500, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
};
qsort(relays, 4, sizeof(test_relay_t), compare_relays);
ASSERT_EQ_STR("relay_b", relays[0].url, "NIP-77 relay with 200ms beats non-NIP with 50ms");
printf("\n--- Dead relays sorted last ---\n");
test_relay_t dead_test[2] = {
{ .url = "alive_relay", .latency_ms = 500, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
{ .url = "dead_relay", .latency_ms = 10, .supports_nip77 = true, .alive = false, .consecutive_failures = 3 },
};
qsort(dead_test, 2, sizeof(test_relay_t), compare_relays);
ASSERT_EQ_STR("alive_relay", dead_test[0].url, "Alive relay sorts before dead");
printf("\n--- Latency tiebreak when same NIP-77 status ---\n");
test_relay_t tiebreak[3] = {
{ .url = "slow", .latency_ms = 300, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
{ .url = "fast", .latency_ms = 50, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
{ .url = "medium", .latency_ms = 150, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
};
qsort(tiebreak, 3, sizeof(test_relay_t), compare_relays);
ASSERT_EQ_STR("fast", tiebreak[0].url, "Fastest NIP-77 relay sorts first");
ASSERT_EQ_STR("medium", tiebreak[1].url, "Medium NIP-77 relay sorts second");
ASSERT_EQ_STR("slow", tiebreak[2].url, "Slowest NIP-77 relay sorts last");
printf("\n--- Failure penalty ---\n");
test_relay_t failures[2] = {
{ .url = "clean", .latency_ms = 200, .supports_nip77 = true, .alive = true, .consecutive_failures = 0 },
{ .url = "flaky", .latency_ms = 50, .supports_nip77 = true, .alive = true, .consecutive_failures = 2 },
};
qsort(failures, 2, sizeof(test_relay_t), compare_relays);
ASSERT_EQ_STR("clean", failures[0].url, "Clean relay beats flaky one");
printf("\n--- Non-NIP-77 relay with failures vs alive non-NIP-77 ---\n");
test_relay_t mixed[2] = {
{ .url = "ok_relay", .latency_ms = 100, .supports_nip77 = false, .alive = true, .consecutive_failures = 0 },
{ .url = "fail_relay", .latency_ms = 50, .supports_nip77 = false, .alive = true, .consecutive_failures = 3 },
};
qsort(mixed, 2, sizeof(test_relay_t), compare_relays);
ASSERT_EQ_STR("ok_relay", mixed[0].url, "Non-failing relay beats one with failures");
printf("\n=== ALL TESTS PASSED ===\n");
return 0;
}
|