use rust_codegen::*;
#[test]
fn block_one_line() {
let mut scope = Scope::new();
let new_fn = scope.new_fn("hello_world");
let mut block = Block::new("");
block.line("println!(\"Hello, world!\");");
new_fn.push_block(block);
let expected = r#"
fn hello_world() {
{
println!("Hello, world!");
}
}"#;
assert_eq!(scope.to_string(), &expected[1..]);
}
#[test]
fn block_multiple_lines() {
let mut scope = Scope::new();
let new_fn = scope.new_fn("hello_world");
let mut block = Block::new("");
block.line("println!(\"Hello, world!\");");
block.line("println!(\"from Rust!\");");
new_fn.push_block(block);
let expected = r#"
fn hello_world() {
{
println!("Hello, world!");
println!("from Rust!");
}
}"#;
assert_eq!(scope.to_string(), &expected[1..]);
}