1 //
2 // Copyright 2019, Google Inc.
3 // All rights reserved.
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 // * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 // * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 // * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 //
31 // Alternatively, this software may be distributed under the terms of the
32 // GNU General Public License ("GPL") version 2 as published by the Free
33 // Software Foundation.
34 //
35
36 use std::fs::File;
37 use std::io::prelude::*;
38 use std::io::BufWriter;
39 use std::path::Path;
40
41 use rand::prelude::*;
42
gen_rand_testdata(path: &Path, size: usize) -> std::io::Result<()>43 pub fn gen_rand_testdata(path: &Path, size: usize) -> std::io::Result<()> {
44 let mut buf = BufWriter::new(File::create(path)?);
45
46 let mut a: Vec<u8> = vec![0; size];
47 thread_rng().fill(a.as_mut_slice());
48
49 buf.write_all(a.as_slice())?;
50
51 Ok(())
52 }
53
54 #[cfg(test)]
55 mod tests {
56 use super::*;
57
58 #[test]
gen_rand_testdata()59 fn gen_rand_testdata() {
60 use super::gen_rand_testdata;
61
62 let path0 = Path::new("/tmp/idk_test00");
63 let path1 = Path::new("/tmp/idk_test01");
64 let sz = 1024;
65
66 gen_rand_testdata(path0, sz).unwrap();
67 gen_rand_testdata(path1, sz).unwrap();
68
69 let mut buf0 = Vec::new();
70 let mut buf1 = Vec::new();
71
72 let mut f = File::open(path0).unwrap();
73 let mut g = File::open(path1).unwrap();
74
75 f.read_to_end(&mut buf0).unwrap();
76 g.read_to_end(&mut buf1).unwrap();
77
78 assert_ne!(buf0, buf1);
79 }
80 }
81