rumtk_core/
lib.rs

1/*
2 * rumtk attempts to implement HL7 and medical protocols for interoperability in medicine.
3 * This toolkit aims to be reliable, simple, performant, and standards compliant.
4 * Copyright (C) 2024  Luis M. Santos, M.D.
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
19 */
20
21//#![feature(unboxed_closures)]
22#![feature(inherent_associated_types)]
23#![feature(type_alias_impl_trait)]
24#![feature(unboxed_closures)]
25#![feature(buf_read_has_data_left)]
26
27pub mod cache;
28pub mod cli;
29pub mod core;
30pub mod json;
31pub mod log;
32pub mod maths;
33pub mod net;
34pub mod queue;
35pub mod scripting;
36pub mod search;
37pub mod strings;
38pub mod threading;
39
40#[cfg(test)]
41mod tests {
42    use super::*;
43    use crate::cache::RUMCache;
44    use crate::search::rumtk_search::*;
45    use crate::strings::{RUMArrayConversions, RUMString, RUMStringConversions, StringUtils};
46    use compact_str::{format_compact, CompactString};
47    use serde::{Deserialize, Serialize};
48    use std::future::IntoFuture;
49    use std::sync::Arc;
50    use tokio::sync::RwLock;
51
52    #[test]
53    fn test_is_escaped_str() {
54        let input = "\r\n\'\\\"";
55        let expected = false;
56        let result = strings::is_escaped_str(input);
57        println!("Input: {} Expected: {} Got: {}", input, expected, result);
58        assert_eq!(
59            expected, result,
60            "Incorrect detection of unescaped string as escaped!!"
61        );
62        println!("Passed!")
63    }
64
65    #[test]
66    fn test_escaping_control() {
67        let input = "\r\n\'\"\\";
68        let expected = "\\r\\n\\'\\\"\\\\";
69        let result = strings::escape(input);
70        println!(
71            "Input: {} Expected: {} Got: {}",
72            input,
73            expected,
74            result.as_str()
75        );
76        assert_eq!(expected, result, "Incorrect string escaping!");
77        println!("Passed!")
78    }
79
80    #[test]
81    fn test_escaping_unicode() {
82        let input = "❤";
83        let expected = "\\u2764";
84        let result = strings::escape(input);
85        println!(
86            "Input: {} Expected: {} Got: {}",
87            input,
88            expected,
89            result.as_str()
90        );
91        assert_eq!(expected, result, "Incorrect string escaping!");
92        println!("Passed!")
93    }
94
95    #[test]
96    fn test_unescaping_unicode() {
97        let input = "❤";
98        let escaped = strings::escape(input);
99        let expected = "❤";
100        let result = RUMString::from_utf8(strings::unescape(escaped.as_str()).unwrap()).unwrap();
101        println!(
102            "Input: {} Expected: {} Got: {}",
103            input,
104            expected,
105            result.as_str()
106        );
107        assert_eq!(expected, result.as_str(), "Incorrect string unescaping!");
108        println!("Passed!")
109    }
110
111    #[test]
112    fn test_unescaping_string() {
113        let input = "I \\u2764 my wife!";
114        let expected = "I ❤ my wife!";
115        let result = strings::unescape_string(input).unwrap();
116        println!(
117            "Input: {} Expected: {} Got: {}",
118            input,
119            expected,
120            result.as_str()
121        );
122        assert_eq!(expected, result.as_str(), "Incorrect string unescaping!");
123        println!("Passed!")
124    }
125
126    #[test]
127    fn test_is_escaped_string() {
128        let input = "I \\u2764 my wife!";
129        let expected = true;
130        let result = strings::is_escaped_str(input);
131        println!("Input: {} Expected: {} Got: {}", input, expected, result);
132        assert_eq!(
133            expected, result,
134            "Escaped string detected as unescaped string!"
135        );
136        println!("Passed!")
137    }
138
139    #[test]
140    fn test_is_unescaped_string() {
141        let input = "I ❤ my wife!";
142        let expected = false;
143        let result = strings::is_escaped_str(input);
144        println!("Input: {} Expected: {} Got: {}", input, expected, result);
145        assert_eq!(
146            expected, result,
147            "Unescaped string detected as escaped string!"
148        );
149        println!("Passed!")
150    }
151
152    #[test]
153    fn test_unique_string() {
154        let input = "I❤mywife!";
155        assert!(input.is_unique(), "String was not detected as unique.");
156    }
157
158    #[test]
159    fn test_non_unique_string() {
160        let input = "I❤❤mywife!";
161        assert!(!input.is_unique(), "String was detected as unique.");
162    }
163
164    #[test]
165    fn test_escaping_string() {
166        let input = "I ❤ my wife!";
167        let expected = "I \\u2764 my wife!";
168        let result = strings::escape(input);
169        println!(
170            "Input: {} Expected: {} Got: {}",
171            input,
172            expected,
173            result.as_str()
174        );
175        assert_eq!(expected, result.as_str(), "Incorrect string escaping!");
176        println!("Passed!")
177    }
178
179    #[test]
180    fn test_autodecode_utf8() {
181        let input = "I ❤ my wife!";
182        let result = strings::try_decode(input.as_bytes());
183        println!(
184            "Input: {} Expected: {} Got: {}",
185            input,
186            input,
187            result.as_str()
188        );
189        assert_eq!(input, result, "Incorrect string decoding!");
190        println!("Passed!")
191    }
192
193    #[test]
194    fn test_autodecode_other() {
195        //TODO: Need an example of other encoding texts.
196        let input = "I ❤ my wife!";
197        let result = input;
198        println!("Input: {} Expected: {} Got: {}", input, input, result);
199        assert_eq!(input, result, "Incorrect string decoding!");
200        println!("Passed!")
201    }
202
203    #[test]
204    fn test_decode() {
205        let input = "I ❤ my wife!";
206        let result = strings::try_decode_with(input.as_bytes(), "utf-8");
207        println!(
208            "Input: {} Expected: {} Got: {}",
209            input,
210            input,
211            result.as_str()
212        );
213        assert_eq!(input, result, "Incorrect string decoding!");
214        println!("Passed!")
215    }
216
217    #[test]
218    fn test_rumcache_insertion() {
219        let mut cache: RUMCache<&str, CompactString> = RUMCache::with_capacity(5);
220        cache.insert("❤", CompactString::from("I ❤ my wife!"));
221        println!("Contents: {:#?}", &cache);
222        assert_eq!(cache.len(), 1, "Incorrect number of items in cache!");
223        println!("Passed!")
224    }
225
226    #[test]
227    fn test_search_string_letters() {
228        let input = "Hello World!";
229        let expr = r"\w";
230        let result = string_search(input, expr, "");
231        let expected: RUMString = RUMString::from("HelloWorld");
232        println!(
233            "Input: {:?} Expected: {:?} Got: {:?}",
234            input, expected, result
235        );
236        assert_eq!(expected, result, "String search results mismatch");
237        println!("Passed!")
238    }
239
240    #[test]
241    fn test_search_string_words() {
242        let input = "Hello World!";
243        let expr = r"\w+";
244        let result = string_search(input, expr, " ");
245        let expected: RUMString = RUMString::from("Hello World");
246        println!(
247            "Input: {:?} Expected: {:?} Got: {:?}",
248            input, expected, result
249        );
250        assert_eq!(expected, result, "String search results mismatch");
251        println!("Passed!")
252    }
253
254    #[test]
255    fn test_search_string_named_groups() {
256        let input = "Hello World!";
257        let expr = r"(?<hello>\w{5}) (?<world>\w{5})";
258        let result = string_search_named_captures(input, expr, "");
259        let expected: RUMString = RUMString::from("World");
260        println!(
261            "Input: {:?} Expected: {:?} Got: {:?}",
262            input, expected, result
263        );
264        assert_eq!(expected, result["world"], "String search results mismatch");
265        println!("Passed!")
266    }
267
268    #[test]
269    fn test_search_string_all_groups() {
270        let input = "Hello World!";
271        let expr = r"(?<hello>\w{5}) (?<world>\w{5})";
272        let result = string_search_all_captures(input, expr, "");
273        let expected: Vec<&str> = vec!["Hello", "World"];
274        println!(
275            "Input: {:?} Expected: {:?} Got: {:?}",
276            input, expected, result
277        );
278        assert_eq!(expected, result, "String search results mismatch");
279        println!("Passed!")
280    }
281
282    ///////////////////////////////////Threading Tests/////////////////////////////////////////////////
283    #[test]
284    fn test_default_num_threads() {
285        use num_cpus;
286        let threads = threading::threading_functions::get_default_system_thread_count();
287        assert_eq!(
288            threads >= num_cpus::get(),
289            true,
290            "Default thread count is incorrect! We got {}, but expected {}!",
291            threads,
292            num_cpus::get()
293        );
294    }
295
296    #[test]
297    fn test_execute_job() {
298        let rt = rumtk_init_threads!();
299        let expected = vec![1, 2, 3];
300        let task_processor = async |args: &SafeTaskArgs<i32>| -> TaskResult<i32> {
301            let owned_args = Arc::clone(args);
302            let lock_future = owned_args.read();
303            let locked_args = lock_future.await;
304            let mut results = TaskItems::<i32>::with_capacity(locked_args.len());
305            print!("Contents: ");
306            for arg in locked_args.iter() {
307                results.push(arg.clone());
308                println!("{} ", &arg);
309            }
310            Ok(results)
311        };
312        let locked_args = RwLock::new(expected.clone());
313        let task_args = SafeTaskArgs::<i32>::new(locked_args);
314        let task_result = rumtk_wait_on_task!(rt, task_processor, &task_args);
315        let result = task_result.unwrap();
316        assert_eq!(&result, &expected, "{}", format_compact!("Task processing returned a different result than expected! Expected {:?} \nResults {:?}", &expected, &result));
317    }
318
319    #[test]
320    fn test_execute_job_macros() {
321        let rt = rumtk_init_threads!();
322        let expected = vec![1, 2, 3];
323        let task_processor = async |args: &SafeTaskArgs<i32>| -> TaskResult<i32> {
324            let owned_args = Arc::clone(args);
325            let lock_future = owned_args.read();
326            let locked_args = lock_future.await;
327            let mut results = TaskItems::<i32>::with_capacity(locked_args.len());
328            print!("Contents: ");
329            for arg in locked_args.iter() {
330                results.push(arg.clone());
331                println!("{} ", &arg);
332            }
333            Ok(results)
334        };
335        let task_args = rumtk_create_task_args!(1, 2, 3);
336        let task_result = rumtk_wait_on_task!(rt, task_processor, &task_args);
337        let result = task_result.unwrap();
338        assert_eq!(&result, &expected, "{}", format_compact!("Task processing returned a different result than expected! Expected {:?} \nResults {:?}", &expected, &result));
339    }
340
341    #[test]
342    fn test_execute_job_macros_one_line() {
343        let rt = rumtk_init_threads!();
344        let expected = vec![1, 2, 3];
345        let result = rumtk_exec_task!(
346            async |args: &SafeTaskArgs<i32>| -> TaskResult<i32> {
347                let owned_args = Arc::clone(args);
348                let lock_future = owned_args.read();
349                let locked_args = lock_future.await;
350                let mut results = TaskItems::<i32>::with_capacity(locked_args.len());
351                print!("Contents: ");
352                for arg in locked_args.iter() {
353                    results.push(arg.clone());
354                    println!("{} ", &arg);
355                }
356                Ok(results)
357            },
358            vec![1, 2, 3]
359        )
360        .unwrap();
361        assert_eq!(&result, &expected, "{}", format_compact!("Task processing returned a different result than expected! Expected {:?} \nResults {:?}", &expected, &result));
362    }
363
364    #[test]
365    fn test_clamp_index_positive_index() {
366        let values = vec![1, 2, 3, 4];
367        let given_index = 3isize;
368        let max_size = values.len() as isize;
369        let index = clamp_index(&given_index, &max_size).unwrap();
370        assert_eq!(
371            index, 3,
372            "Index mismatch! Requested index {} but got {}",
373            &given_index, &index
374        );
375        assert_eq!(
376            values[index], 4,
377            "Value mismatch! Expected {} but got {}",
378            &values[3], &values[index]
379        );
380    }
381
382    #[test]
383    fn test_clamp_index_reverse_index() {
384        let values = vec![1, 2, 3, 4];
385        let given_index = -1isize;
386        let max_size = values.len() as isize;
387        let index = clamp_index(&given_index, &max_size).unwrap();
388        assert_eq!(
389            index, 4,
390            "Index mismatch! Requested index {} but got {}",
391            &given_index, &index
392        );
393        assert_eq!(
394            values[index - 1],
395            4,
396            "Value mismatch! Expected {} but got {}",
397            &values[3],
398            &values[index]
399        );
400    }
401
402    ///////////////////////////////////Queue Tests/////////////////////////////////////////////////
403    use crate::cli::cli_utils::print_license_notice;
404    use crate::core::clamp_index;
405    use crate::net::tcp::LOCALHOST;
406    use crate::threading::thread_primitives::{SafeTaskArgs, TaskItems, TaskResult};
407    use crate::threading::threading_functions::sleep;
408    use queue::queue::*;
409
410    #[test]
411    fn test_queue_data() {
412        let expected = vec![
413            RUMString::from("Hello"),
414            RUMString::from("World!"),
415            RUMString::from("Overcast"),
416            RUMString::from("and"),
417            RUMString::from("Sad"),
418        ];
419        let mut queue = TaskQueue::<RUMString>::new(&5).unwrap();
420        let locked_args = RwLock::new(expected.clone());
421        let task_args = SafeTaskArgs::<RUMString>::new(locked_args);
422        let processor = rumtk_create_task!(
423            async |args: &SafeTaskArgs<RUMString>| -> TaskResult<RUMString> {
424                let owned_args = Arc::clone(args);
425                let lock_future = owned_args.read();
426                let locked_args = lock_future.await;
427                let mut results = TaskItems::<RUMString>::with_capacity(locked_args.len());
428                print!("Contents: ");
429                for arg in locked_args.iter() {
430                    print!("{} ", &arg);
431                    results.push(RUMString::new(arg));
432                }
433                Ok(results)
434            },
435            task_args
436        );
437        queue.add_task::<_>(processor);
438        let results = queue.wait();
439        let mut result_data = Vec::<RUMString>::with_capacity(5);
440        for r in results {
441            for v in r.unwrap().iter() {
442                result_data.push(v.clone());
443            }
444        }
445        assert_eq!(result_data, expected, "Results do not match expected!");
446    }
447
448    ///////////////////////////////////Net Tests/////////////////////////////////////////////////
449    #[test]
450    fn test_server_start() {
451        let mut server = match rumtk_create_server!("localhost", 0) {
452            Ok(server) => server,
453            Err(e) => panic!("Failed to create server because {}", e),
454        };
455        match server.start(false) {
456            Ok(_) => (),
457            Err(e) => panic!("Failed to start server because {}", e),
458        }
459    }
460
461    #[test]
462    fn test_server_send() {
463        let msg = RUMString::from("Hello World!");
464        let mut server = match rumtk_create_server!(LOCALHOST, 0, 1) {
465            Ok(server) => server,
466            Err(e) => panic!("Failed to create server because {}", e),
467        };
468        match server.start(false) {
469            Ok(_) => (),
470            Err(e) => panic!("Failed to start server because {}", e),
471        };
472        let address_info = server.get_address_info().unwrap();
473        let (ip, port) = rumtk_get_ip_port!(address_info);
474        println!("Sleeping");
475        rumtk_sleep!(1);
476        let mut client = match rumtk_connect!(port) {
477            Ok(client) => client,
478            Err(e) => panic!("Failed to create server because {}", e),
479        };
480        let client_id = client.get_address().unwrap();
481        rumtk_sleep!(1);
482        match server.send(&client_id, &msg.to_raw()) {
483            Ok(_) => (),
484            Err(e) => panic!("Server failed to send message because {}", e),
485        };
486        rumtk_sleep!(1);
487        let received_message = client.receive().unwrap();
488        assert_eq!(
489            &msg.to_raw(),
490            &received_message,
491            "{}",
492            format_compact!(
493                "Received message does not match sent message by server {:?}",
494                &received_message
495            )
496        );
497    }
498
499    #[test]
500    fn test_server_receive() {
501        let msg = RUMString::from("Hello World!");
502        let mut server = match rumtk_create_server!(LOCALHOST, 0) {
503            Ok(server) => server,
504            Err(e) => panic!("Failed to create server because {}", e),
505        };
506        match server.start(false) {
507            Ok(_) => (),
508            Err(e) => panic!("Failed to start server because {}", e),
509        };
510        let address_info = server.get_address_info().unwrap();
511        let (ip, port) = rumtk_get_ip_port!(address_info);
512        println!("Sleeping");
513        rumtk_sleep!(1);
514        let mut client = match rumtk_connect!(port) {
515            Ok(client) => client,
516            Err(e) => panic!("Failed to create server because {}", e),
517        };
518        match client.send(&msg.to_raw()) {
519            Ok(_) => (),
520            Err(e) => panic!("Failed to send message because {}", e),
521        };
522        rumtk_sleep!(1);
523        let client_id = client.get_address().expect("Failed to get client id");
524        let incoming_message = server.receive(&client_id).unwrap().to_rumstring();
525        println!("Received message => {:?}", &incoming_message);
526        assert_eq!(&incoming_message, msg, "Received message corruption!");
527    }
528
529    #[test]
530    fn test_server_get_clients() {
531        let mut server = match rumtk_create_server!(LOCALHOST, 0) {
532            Ok(server) => server,
533            Err(e) => panic!("Failed to create server because {}", e),
534        };
535        match server.start(false) {
536            Ok(_) => (),
537            Err(e) => panic!("Failed to start server because {}", e),
538        };
539        let address_info = server.get_address_info().unwrap();
540        let (ip, port) = rumtk_get_ip_port!(address_info);
541        println!("Sleeping");
542        rumtk_sleep!(1);
543        let mut client = match rumtk_connect!(port) {
544            Ok(client) => client,
545            Err(e) => panic!("Failed to create server because {}", e),
546        };
547        rumtk_sleep!(1);
548        let expected_client_id = client.get_address().expect("Failed to get client id");
549        let clients = server.get_client_ids();
550        let incoming_client_id = clients.get(0).expect("Expected client to have connected!");
551        println!("Connected client id => {}", &incoming_client_id);
552        assert_eq!(
553            &incoming_client_id, &expected_client_id,
554            "Connected client does not match the connecting client! Client id => {}",
555            &incoming_client_id
556        );
557    }
558
559    #[test]
560    fn test_server_stop() {
561        let msg = RUMString::from("Hello World!");
562        let mut server = match rumtk_create_server!("localhost", 0) {
563            Ok(server) => server,
564            Err(e) => panic!("Failed to create server because {}", e),
565        };
566        match server.start(false) {
567            Ok(_) => (),
568            Err(e) => panic!("Failed to start server because {}", e),
569        };
570        println!("Sleeping");
571        rumtk_sleep!(1);
572        match server.stop() {
573            Ok(_) => (),
574            Err(e) => panic!("Failed to stop server because {}", e),
575        };
576    }
577
578    #[test]
579    fn test_server_get_address_info() {
580        let msg = RUMString::from("Hello World!");
581        let mut server = match rumtk_create_server!("localhost", 0) {
582            Ok(server) => server,
583            Err(e) => panic!("Failed to create server because {}", e),
584        };
585        match server.start(false) {
586            Ok(_) => (),
587            Err(e) => panic!("Failed to start server because {}", e),
588        };
589        println!("Sleeping");
590        rumtk_sleep!(1);
591        match server.get_address_info() {
592            Some(addr) => println!("Server address info => {}", addr),
593            None => panic!("No address. Perhaps the server was never initialized?"),
594        };
595    }
596
597    #[test]
598    fn test_client_send() {
599        let msg = RUMString::from("Hello World!");
600        let mut server = match rumtk_create_server!(LOCALHOST, 0) {
601            Ok(server) => server,
602            Err(e) => panic!("Failed to create server because {}", e),
603        };
604        match server.start(false) {
605            Ok(_) => (),
606            Err(e) => panic!("Failed to start server because {}", e),
607        };
608        let address_info = server.get_address_info().unwrap();
609        let (ip, port) = rumtk_get_ip_port!(address_info);
610        println!("Sleeping");
611        rumtk_sleep!(1);
612        let mut client = match rumtk_connect!(port) {
613            Ok(client) => client,
614            Err(e) => panic!("Failed to create server because {}", e),
615        };
616        rumtk_sleep!(2);
617        match client.send(&msg.to_raw()) {
618            Ok(_) => (),
619            Err(e) => panic!("Failed to send message because {}", e),
620        };
621        rumtk_sleep!(1);
622        let clients = server.get_client_ids();
623        let incoming_client_id = clients.first().expect("Expected client to have connected!");
624        let mut received_message = server.receive(incoming_client_id).unwrap();
625        if received_message.is_empty() {
626            rumtk_sleep!(1);
627            received_message = server.receive(incoming_client_id).unwrap();
628        }
629        assert_eq!(
630            &msg.to_raw(),
631            &received_message,
632            "{}",
633            format_compact!(
634                "Received message does not match sent message by client {:?}",
635                &received_message
636            )
637        );
638    }
639
640    ////////////////////////////JSON Tests/////////////////////////////////
641
642    #[test]
643    fn test_serialize_json() {
644        #[derive(Serialize)]
645        struct MyStruct {
646            hello: RUMString,
647        }
648
649        let hw = MyStruct {
650            hello: RUMString::from("World"),
651        };
652        let hw_str = rumtk_serialize!(&hw, true).unwrap();
653
654        assert!(
655            !hw_str.is_empty(),
656            "Empty JSON string generated from the test struct!"
657        );
658    }
659
660    #[test]
661    fn test_deserialize_json() {
662        #[derive(Serialize, Deserialize, PartialEq)]
663        struct MyStruct {
664            hello: RUMString,
665        }
666
667        let hw = MyStruct {
668            hello: RUMString::from("World"),
669        };
670        let hw_str = rumtk_serialize!(&hw, true).unwrap();
671        let new_hw: MyStruct = rumtk_deserialize!(&hw_str).unwrap();
672
673        assert!(
674            new_hw == hw,
675            "Deserialized JSON does not match the expected value!"
676        );
677    }
678
679    /*
680    #[test]
681    fn test_escape_unescape_json() {
682        #[derive(Serialize, Deserialize, PartialEq)]
683        struct MyStruct {
684            hello: RUMString,
685        }
686
687        let hw = MyStruct {
688            hello: RUMString::from("World"),
689        };
690
691        let hw_str = rumtk_serialize!(&hw, true).unwrap();
692        let hw_escaped_str = strings::basic_escape(&hw_str);
693        println!("Escaped => {}", hw_escaped_str);
694
695        let hw_unescaped_str = strings::unescape_string(&hw_escaped_str).unwrap();
696        println!("Unescaped => {}", hw_unescaped_str);
697        assert_eq!(
698            hw_str.to_rumstring(),
699            hw_unescaped_str.to_rumstring(),
700            "Unescaped serialized JSON mismatch!"
701        );
702
703        let new_hw: MyStruct = rumtk_deserialize!(&hw_unescaped_str).unwrap();
704
705        assert!(
706            new_hw == hw,
707            "Deserialized JSON does not match the expected value!"
708        );
709    }
710    */
711
712    ////////////////////////////CLI Tests/////////////////////////////////
713
714    #[test]
715    fn test_print_license_notice() {
716        print_license_notice("RUMTK", "2025", &vec!["Luis M. Santos, M.D."]);
717    }
718
719    //////////////////////////////////////////////////////////////////////////////////////////////
720}
rumtk_core/lib.rs

rumtk_core/
lib.rs
