;\n"), ("a.py", "def f():\n return 'x'\n"), ("a.go", "package main\nfunc main() {}\n"), ("a.c", "int main(void) { return 0; }\n"), ("a.cpp", "class A {};\n"), ("A.java", "class A { void f() {} }\n"), ("Gemfile", "gem 'x'\n"), ("a.json", "{\"key\": 1}\n"), ("Cargo.lock", "[[package]]\nname = \"x\"\n"), ("a.yml", "a: 'b'\n"), ("a.css", "a { color: red; }\n"), ("a.lua", "local x = 'y'\n"), ("a.zig", "const x = 1;\n"), ("run", "#!/usr/bin/env bash\necho 'hi'\n"), ("tool", "#!/usr/bin/python3.12\nx = 'y'\n"), ("cli", "#!/usr/bin/env node\nlet a = 'b'\n"), ("task", "#!/usr/bin/ruby\nputs 'x'\n"), ("main.tf", "resource \"a\" \"b\" {\n n = 1 # c\n}\n"), ("README.md", "# Title\n\n- `code`\n"), ("Dockerfile", "FROM alpine AS base\nRUN echo 'hi'\n"), ("icon.svg", "\n"), ("fix.patch", "--- a/x\n+++ b/x\n@@ -1 +1 @@\n-a\n+b\n"), ("flake.nix", "{ x = \"y\"; }\n"), ("a.php", "

\n"), ("A.cs", "class A { int f() { return 1; } }\n"), ("a.swift", "let x = \"y\"\n"), ]; for (path, text) in samples { let r = code(path, text).expect(path); assert!(!r.encode().is_empty(), "{path}"); } let json = "{\"key\": 1}\n"; assert_eq!( spans(&code("a.json", json).unwrap(), json, "attr"), ["\"key\""] ); } #[test] fn html_carries_its_scripts_and_styles() { let html = "y\n"; let r = code("index.html", html).unwrap(); assert_eq!(spans(&r, html, "attr"), ["href"]); assert!( spans(&r, html, "string").contains(&"'q'".to_string()), "{:?}", r ); assert!(spans(&r, html, "tag").contains(&"script".to_string())); } #[test] fn svelte_carries_its_scripts_styles_and_expressions() { let svelte = "\n\ {#if a > 0}{a + 2}{/if}\n\ \n"; let r = code("Avatar.svelte", svelte).unwrap(); assert_eq!(spans(&r, svelte, "attr"), ["lang", "class", "title"]); assert!(spans(&r, svelte, "type").contains(&"number".to_string())); let numbers = spans(&r, svelte, "number"); assert!( ["1", "0", "2"] .iter() .all(|n| numbers.contains(&n.to_string())) ); assert!(spans(&r, svelte, "string").contains(&"'t'".to_string())); assert!(spans(&r, svelte, "keyword").contains(&"#if".to_string())); assert!(spans(&r, svelte, "tag").contains(&"style".to_string())); } #[test] fn markdown_fences_take_their_language() { let md = "# Hi\n\n```rs\nfn f() {}\n```\n\nSee [it](x.md).\n"; let r = code("README.md", md).unwrap(); assert_eq!(spans(&r, md, "function"), ["# Hi\n", "f"]); assert_eq!(spans(&r, md, "attr"), ["rs", "it"]); assert_eq!(spans(&r, md, "string"), ["x.md"]); } #[test] fn what_it_leaves_plain() { assert_eq!(code("notes.unknown-kind", "plain\n"), None); assert_eq!(code("script", "#!/bin/perl\n"), None); assert_eq!(code("big.rs", &"x".repeat(LIMIT + 1)), None); let mut empty = Ranges::new(CODE); empty.push(0, 4, 4); assert_eq!(empty.encode(), ""); } fn row(kind: Kind, old: Option, new: Option, text: &str) -> Row { Row { kind, old, new, text: text.into(), } } #[test] fn diff_rows() { let rows = [ row(Kind::Hunk, None, None, "@@ -1,3 +1,3 @@"), row(Kind::Context, None, None, "/* a"), row(Kind::Del, None, None, "b */ let x = 1;"), row(Kind::Add, None, None, "b */ let x = 2;"), row(Kind::Add, None, None, "let é = \"q\";"), row(Kind::Note, None, None, "\\ No newline at end of file"), row(Kind::Del, None, None, "gone"), row(Kind::Add, None, None, "here"), ]; let r = super::rows("a.rs", &rows, Sides::default()); let shown = "/* a\nb */ let x = 1;\nb */ let x = 2;\nlet é = \"q\";\ngone\nhere\n"; // The comment spans two rows, and is cut at the row between. assert_eq!(spans(&r, shown, "comment"), ["/* a", "b */", "b */"]); assert_eq!(spans(&r, shown, "number"), ["1", "2"]); assert_eq!(spans(&r, shown, "string"), ["\"q\""]); assert_eq!(spans(&r, shown, "xdel"), ["1"]); assert_eq!(spans(&r, shown, "xins"), ["2"]); assert_eq!( super::rows("notes.unknown-kind", &rows, Sides::default()).encode(), "xins:34,1;xdel:18,1" ); assert_eq!(changed("abc", "xyz"), None); assert_eq!(changed("gone", "here"), None); assert_eq!(changed("aé", "aéb"), Some(((2, 2), (2, 3)))); assert_eq!(changed("aa", "a"), Some(((1, 2), (1, 1)))); } #[test] fn diff_rows_in_their_files() { // The first hunk opens a string the second never sees close: read // alone, the string is taken for code. let old = "const Q: &str = \"a\nb\nc\nd\ne\";\n/// doc\nfn f() {}\n"; let new = "const Q: &str = \"a\r\nB\r\nc\r\nd\r\ne\";\r\n/// doc\r\nfn g() {}\r\n"; let rows = [ row(Kind::Hunk, None, None, "@@ -1,2 +1,2 @@"), row(Kind::Context, Some(1), Some(1), "const Q: &str = \"a"), row(Kind::Del, Some(2), None, "b"), row(Kind::Add, None, Some(2), "B"), row(Kind::Hunk, None, None, "@@ -6,2 +6,2 @@"), row(Kind::Context, Some(6), Some(6), "/// doc"), row(Kind::Del, Some(7), None, "fn f() {}"), row(Kind::Add, None, Some(7), "fn g() {}"), ]; let shown = "const Q: &str = \"a\nb\nB\n/// doc\nfn f() {}\nfn g() {}\n"; let both = Sides { old: Some(old), new: Some(new), }; let r = super::rows("a.rs", &rows, both); assert_eq!(spans(&r, shown, "string"), ["\"a", "b", "B"]); assert_eq!(spans(&r, shown, "comment"), ["/// doc"]); assert_eq!(spans(&r, shown, "function"), ["f", "g"]); let alone = super::rows("a.rs", &rows, Sides::default()); assert!(spans(&alone, shown, "string").is_empty(), "{alone:?}"); // A side whose lines are not the diff's is no help: the rows are // read alone, as they are without it. let wrong = Sides { old: Some("nothing like it\n"), new: Some(""), }; assert_eq!(super::rows("a.rs", &rows, wrong), alone); // Nor is one in a language without a grammar here. assert_eq!( super::rows("a.unknown-kind", &rows, both), super::rows("a.unknown-kind", &rows, Sides::default()) ); // A side with no rows of its own still lines up. let added = [row(Kind::Add, None, Some(2), "B")]; let r = super::rows("a.rs", &added, both); assert_eq!(spans(&r, "B\n", "string"), ["B"]); } }