Skip to content

Commit 0ecd250

Browse files
committed
Introduce code module
1 parent 57ada8a commit 0ecd250

File tree

3 files changed

+42
-41
lines changed

3 files changed

+42
-41
lines changed

src/cb1.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ const SEEDS: [[u32; 16]; 3] = [
9797
#[cfg(test)]
9898
mod tests {
9999
use super::*;
100-
use crate::{format_code, parse_code};
100+
use crate::code;
101101

102102
struct Test {
103103
decrypted: &'static str,
@@ -136,36 +136,36 @@ mod tests {
136136
#[test]
137137
fn test_encrypt_code() {
138138
for t in tests().iter() {
139-
let code = parse_code(t.decrypted);
139+
let code = code::parse(t.decrypted);
140140
let result = encrypt_code(code.0, code.1);
141-
assert_eq!(t.encrypted, format_code(result));
141+
assert_eq!(t.encrypted, code::format(result));
142142
}
143143
}
144144

145145
#[test]
146146
fn test_encrypt_code_mut() {
147147
for t in tests().iter() {
148-
let mut code = parse_code(t.decrypted);
148+
let mut code = code::parse(t.decrypted);
149149
encrypt_code_mut(&mut code.0, &mut code.1);
150-
assert_eq!(t.encrypted, format_code(code));
150+
assert_eq!(t.encrypted, code::format(code));
151151
}
152152
}
153153

154154
#[test]
155155
fn test_decrypt_code() {
156156
for t in tests().iter() {
157-
let code = parse_code(t.encrypted);
157+
let code = code::parse(t.encrypted);
158158
let result = decrypt_code(code.0, code.1);
159-
assert_eq!(t.decrypted, format_code(result));
159+
assert_eq!(t.decrypted, code::format(result));
160160
}
161161
}
162162

163163
#[test]
164164
fn test_decrypt_code_mut() {
165165
for t in tests().iter() {
166-
let mut code = parse_code(t.encrypted);
166+
let mut code = code::parse(t.encrypted);
167167
decrypt_code_mut(&mut code.0, &mut code.1);
168-
assert_eq!(t.decrypted, format_code(code));
168+
assert_eq!(t.decrypted, code::format(code));
169169
}
170170
}
171171
}

src/cb7.rs

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -477,7 +477,7 @@ const SEEDS: [[u8; 256]; 5] = [
477477
#[cfg(test)]
478478
mod tests {
479479
use super::*;
480-
use crate::{format_code, parse_code};
480+
use crate::code;
481481

482482
fn mul_tests() -> Vec<(u32, u32, u32)> {
483483
vec![
@@ -616,14 +616,14 @@ mod tests {
616616
#[test]
617617
fn test_encrypt_code() {
618618
for t in tests().iter() {
619-
let code = parse_code(t.beefcode);
619+
let code = code::parse(t.beefcode);
620620
let mut cb7 = Cb7::new();
621621
cb7.beefcode(code.0, code.1);
622622

623623
for (i, line) in t.decrypted.iter().enumerate() {
624-
let code = parse_code(line);
624+
let code = code::parse(line);
625625
let result = cb7.encrypt_code(code.0, code.1);
626-
assert_eq!(t.encrypted[i], format_code(result));
626+
assert_eq!(t.encrypted[i], code::format(result));
627627

628628
if is_beefcode(code.0) {
629629
cb7.beefcode(code.0, code.1)
@@ -635,15 +635,15 @@ mod tests {
635635
#[test]
636636
fn test_encrypt_code_mut() {
637637
for t in tests().iter() {
638-
let code = parse_code(t.beefcode);
638+
let code = code::parse(t.beefcode);
639639
let mut cb7 = Cb7::new();
640640
cb7.beefcode(code.0, code.1);
641641

642642
for (i, line) in t.decrypted.iter().enumerate() {
643-
let mut code = parse_code(line);
643+
let mut code = code::parse(line);
644644
let oldcode = code;
645645
cb7.encrypt_code_mut(&mut code.0, &mut code.1);
646-
assert_eq!(t.encrypted[i], format_code(code));
646+
assert_eq!(t.encrypted[i], code::format(code));
647647

648648
if is_beefcode(oldcode.0) {
649649
cb7.beefcode(oldcode.0, oldcode.1)
@@ -655,14 +655,14 @@ mod tests {
655655
#[test]
656656
fn test_decrypt_code() {
657657
for t in tests().iter() {
658-
let code = parse_code(t.beefcode);
658+
let code = code::parse(t.beefcode);
659659
let mut cb7 = Cb7::new();
660660
cb7.beefcode(code.0, code.1);
661661

662662
for (i, line) in t.encrypted.iter().enumerate() {
663-
let code = parse_code(line);
663+
let code = code::parse(line);
664664
let result = cb7.decrypt_code(code.0, code.1);
665-
assert_eq!(t.decrypted[i], format_code(result));
665+
assert_eq!(t.decrypted[i], code::format(result));
666666

667667
if is_beefcode(result.0) {
668668
cb7.beefcode(result.0, result.1)
@@ -674,14 +674,14 @@ mod tests {
674674
#[test]
675675
fn test_decrypt_code_mut() {
676676
for t in tests().iter() {
677-
let code = parse_code(t.beefcode);
677+
let code = code::parse(t.beefcode);
678678
let mut cb7 = Cb7::new();
679679
cb7.beefcode(code.0, code.1);
680680

681681
for (i, line) in t.encrypted.iter().enumerate() {
682-
let mut code = parse_code(line);
682+
let mut code = code::parse(line);
683683
cb7.decrypt_code_mut(&mut code.0, &mut code.1);
684-
assert_eq!(t.decrypted[i], format_code(code));
684+
assert_eq!(t.decrypted[i], code::format(code));
685685

686686
if is_beefcode(code.0) {
687687
cb7.beefcode(code.0, code.1)

src/lib.rs

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -342,9 +342,9 @@ mod tests {
342342
fn test_encrypt_code() {
343343
for t in tests().iter_mut() {
344344
for (i, line) in t.decrypted.iter().enumerate() {
345-
let code = parse_code(line);
345+
let code = code::parse(line);
346346
let result = t.cb.encrypt_code(code.0, code.1);
347-
assert_eq!(t.encrypted[i], format_code(result));
347+
assert_eq!(t.encrypted[i], code::format(result));
348348
}
349349
}
350350
}
@@ -353,9 +353,9 @@ mod tests {
353353
fn test_encrypt_code_mut() {
354354
for t in tests().iter_mut() {
355355
for (i, line) in t.decrypted.iter().enumerate() {
356-
let mut code = parse_code(line);
356+
let mut code = code::parse(line);
357357
t.cb.encrypt_code_mut(&mut code.0, &mut code.1);
358-
assert_eq!(t.encrypted[i], format_code(code));
358+
assert_eq!(t.encrypted[i], code::format(code));
359359
}
360360
}
361361
}
@@ -364,9 +364,9 @@ mod tests {
364364
fn test_decrypt_code() {
365365
for t in tests().iter_mut() {
366366
for (i, line) in t.encrypted.iter().enumerate() {
367-
let code = parse_code(line);
367+
let code = code::parse(line);
368368
let result = t.cb.decrypt_code(code.0, code.1);
369-
assert_eq!(t.decrypted[i], format_code(result));
369+
assert_eq!(t.decrypted[i], code::format(result));
370370
}
371371
}
372372
}
@@ -375,9 +375,9 @@ mod tests {
375375
fn test_decrypt_code_mut() {
376376
for t in tests().iter_mut() {
377377
for (i, line) in t.encrypted.iter().enumerate() {
378-
let mut code = parse_code(line);
378+
let mut code = code::parse(line);
379379
t.cb.decrypt_code_mut(&mut code.0, &mut code.1);
380-
assert_eq!(t.decrypted[i], format_code(code));
380+
assert_eq!(t.decrypted[i], code::format(code));
381381
}
382382
}
383383
}
@@ -469,9 +469,9 @@ mod tests {
469469
for t in auto_tests().iter_mut() {
470470
let mut cb = Codebreaker::new();
471471
for (i, line) in t.input.iter().enumerate() {
472-
let code = parse_code(line);
472+
let code = code::parse(line);
473473
let result = cb.auto_decrypt_code(code.0, code.1);
474-
assert_eq!(t.output[i], format_code(result));
474+
assert_eq!(t.output[i], code::format(result));
475475
}
476476
}
477477
}
@@ -481,21 +481,22 @@ mod tests {
481481
for t in auto_tests().iter_mut() {
482482
let mut cb = Codebreaker::new();
483483
for (i, line) in t.input.iter().enumerate() {
484-
let mut code = parse_code(line);
484+
let mut code = code::parse(line);
485485
cb.auto_decrypt_code_mut(&mut code.0, &mut code.1);
486-
assert_eq!(t.output[i], format_code(code));
486+
assert_eq!(t.output[i], code::format(code));
487487
}
488488
}
489489
}
490490
}
491491

492492
#[cfg(test)]
493-
fn parse_code(line: &str) -> (u32, u32) {
494-
let code: Vec<u32> = line.split(' ').map(|v| u32::from_str_radix(v, 16).unwrap()).collect();
495-
(code[0], code[1])
496-
}
493+
mod code {
494+
pub(crate) fn parse(line: &str) -> (u32, u32) {
495+
let code: Vec<u32> = line.split(' ').map(|v| u32::from_str_radix(v, 16).unwrap()).collect();
496+
(code[0], code[1])
497+
}
497498

498-
#[cfg(test)]
499-
fn format_code(code: (u32, u32)) -> String {
500-
format!("{:08X} {:08X}", code.0, code.1)
499+
pub(crate) fn format(code: (u32, u32)) -> String {
500+
format!("{:08X} {:08X}", code.0, code.1)
501+
}
501502
}

0 commit comments

Comments
 (0)