-
Notifications
You must be signed in to change notification settings - Fork 409
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test(binaryinstall): Add initial tests
- Loading branch information
Showing
5 changed files
with
203 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -18,3 +18,6 @@ is_executable = "0.1.2" | |
siphasher = "0.2.3" | ||
tar = "0.4.16" | ||
zip = "0.5.0" | ||
|
||
[dev-dependencies] | ||
tempfile = "3.0.5" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,68 @@ | ||
use binary_install::Cache; | ||
use utils; | ||
|
||
#[test] | ||
fn it_returns_none_if_install_is_not_permitted() { | ||
let binary_name = "wasm-pack"; | ||
let binaries = vec![binary_name]; | ||
|
||
let dir = tempfile::TempDir::new().unwrap(); | ||
let cache = Cache::at(dir.path()); | ||
|
||
let dl = cache.download( | ||
false, | ||
binary_name, | ||
&binaries, | ||
&format!("{}/{}.tar.gz", "", binary_name), | ||
); | ||
|
||
assert!(dl.is_ok()); | ||
assert!(dl.unwrap().is_none()) | ||
} | ||
|
||
#[test] | ||
#[ignore] | ||
fn it_returns_destination_if_binary_already_exists() { | ||
let binary_name = "wasm-pack"; | ||
let binaries = vec![binary_name]; | ||
|
||
let dir = tempfile::TempDir::new().unwrap(); | ||
let cache = Cache::at(dir.path()); | ||
|
||
let dl = cache.download( | ||
true, | ||
binary_name, | ||
&binaries, | ||
&format!("{}/{}.tar.gz", "http://localhost:7878", binary_name), | ||
); | ||
|
||
assert!(dl.is_ok()); | ||
assert!(dl.unwrap().is_some()) | ||
} | ||
|
||
#[test] | ||
fn it_downloads_tarball() { | ||
let server_port = 7880; | ||
let url = format!("http://{}:{}", utils::TEST_SERVER_HOST, server_port); | ||
let binary_name = "wasm-pack"; | ||
let binaries = vec![binary_name]; | ||
|
||
// Create a temporary tarball. | ||
let tarball = utils::create_tarball(binary_name).ok(); | ||
|
||
// Spin up a local TcpListener. | ||
utils::start_server(server_port, tarball); | ||
|
||
let dir = tempfile::TempDir::new().unwrap(); | ||
let cache = Cache::at(dir.path()); | ||
|
||
let dl = cache.download( | ||
true, | ||
binary_name, | ||
&binaries, | ||
&format!("{}/{}.tar.gz", &url, binary_name), | ||
); | ||
|
||
assert!(dl.is_ok()); | ||
assert!(dl.unwrap().is_some()) | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,6 @@ | ||
extern crate binary_install; | ||
extern crate flate2; | ||
extern crate tar; | ||
|
||
pub mod utils; | ||
pub mod cache; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,67 @@ | ||
use std::fs::{File, OpenOptions}; | ||
use std::io::{self, Read, Write}; | ||
use std::net::TcpListener; | ||
use std::thread; | ||
use flate2::write::GzEncoder; | ||
use flate2::Compression; | ||
|
||
pub const TEST_SERVER_HOST: &'static str = "localhost"; | ||
|
||
pub fn start_server(port: u32, tarball: Option<Vec<u8>>) -> thread::JoinHandle<TcpListener> { | ||
thread::spawn(move || { | ||
let listener = TcpListener::bind(format!("{}:{}", TEST_SERVER_HOST, port)).unwrap(); | ||
|
||
for stream in listener.incoming() { | ||
let mut stream = stream.unwrap(); | ||
|
||
let mut buffer = [0; 512]; | ||
|
||
stream.read(&mut buffer).unwrap(); | ||
|
||
let response = "HTTP/1.1 200 OK\r\n\r\n"; | ||
|
||
stream.write(response.as_bytes()).unwrap(); | ||
|
||
match tarball.to_owned() { | ||
Some(tar) => { | ||
stream.write(tar.as_ref()).unwrap(); | ||
} | ||
None => {} | ||
} | ||
|
||
stream.flush().unwrap(); | ||
} | ||
listener | ||
}) | ||
} | ||
|
||
pub fn create_tarball(binary_name: &str) -> Result<Vec<u8>, io::Error> { | ||
let temp_dir = tempfile::TempDir::new().unwrap(); | ||
let full_path = temp_dir.path().join(binary_name.to_owned() + ".tar.gz"); | ||
|
||
let tar = OpenOptions::new() | ||
.create(true) | ||
.read(true) | ||
.write(true) | ||
.open(&full_path)?; | ||
|
||
let mut file = OpenOptions::new() | ||
.create(true) | ||
.read(true) | ||
.write(true) | ||
.open(temp_dir.path().join(binary_name))?; | ||
|
||
let mut encoder = GzEncoder::new(tar, Compression::default()); | ||
{ | ||
let mut archive = tar::Builder::new(&mut encoder); | ||
archive.append_file(binary_name, &mut file)?; | ||
} | ||
|
||
let mut contents = vec![]; | ||
|
||
encoder.finish()?; | ||
|
||
File::open(temp_dir.path().join(&full_path))?.read_to_end(&mut contents)?; | ||
|
||
Ok(contents) | ||
} |