crates/ide/src/syntax_highlighting/tests.rs

   1 use std::time::Instant;
   2
   3 use expect_test::{expect_file, ExpectFile};
   4 use ide_db::SymbolKind;
   5 use test_utils::{bench, bench_fixture, skip_slow_tests, AssertLinear};
   6
   7 use crate::{fixture, FileRange, HlTag, TextRange};
   8
   9 #[test]
  10 fn test_highlighting() {
  11     check_highlighting(
  12         r#"
  13 //- proc_macros: identity, mirror
  14 //- /main.rs crate:main deps:foo
  15 use inner::{self as inner_mod};
  16 mod inner {}
  17
  18 #[rustc_builtin_macro]
  19 macro Copy {}
  20
  21 // Needed for function consuming vs normal
  22 pub mod marker {
  23     #[lang = "copy"]
  24     pub trait Copy {}
  25 }
  26
  27 #[proc_macros::identity]
  28 pub mod ops {
  29     #[lang = "fn_once"]
  30     pub trait FnOnce<Args> {}
  31
  32     #[lang = "fn_mut"]
  33     pub trait FnMut<Args>: FnOnce<Args> {}
  34
  35     #[lang = "fn"]
  36     pub trait Fn<Args>: FnMut<Args> {}
  37 }
  38
  39 proc_macros::mirror! {
  40     {
  41         ,i32 :x pub
  42         ,i32 :y pub
  43     } Foo struct
  44 }
  45
  46 trait Bar where Self: {
  47     fn bar(&self) -> i32;
  48 }
  49
  50 impl Bar for Foo where Self: {
  51     fn bar(&self) -> i32 {
  52         self.x
  53     }
  54 }
  55
  56 impl Foo {
  57     fn baz(mut self, f: Foo) -> i32 {
  58         f.baz(self)
  59     }
  60
  61     fn qux(&mut self) {
  62         self.x = 0;
  63     }
  64
  65     fn quop(&self) -> i32 {
  66         self.x
  67     }
  68 }
  69
  70 #[derive(Copy)]
  71 struct FooCopy {
  72     x: u32,
  73 }
  74
  75 impl FooCopy {
  76     fn baz(self, f: FooCopy) -> u32 {
  77         f.baz(self)
  78     }
  79
  80     fn qux(&mut self) {
  81         self.x = 0;
  82     }
  83
  84     fn quop(&self) -> u32 {
  85         self.x
  86     }
  87 }
  88
  89 fn str() {
  90     str();
  91 }
  92
  93 fn foo<'a, T>() -> T {
  94     foo::<'a, i32>()
  95 }
  96
  97 fn never() -> ! {
  98     loop {}
  99 }
 100
 101 fn const_param<const FOO: usize>() -> usize {
 102     FOO
 103 }
 104
 105 use ops::Fn;
 106 fn baz<F: Fn() -> ()>(f: F) {
 107     f()
 108 }
 109
 110 fn foobar() -> impl Copy {}
 111
 112 fn foo() {
 113     let bar = foobar();
 114 }
 115
 116 macro_rules! def_fn {
 117     ($($tt:tt)*) => {$($tt)*}
 118 }
 119
 120 def_fn! {
 121     fn bar() -> u32 {
 122         100
 123     }
 124 }
 125
 126 macro_rules! dont_color_me_braces {
 127     () => {0}
 128 }
 129
 130 macro_rules! noop {
 131     ($expr:expr) => {
 132         $expr
 133     }
 134 }
 135
 136 macro_rules! keyword_frag {
 137     ($type:ty) => ($type)
 138 }
 139
 140 macro with_args($i:ident) {
 141     $i
 142 }
 143
 144 macro without_args {
 145     ($i:ident) => {
 146         $i
 147     }
 148 }
 149
 150 // comment
 151 fn main() {
 152     println!("Hello, {}!", 92);
 153     dont_color_me_braces!();
 154
 155     let mut vec = Vec::new();
 156     if true {
 157         let x = 92;
 158         vec.push(Foo { x, y: 1 });
 159     }
 160
 161     for e in vec {
 162         // Do nothing
 163     }
 164
 165     noop!(noop!(1));
 166
 167     let mut x = 42;
 168     x += 1;
 169     let y = &mut x;
 170     let z = &y;
 171
 172     let Foo { x: z, y } = Foo { x: z, y };
 173
 174     y;
 175
 176     let mut foo = Foo { x, y: x };
 177     let foo2 = Foo { x, y: x };
 178     foo.quop();
 179     foo.qux();
 180     foo.baz(foo2);
 181
 182     let mut copy = FooCopy { x };
 183     copy.quop();
 184     copy.qux();
 185     copy.baz(copy);
 186
 187     let a = |x| x;
 188     let bar = Foo::baz;
 189
 190     let baz = (-42,);
 191     let baz = -baz.0;
 192
 193     let _ = !true;
 194
 195     'foo: loop {
 196         break 'foo;
 197         continue 'foo;
 198     }
 199 }
 200
 201 enum Option<T> {
 202     Some(T),
 203     None,
 204 }
 205 use Option::*;
 206
 207 impl<T> Option<T> {
 208     fn and<U>(self, other: Option<U>) -> Option<(T, U)> {
 209         match other {
 210             None => unimplemented!(),
 211             Nope => Nope,
 212         }
 213     }
 214 }
 215
 216 async fn learn_and_sing() {
 217     let song = learn_song().await;
 218     sing_song(song).await;
 219 }
 220
 221 async fn async_main() {
 222     let f1 = learn_and_sing();
 223     let f2 = dance();
 224     futures::join!(f1, f2);
 225 }
 226
 227 fn use_foo_items() {
 228     let bob = foo::Person {
 229         name: "Bob",
 230         age: foo::consts::NUMBER,
 231     };
 232
 233     let control_flow = foo::identity(foo::ControlFlow::Continue);
 234
 235     if control_flow.should_die() {
 236         foo::die!();
 237     }
 238 }
 239
 240 pub enum Bool { True, False }
 241
 242 impl Bool {
 243     pub const fn to_primitive(self) -> bool {
 244         matches!(self, Self::True)
 245     }
 246 }
 247 const USAGE_OF_BOOL:bool = Bool::True.to_primitive();
 248
 249 //- /foo.rs crate:foo
 250 pub struct Person {
 251     pub name: &'static str,
 252     pub age: u8,
 253 }
 254
 255 pub enum ControlFlow {
 256     Continue,
 257     Die,
 258 }
 259
 260 impl ControlFlow {
 261     pub fn should_die(self) -> bool {
 262         matches!(self, ControlFlow::Die)
 263     }
 264 }
 265
 266 pub fn identity<T>(x: T) -> T { x }
 267
 268 pub mod consts {
 269     pub const NUMBER: i64 = 92;
 270 }
 271
 272 macro_rules! die {
 273     () => {
 274         panic!();
 275     };
 276 }
 277 "#
 278         .trim(),
 279         expect_file!["./test_data/highlighting.html"],
 280         false,
 281     );
 282 }
 283
 284 #[test]
 285 fn test_rainbow_highlighting() {
 286     check_highlighting(
 287         r#"
 288 fn main() {
 289     let hello = "hello";
 290     let x = hello.to_string();
 291     let y = hello.to_string();
 292
 293     let x = "other color please!";
 294     let y = x.to_string();
 295 }
 296
 297 fn bar() {
 298     let mut hello = "hello";
 299 }
 300 "#
 301         .trim(),
 302         expect_file!["./test_data/rainbow_highlighting.html"],
 303         true,
 304     );
 305 }
 306
 307 #[test]
 308 fn benchmark_syntax_highlighting_long_struct() {
 309     if skip_slow_tests() {
 310         return;
 311     }
 312
 313     let fixture = bench_fixture::big_struct();
 314     let (analysis, file_id) = fixture::file(&fixture);
 315
 316     let hash = {
 317         let _pt = bench("syntax highlighting long struct");
 318         analysis
 319             .highlight(file_id)
 320             .unwrap()
 321             .iter()
 322             .filter(|it| it.highlight.tag == HlTag::Symbol(SymbolKind::Struct))
 323             .count()
 324     };
 325     assert_eq!(hash, 2001);
 326 }
 327
 328 #[test]
 329 fn syntax_highlighting_not_quadratic() {
 330     if skip_slow_tests() {
 331         return;
 332     }
 333
 334     let mut al = AssertLinear::default();
 335     while al.next_round() {
 336         for i in 6..=10 {
 337             let n = 1 << i;
 338
 339             let fixture = bench_fixture::big_struct_n(n);
 340             let (analysis, file_id) = fixture::file(&fixture);
 341
 342             let time = Instant::now();
 343
 344             let hash = analysis
 345                 .highlight(file_id)
 346                 .unwrap()
 347                 .iter()
 348                 .filter(|it| it.highlight.tag == HlTag::Symbol(SymbolKind::Struct))
 349                 .count();
 350             assert!(hash > n as usize);
 351
 352             let elapsed = time.elapsed();
 353             al.sample(n as f64, elapsed.as_millis() as f64);
 354         }
 355     }
 356 }
 357
 358 #[test]
 359 fn benchmark_syntax_highlighting_parser() {
 360     if skip_slow_tests() {
 361         return;
 362     }
 363
 364     let fixture = bench_fixture::glorious_old_parser();
 365     let (analysis, file_id) = fixture::file(&fixture);
 366
 367     let hash = {
 368         let _pt = bench("syntax highlighting parser");
 369         analysis
 370             .highlight(file_id)
 371             .unwrap()
 372             .iter()
 373             .filter(|it| it.highlight.tag == HlTag::Symbol(SymbolKind::Function))
 374             .count()
 375     };
 376     assert_eq!(hash, 1616);
 377 }
 378
 379 #[test]
 380 fn test_ranges() {
 381     let (analysis, file_id) = fixture::file(
 382         r#"
 383 #[derive(Clone, Debug)]
 384 struct Foo {
 385     pub x: i32,
 386     pub y: i32,
 387 }
 388 "#,
 389     );
 390
 391     // The "x"
 392     let highlights = &analysis
 393         .highlight_range(FileRange { file_id, range: TextRange::at(45.into(), 1.into()) })
 394         .unwrap();
 395
 396     assert_eq!(&highlights[0].highlight.to_string(), "field.declaration.public");
 397 }
 398
 399 #[test]
 400 fn test_flattening() {
 401     check_highlighting(
 402         r##"
 403 fn fixture(ra_fixture: &str) {}
 404
 405 fn main() {
 406     fixture(r#"
 407         trait Foo {
 408             fn foo() {
 409                 println!("2 + 2 = {}", 4);
 410             }
 411         }"#
 412     );
 413 }"##
 414         .trim(),
 415         expect_file!["./test_data/highlight_injection.html"],
 416         false,
 417     );
 418 }
 419
 420 #[test]
 421 fn ranges_sorted() {
 422     let (analysis, file_id) = fixture::file(
 423         r#"
 424 #[foo(bar = "bar")]
 425 macro_rules! test {}
 426 }"#
 427         .trim(),
 428     );
 429     let _ = analysis.highlight(file_id).unwrap();
 430 }
 431
 432 #[test]
 433 fn test_string_highlighting() {
 434     // The format string detection is based on macro-expansion,
 435     // thus, we have to copy the macro definition from `std`
 436     check_highlighting(
 437         r#"
 438 macro_rules! println {
 439     ($($arg:tt)*) => ({
 440         $crate::io::_print($crate::format_args_nl!($($arg)*));
 441     })
 442 }
 443 #[rustc_builtin_macro]
 444 #[macro_export]
 445 macro_rules! format_args {}
 446 #[rustc_builtin_macro]
 447 #[macro_export]
 448 macro_rules! const_format_args {}
 449 #[rustc_builtin_macro]
 450 #[macro_export]
 451 macro_rules! format_args_nl {}
 452
 453 mod panic {
 454     pub macro panic_2015 {
 455         () => (
 456             $crate::panicking::panic("explicit panic")
 457         ),
 458         ($msg:literal $(,)?) => (
 459             $crate::panicking::panic($msg)
 460         ),
 461         // Use `panic_str` instead of `panic_display::<&str>` for non_fmt_panic lint.
 462         ($msg:expr $(,)?) => (
 463             $crate::panicking::panic_str($msg)
 464         ),
 465         // Special-case the single-argument case for const_panic.
 466         ("{}", $arg:expr $(,)?) => (
 467             $crate::panicking::panic_display(&$arg)
 468         ),
 469         ($fmt:expr, $($arg:tt)+) => (
 470             $crate::panicking::panic_fmt($crate::const_format_args!($fmt, $($arg)+))
 471         ),
 472     }
 473 }
 474
 475 #[rustc_builtin_macro(std_panic)]
 476 #[macro_export]
 477 macro_rules! panic {}
 478 #[rustc_builtin_macro]
 479 macro_rules! assert {}
 480
 481 macro_rules! todo {
 482     () => ($crate::panic!("not yet implemented"));
 483     ($($arg:tt)+) => ($crate::panic!("not yet implemented: {}", $crate::format_args!($($arg)+)));
 484 }
 485
 486 fn main() {
 487     // from https://doc.rust-lang.org/std/fmt/index.html
 488     println!("Hello");                 // => "Hello"
 489     println!("Hello, {}!", "world");   // => "Hello, world!"
 490     println!("The number is {}", 1);   // => "The number is 1"
 491     println!("{:?}", (3, 4));          // => "(3, 4)"
 492     println!("{value}", value=4);      // => "4"
 493     println!("{} {}", 1, 2);           // => "1 2"
 494     println!("{:04}", 42);             // => "0042" with leading zerosV
 495     println!("{1} {} {0} {}", 1, 2);   // => "2 1 1 2"
 496     println!("{argument}", argument = "test");   // => "test"
 497     println!("{name} {}", 1, name = 2);          // => "2 1"
 498     println!("{a} {c} {b}", a="a", b='b', c=3);  // => "a 3 b"
 499     println!("{{{}}}", 2);                       // => "{2}"
 500     println!("Hello {:5}!", "x");
 501     println!("Hello {:1$}!", "x", 5);
 502     println!("Hello {1:0$}!", 5, "x");
 503     println!("Hello {:width$}!", "x", width = 5);
 504     println!("Hello {:<5}!", "x");
 505     println!("Hello {:-<5}!", "x");
 506     println!("Hello {:^5}!", "x");
 507     println!("Hello {:>5}!", "x");
 508     println!("Hello {:+}!", 5);
 509     println!("{:#x}!", 27);
 510     println!("Hello {:05}!", 5);
 511     println!("Hello {:05}!", -5);
 512     println!("{:#010x}!", 27);
 513     println!("Hello {0} is {1:.5}", "x", 0.01);
 514     println!("Hello {1} is {2:.0$}", 5, "x", 0.01);
 515     println!("Hello {0} is {2:.1$}", "x", 5, 0.01);
 516     println!("Hello {} is {:.*}",    "x", 5, 0.01);
 517     println!("Hello {} is {2:.*}",   "x", 5, 0.01);
 518     println!("Hello {} is {number:.prec$}", "x", prec = 5, number = 0.01);
 519     println!("{}, `{name:.*}` has 3 fractional digits", "Hello", 3, name=1234.56);
 520     println!("{}, `{name:.*}` has 3 characters", "Hello", 3, name="1234.56");
 521     println!("{}, `{name:>8.*}` has 3 right-aligned characters", "Hello", 3, name="1234.56");
 522     println!("Hello {{}}");
 523     println!("{{ Hello");
 524
 525     println!(r"Hello, {}!", "world");
 526
 527     // escape sequences
 528     println!("Hello\nWorld");
 529     println!("\u{48}\x65\x6C\x6C\x6F World");
 530
 531     println!("{\x41}", A = 92);
 532     println!("{ничоси}", ничоси = 92);
 533
 534     println!("{:x?} {} ", thingy, n2);
 535     panic!("{}", 0);
 536     panic!("more {}", 1);
 537     assert!(true, "{}", 1);
 538     assert!(true, "{} asdasd", 1);
 539     todo!("{}fmt", 0);
 540 }"#
 541         .trim(),
 542         expect_file!["./test_data/highlight_strings.html"],
 543         false,
 544     );
 545 }
 546
 547 #[test]
 548 fn test_unsafe_highlighting() {
 549     check_highlighting(
 550         r#"
 551 static mut MUT_GLOBAL: Struct = Struct { field: 0 };
 552 static GLOBAL: Struct = Struct { field: 0 };
 553 unsafe fn unsafe_fn() {}
 554
 555 union Union {
 556     a: u32,
 557     b: f32,
 558 }
 559
 560 struct Struct { field: i32 }
 561 impl Struct {
 562     unsafe fn unsafe_method(&self) {}
 563 }
 564
 565 #[repr(packed)]
 566 struct Packed {
 567     a: u16,
 568 }
 569
 570 unsafe trait UnsafeTrait {}
 571 unsafe impl UnsafeTrait for Packed {}
 572 impl !UnsafeTrait for () {}
 573
 574 fn unsafe_trait_bound<T: UnsafeTrait>(_: T) {}
 575
 576 trait DoTheAutoref {
 577     fn calls_autoref(&self);
 578 }
 579
 580 impl DoTheAutoref for u16 {
 581     fn calls_autoref(&self) {}
 582 }
 583
 584 fn main() {
 585     let x = &5 as *const _ as *const usize;
 586     let u = Union { b: 0 };
 587     unsafe {
 588         // unsafe fn and method calls
 589         unsafe_fn();
 590         let b = u.b;
 591         match u {
 592             Union { b: 0 } => (),
 593             Union { a } => (),
 594         }
 595         Struct { field: 0 }.unsafe_method();
 596
 597         // unsafe deref
 598         *x;
 599
 600         // unsafe access to a static mut
 601         MUT_GLOBAL.field;
 602         GLOBAL.field;
 603
 604         // unsafe ref of packed fields
 605         let packed = Packed { a: 0 };
 606         let a = &packed.a;
 607         let ref a = packed.a;
 608         let Packed { ref a } = packed;
 609         let Packed { a: ref _a } = packed;
 610
 611         // unsafe auto ref of packed field
 612         packed.a.calls_autoref();
 613     }
 614 }
 615 "#
 616         .trim(),
 617         expect_file!["./test_data/highlight_unsafe.html"],
 618         false,
 619     );
 620 }
 621
 622 #[test]
 623 fn test_highlight_doc_comment() {
 624     check_highlighting(
 625         r#"
 626 //! This is a module to test doc injection.
 627 //! ```
 628 //! fn test() {}
 629 //! ```
 630
 631 /// ```
 632 /// let _ = "early doctests should not go boom";
 633 /// ```
 634 struct Foo {
 635     bar: bool,
 636 }
 637
 638 /// This is an impl with a code block.
 639 ///
 640 /// ```
 641 /// fn foo() {
 642 ///
 643 /// }
 644 /// ```
 645 impl Foo {
 646     /// ```
 647     /// let _ = "Call me
 648     //    KILLER WHALE
 649     ///     Ishmael.";
 650     /// ```
 651     pub const bar: bool = true;
 652
 653     /// Constructs a new `Foo`.
 654     ///
 655     /// # Examples
 656     ///
 657     /// ```
 658     /// # #![allow(unused_mut)]
 659     /// let mut foo: Foo = Foo::new();
 660     /// ```
 661     pub const fn new() -> Foo {
 662         Foo { bar: true }
 663     }
 664
 665     /// `bar` method on `Foo`.
 666     ///
 667     /// # Examples
 668     ///
 669     /// ```
 670     /// use x::y;
 671     ///
 672     /// let foo = Foo::new();
 673     ///
 674     /// // calls bar on foo
 675     /// assert!(foo.bar());
 676     ///
 677     /// let bar = foo.bar || Foo::bar;
 678     ///
 679     /// /* multi-line
 680     ///        comment */
 681     ///
 682     /// let multi_line_string = "Foo
 683     ///   bar\n
 684     ///          ";
 685     ///
 686     /// ```
 687     ///
 688     /// ```rust,no_run
 689     /// let foobar = Foo::new().bar();
 690     /// ```
 691     ///
 692     /// ```sh
 693     /// echo 1
 694     /// ```
 695     pub fn foo(&self) -> bool {
 696         true
 697     }
 698 }
 699
 700 /// [`Foo`](Foo) is a struct
 701 /// This function is > [`all_the_links`](all_the_links) <
 702 /// [`noop`](noop) is a macro below
 703 /// [`Item`] is a struct in the module [`module`]
 704 ///
 705 /// [`Item`]: module::Item
 706 /// [mix_and_match]: ThisShouldntResolve
 707 pub fn all_the_links() {}
 708
 709 pub mod module {
 710     pub struct Item;
 711 }
 712
 713 /// ```
 714 /// noop!(1);
 715 /// ```
 716 macro_rules! noop {
 717     ($expr:expr) => {
 718         $expr
 719     }
 720 }
 721
 722 /// ```rust
 723 /// let _ = example(&[1, 2, 3]);
 724 /// ```
 725 ///
 726 /// ```
 727 /// loop {}
 728 #[cfg_attr(not(feature = "false"), doc = "loop {}")]
 729 #[doc = "loop {}"]
 730 /// ```
 731 ///
 732 #[cfg_attr(feature = "alloc", doc = "```rust")]
 733 #[cfg_attr(not(feature = "alloc"), doc = "```ignore")]
 734 /// let _ = example(&alloc::vec![1, 2, 3]);
 735 /// ```
 736 pub fn mix_and_match() {}
 737
 738 /**
 739 It is beyond me why you'd use these when you got ///
 740 ```rust
 741 let _ = example(&[1, 2, 3]);
 742 ```
 743 [`block_comments2`] tests these with indentation
 744  */
 745 pub fn block_comments() {}
 746
 747 /**
 748     Really, I don't get it
 749     ```rust
 750     let _ = example(&[1, 2, 3]);
 751     ```
 752     [`block_comments`] tests these without indentation
 753 */
 754 pub fn block_comments2() {}
 755 "#
 756         .trim(),
 757         expect_file!["./test_data/highlight_doctest.html"],
 758         false,
 759     );
 760 }
 761
 762 #[test]
 763 fn test_extern_crate() {
 764     check_highlighting(
 765         r#"
 766         //- /main.rs crate:main deps:std,alloc
 767         extern crate std;
 768         extern crate alloc as abc;
 769         //- /std/lib.rs crate:std
 770         pub struct S;
 771         //- /alloc/lib.rs crate:alloc
 772         pub struct A
 773         "#,
 774         expect_file!["./test_data/highlight_extern_crate.html"],
 775         false,
 776     );
 777 }
 778
 779 #[test]
 780 fn test_associated_function() {
 781     check_highlighting(
 782         r#"
 783 fn not_static() {}
 784
 785 struct foo {}
 786
 787 impl foo {
 788     pub fn is_static() {}
 789     pub fn is_not_static(&self) {}
 790 }
 791
 792 trait t {
 793     fn t_is_static() {}
 794     fn t_is_not_static(&self) {}
 795 }
 796
 797 impl t for foo {
 798     pub fn is_static() {}
 799     pub fn is_not_static(&self) {}
 800 }
 801         "#,
 802         expect_file!["./test_data/highlight_assoc_functions.html"],
 803         false,
 804     )
 805 }
 806
 807 #[test]
 808 fn test_injection() {
 809     check_highlighting(
 810         r##"
 811 fn f(ra_fixture: &str) {}
 812 fn main() {
 813     f(r"
 814 fn foo() {
 815     foo(\$0{
 816         92
 817     }\$0)
 818 }");
 819 }
 820     "##,
 821         expect_file!["./test_data/injection.html"],
 822         false,
 823     );
 824 }
 825
 826 /// Highlights the code given by the `ra_fixture` argument, renders the
 827 /// result as HTML, and compares it with the HTML file given as `snapshot`.
 828 /// Note that the `snapshot` file is overwritten by the rendered HTML.
 829 fn check_highlighting(ra_fixture: &str, expect: ExpectFile, rainbow: bool) {
 830     let (analysis, file_id) = fixture::file(ra_fixture);
 831     let actual_html = &analysis.highlight_as_html(file_id, rainbow).unwrap();
 832     expect.assert_eq(actual_html)
 833 }