diff options
Diffstat (limited to 'tests/ui-fulldeps/rustc_public/smir_serde.rs')
| -rw-r--r-- | tests/ui-fulldeps/rustc_public/smir_serde.rs | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/tests/ui-fulldeps/rustc_public/smir_serde.rs b/tests/ui-fulldeps/rustc_public/smir_serde.rs new file mode 100644 index 00000000000..972bc5efe20 --- /dev/null +++ b/tests/ui-fulldeps/rustc_public/smir_serde.rs @@ -0,0 +1,68 @@ +//@ run-pass +//! Test that users are able to use serialize stable MIR constructs. + +//@ ignore-stage1 +//@ ignore-cross-compile +//@ ignore-remote +//@ edition: 2021 + +#![feature(rustc_private)] +#![feature(assert_matches)] + + +extern crate rustc_driver; +extern crate rustc_interface; +extern crate rustc_middle; +extern crate serde; +extern crate serde_json; +#[macro_use] +extern crate rustc_public; + +use rustc_middle::ty::TyCtxt; +use serde_json::to_string; +use rustc_public::mir::Body; +use std::io::{BufWriter, Write}; +use std::ops::ControlFlow; + +const CRATE_NAME: &str = "input"; + +fn serialize_to_json(_tcx: TyCtxt<'_>) -> ControlFlow<()> { + let path = "output.json"; + let mut writer = BufWriter::new(std::fs::File::create(path).expect("Failed to create path")); + let local_crate = rustc_public::local_crate(); + let items: Vec<Body> = + rustc_public::all_local_items().iter().map(|item| item.expect_body()).collect(); + let crate_data = (local_crate.name, items); + writer + .write_all(to_string(&crate_data).expect("serde_json failed").as_bytes()) + .expect("JSON serialization failed"); + ControlFlow::Continue(()) +} + +/// This test will generate and analyze a dummy crate using the stable mir. +/// For that, it will first write the dummy crate into a file. +/// Then it will create a `RustcPublic` using custom arguments and then +/// it will run the compiler. +fn main() { + let path = "internal_input.rs"; + generate_input(&path).unwrap(); + let args = &[ + "rustc".to_string(), + "--crate-name".to_string(), + CRATE_NAME.to_string(), + path.to_string(), + ]; + run_with_tcx!(args, serialize_to_json).unwrap(); +} + +fn generate_input(path: &str) -> std::io::Result<()> { + let mut file = std::fs::File::create(path)?; + write!( + file, + r#" + pub fn main() {{ + }} + "# + )?; + Ok(()) +} |
