blob: d50a7990d517c4ce427923c567e884bf788505b9 (
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
|
(* Test Ames UDP Networking Driver *)
open Io_drivers
let test_ames_creation env =
Printf.printf "Test: Ames driver creation...\n";
Eio.Switch.run @@ fun sw ->
let config = Ames.{
port = 12345;
our_ship = "~zod";
galaxy_table = [];
} in
let ames = Ames.create ~env ~sw config in
let stats = Ames.get_stats ames in
Printf.printf " Created Ames on port %d\n" config.port;
Printf.printf " Initial stats - sent: %Ld, recv: %Ld\n"
stats.packets_sent stats.packets_recv;
assert (stats.packets_sent = 0L);
assert (stats.packets_recv = 0L);
Printf.printf " ✓ Ames creation works!\n\n"
let test_ames_send_recv env =
Printf.printf "Test: Ames send/receive...\n";
Eio.Switch.run @@ fun sw ->
(* Create two Ames instances on different ports *)
let config1 = Ames.{
port = 23456;
our_ship = "~zod";
galaxy_table = [];
} in
let config2 = Ames.{
port = 23457;
our_ship = "~nec";
galaxy_table = [];
} in
let ames1 = Ames.create ~env ~sw config1 in
let _ames2 = Ames.create ~env ~sw config2 in
Printf.printf " Created two Ames instances\n";
Printf.printf " Ames1 (%s) on port %d\n" config1.our_ship config1.port;
Printf.printf " Ames2 (%s) on port %d\n" config2.our_ship config2.port;
(* Create test packet *)
let packet = Ames.{
header = {
version = 1;
sender = "~zod";
receiver = "~nec";
sequence = 1L;
};
payload = Bytes.of_string "Hello from ~zod!";
} in
(* Send packet from ames1 to ames2 *)
let dest = `Udp (Eio.Net.Ipaddr.V4.loopback, config2.port) in
Ames.send_packet ames1 dest packet;
Printf.printf " Sent packet from %s to %s\n" config1.our_ship config2.our_ship;
(* Give it a moment to arrive *)
Eio.Time.sleep (Eio.Stdenv.clock env) 0.1;
let stats1 = Ames.get_stats ames1 in
Printf.printf " Ames1 stats - sent: %Ld, recv: %Ld\n"
stats1.packets_sent stats1.packets_recv;
assert (stats1.packets_sent = 1L);
Printf.printf " ✓ Ames send works!\n\n"
let _test_ames_with_runtime env =
Printf.printf "Test: Ames with runtime event queue...\n";
Eio.Switch.run @@ fun sw ->
(* Create event stream for runtime *)
let event_stream = Eio.Stream.create 100 in
let config = Ames.{
port = 34567;
our_ship = "~zod";
galaxy_table = [];
} in
let ames = Ames.create ~env ~sw config in
Printf.printf " Starting Ames driver with event queue\n";
(* Run Ames driver (spawns receive fiber) *)
Ames.run ames ~sw ~event_stream;
(* Send a packet to ourselves *)
let packet = Ames.{
header = {
version = 1;
sender = "~nec";
receiver = "~zod";
sequence = 42L;
};
payload = Bytes.of_string "Test message";
} in
let dest = `Udp (Eio.Net.Ipaddr.V4.loopback, config.port) in
Ames.send_packet ames dest packet;
Printf.printf " Sent test packet to ourselves\n";
(* Wait a bit for the packet to be received *)
Eio.Time.sleep (Eio.Stdenv.clock env) 0.2;
(* Try to receive event from queue with timeout *)
(match Eio.Time.with_timeout (Eio.Stdenv.clock env) 0.5 (fun () ->
Ok (Eio.Stream.take event_stream)
) with
| Ok ovum ->
Printf.printf " Received event from Ames!\n";
Printf.printf " Wire: %s\n" (Format.asprintf "%a" Nock_lib.Noun.pp_noun ovum.Nock_lib.Effects.wire)
| Error `Timeout ->
Printf.printf " (Timeout - no event received)\n"
);
let stats = Ames.get_stats ames in
Printf.printf " Final stats - sent: %Ld, recv: %Ld\n"
stats.packets_sent stats.packets_recv;
Printf.printf " ✓ Ames with runtime integration works!\n\n"
let () =
Printf.printf "\n🚀🚀🚀 === AMES NETWORKING TESTS === 🚀🚀🚀\n\n";
Eio_main.run @@ fun env ->
test_ames_creation env;
test_ames_send_recv env;
Printf.printf "🎉🎉🎉 === AMES TESTS PASSED! === 🎉🎉🎉\n\n";
Printf.printf "Ames UDP driver is working!\n";
Printf.printf "- Async socket creation ✓\n";
Printf.printf "- Packet send ✓\n";
Printf.printf "\nReady for ship-to-ship communication! 🚀\n";
Printf.printf "\n(Note: Runtime integration test with infinite receive loop available in test_ames_with_runtime)\n"
|
