@@ -88,103 +88,74 @@ pub async fn handle_event(event: Event, stats_repository: &Repository) {
88
88
89
89
#[ cfg( test) ]
90
90
mod tests {
91
- use crate :: statistics:: event:: handler:: handle_event;
92
- use crate :: statistics:: event:: Event ;
93
- use crate :: statistics:: repository:: Repository ;
91
+ use crate :: packages :: udp_tracker_core :: statistics:: event:: handler:: handle_event;
92
+ use crate :: packages :: udp_tracker_core :: statistics:: event:: Event ;
93
+ use crate :: packages :: udp_tracker_core :: statistics:: repository:: Repository ;
94
94
95
95
#[ tokio:: test]
96
- async fn should_increase_the_udp_abort_counter_when_it_receives_a_udp_abort_event ( ) {
96
+ async fn should_increase_the_udp4_connections_counter_when_it_receives_a_udp4_connect_event ( ) {
97
97
let stats_repository = Repository :: new ( ) ;
98
98
99
- handle_event ( Event :: UdpRequestAborted , & stats_repository) . await ;
100
- let stats = stats_repository. get_stats ( ) . await ;
101
- assert_eq ! ( stats. udp_requests_aborted, 1 ) ;
102
- }
103
- #[ tokio:: test]
104
- async fn should_increase_the_udp_ban_counter_when_it_receives_a_udp_banned_event ( ) {
105
- let stats_repository = Repository :: new ( ) ;
106
-
107
- handle_event ( Event :: UdpRequestBanned , & stats_repository) . await ;
108
- let stats = stats_repository. get_stats ( ) . await ;
109
- assert_eq ! ( stats. udp_requests_banned, 1 ) ;
110
- }
111
-
112
- #[ tokio:: test]
113
- async fn should_increase_the_udp4_requests_counter_when_it_receives_a_udp4_request_event ( ) {
114
- let stats_repository = Repository :: new ( ) ;
115
-
116
- handle_event ( Event :: Udp4IncomingRequest , & stats_repository) . await ;
99
+ handle_event ( Event :: Udp4Connect , & stats_repository) . await ;
117
100
118
101
let stats = stats_repository. get_stats ( ) . await ;
119
102
120
- assert_eq ! ( stats. udp4_requests , 1 ) ;
103
+ assert_eq ! ( stats. udp4_connections_handled , 1 ) ;
121
104
}
122
105
123
106
#[ tokio:: test]
124
- async fn should_increase_the_udp4_responses_counter_when_it_receives_a_udp4_response_event ( ) {
107
+ async fn should_increase_the_udp4_announces_counter_when_it_receives_a_udp4_announce_event ( ) {
125
108
let stats_repository = Repository :: new ( ) ;
126
109
127
- handle_event (
128
- Event :: Udp4Response {
129
- kind : crate :: statistics:: event:: UdpResponseKind :: Announce ,
130
- req_processing_time : std:: time:: Duration :: from_secs ( 1 ) ,
131
- } ,
132
- & stats_repository,
133
- )
134
- . await ;
110
+ handle_event ( Event :: Udp4Announce , & stats_repository) . await ;
135
111
136
112
let stats = stats_repository. get_stats ( ) . await ;
137
113
138
- assert_eq ! ( stats. udp4_responses , 1 ) ;
114
+ assert_eq ! ( stats. udp4_announces_handled , 1 ) ;
139
115
}
140
116
141
117
#[ tokio:: test]
142
- async fn should_increase_the_udp4_errors_counter_when_it_receives_a_udp4_error_event ( ) {
118
+ async fn should_increase_the_udp4_scrapes_counter_when_it_receives_a_udp4_scrape_event ( ) {
143
119
let stats_repository = Repository :: new ( ) ;
144
120
145
- handle_event ( Event :: Udp4Error , & stats_repository) . await ;
121
+ handle_event ( Event :: Udp4Scrape , & stats_repository) . await ;
146
122
147
123
let stats = stats_repository. get_stats ( ) . await ;
148
124
149
- assert_eq ! ( stats. udp4_errors_handled , 1 ) ;
125
+ assert_eq ! ( stats. udp4_scrapes_handled , 1 ) ;
150
126
}
151
127
152
128
#[ tokio:: test]
153
- async fn should_increase_the_udp6_requests_counter_when_it_receives_a_udp6_request_event ( ) {
129
+ async fn should_increase_the_udp6_connections_counter_when_it_receives_a_udp6_connect_event ( ) {
154
130
let stats_repository = Repository :: new ( ) ;
155
131
156
- handle_event ( Event :: Udp6IncomingRequest , & stats_repository) . await ;
132
+ handle_event ( Event :: Udp6Connect , & stats_repository) . await ;
157
133
158
134
let stats = stats_repository. get_stats ( ) . await ;
159
135
160
- assert_eq ! ( stats. udp6_requests , 1 ) ;
136
+ assert_eq ! ( stats. udp6_connections_handled , 1 ) ;
161
137
}
162
138
163
139
#[ tokio:: test]
164
- async fn should_increase_the_udp6_response_counter_when_it_receives_a_udp6_response_event ( ) {
140
+ async fn should_increase_the_udp6_announces_counter_when_it_receives_a_udp6_announce_event ( ) {
165
141
let stats_repository = Repository :: new ( ) ;
166
142
167
- handle_event (
168
- Event :: Udp6Response {
169
- kind : crate :: statistics:: event:: UdpResponseKind :: Announce ,
170
- req_processing_time : std:: time:: Duration :: from_secs ( 1 ) ,
171
- } ,
172
- & stats_repository,
173
- )
174
- . await ;
143
+ handle_event ( Event :: Udp6Announce , & stats_repository) . await ;
175
144
176
145
let stats = stats_repository. get_stats ( ) . await ;
177
146
178
- assert_eq ! ( stats. udp6_responses , 1 ) ;
147
+ assert_eq ! ( stats. udp6_announces_handled , 1 ) ;
179
148
}
149
+
180
150
#[ tokio:: test]
181
- async fn should_increase_the_udp6_errors_counter_when_it_receives_a_udp6_error_event ( ) {
151
+ async fn should_increase_the_udp6_scrapes_counter_when_it_receives_a_udp6_scrape_event ( ) {
182
152
let stats_repository = Repository :: new ( ) ;
183
153
184
- handle_event ( Event :: Udp6Error , & stats_repository) . await ;
154
+ handle_event ( Event :: Udp6Scrape , & stats_repository) . await ;
185
155
186
156
let stats = stats_repository. get_stats ( ) . await ;
187
157
188
- assert_eq ! ( stats. udp6_errors_handled, 1 ) ;
158
+ assert_eq ! ( stats. udp6_scrapes_handled, 1 ) ;
159
+ >>>>>>> parent of 96 c568d8 ( ✅ tests for udp event stats) : src/packages/udp_tracker_core/statistics/event/handler. rs
189
160
}
190
161
}
0 commit comments