build.rs 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. use std::env;
  2. use std::fs::File;
  3. use std::io::Write;
  4. use std::path::Path;
  5. const TEST_PREFIX: &str = "
  6. #[cfg(test)]
  7. use pretty_assertions::assert_eq;
  8. use crate::grammar;
  9. use std::io::Write;
  10. // to let us use pretty_assertions with strings, we write a newtype
  11. // that delegates Debug to Display
  12. #[derive(PartialEq, Eq)]
  13. struct StringWrapper<'a> {
  14. wrapped: &'a str,
  15. }
  16. impl<'a> core::fmt::Debug for StringWrapper<'a> {
  17. fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
  18. core::fmt::Display::fmt(self.wrapped, f)
  19. }
  20. }
  21. fn assert_eq(x: &str, y: &str) {
  22. assert_eq!(StringWrapper {wrapped: x}, StringWrapper {wrapped: y});
  23. }
  24. ";
  25. const TEST_TEMPLATE: &str = "
  26. #[test]
  27. fn test_%PREFIX%() {
  28. let source = include_str!(\"%ROOT%/tests/%PREFIX%.matzo\");
  29. let ast = grammar::StmtsParser::new().parse(source);
  30. assert!(ast.is_ok());
  31. let ast = ast.unwrap();
  32. let mut buf = Vec::new();
  33. writeln!(buf, \"{:#?}\", ast).unwrap();
  34. assert_eq(
  35. std::str::from_utf8(&buf).unwrap().trim(),
  36. include_str!(\"%ROOT%/tests/%PREFIX%.parsed\").trim(),
  37. );
  38. }
  39. ";
  40. fn main() -> Result<(), Box<dyn std::error::Error>> {
  41. lalrpop::process_root()?;
  42. let out_dir = env::var("OUT_DIR")?;
  43. let manifest_dir = env::var("CARGO_MANIFEST_DIR")?;
  44. let dest = Path::new(&out_dir).join("exp_tests.rs");
  45. let mut test_file = File::create(&dest)?;
  46. writeln!(test_file, "{}", TEST_PREFIX)?;
  47. for exp in std::fs::read_dir("tests")? {
  48. let exp = exp?.path().canonicalize()?;
  49. let fname = exp.file_name().ok_or("bad file name")?.to_string_lossy();
  50. if let Some(prefix) = fname.strip_suffix(".matzo") {
  51. let test = TEST_TEMPLATE
  52. .replace("%FILE%", &fname)
  53. .replace("%PREFIX%", prefix)
  54. .replace("%ROOT%", &manifest_dir);
  55. writeln!(test_file, "{}", test)?;
  56. }
  57. }
  58. Ok(())
  59. }