[red-knot] Fix file watching for new non-project files

Author: MichaReiser

crates/red_knot/tests/file_watching.rs

@@ -462,6 +462,41 @@ fn new_ignored_file() -> anyhow::Result<()> {
     Ok(())
 }
 
+#[test]
+fn new_non_project_file() -> anyhow::Result<()> {
+    let mut case = setup_with_options([("bar.py", "")], |context| {
+        Some(Options {
+            environment: Some(EnvironmentOptions {
+                extra_paths: Some(vec![RelativePathBuf::cli(
+                    context.join_root_path("site_packages"),
+                )]),
+                ..EnvironmentOptions::default()
+            }),
+            ..Options::default()
+        })
+    })?;
+
+    let bar_path = case.project_path("bar.py");
+    let bar_file = case.system_file(&bar_path).unwrap();
+
+    assert_eq!(&case.collect_project_files(), &[bar_file]);
+
+    // Add a file to site packages
+    let black_path = case.root_path().join("site_packages/black.py");
+    std::fs::write(black_path.as_std_path(), "print('Hello')")?;
+
+    let changes = case.stop_watch(event_for_file("black.py"));
+
+    case.apply_changes(changes);
+
+    assert!(case.system_file(&black_path).is_ok());
+
+    // The file should not have been added to the project files
+    assert_eq!(&case.collect_project_files(), &[bar_file]);
+
+    Ok(())
+}
+
 #[test]
 fn changed_file() -> anyhow::Result<()> {
     let foo_source = "print('Hello, world!')";
@@ -1075,6 +1110,7 @@ fn hard_links_in_project() -> anyhow::Result<()> {
 
     assert_eq!(source_text(case.db(), foo).as_str(), "print('Version 1')");
     assert_eq!(source_text(case.db(), bar).as_str(), "print('Version 1')");
+    assert_eq!(case.collect_project_files(), &[bar, foo]);
 
     // Write to the hard link target.
     update_file(foo_path, "print('Version 2')").context("Failed to update foo.py")?;
@@ -1268,6 +1304,8 @@ mod unix {
             Some(&*baz_project)
         );
 
+        assert_eq!(case.collect_project_files(), &[baz.file()]);
+
         let baz_original = case.root_path().join("bar/baz.py");
 
         // Write to the symlink target.
@@ -1283,6 +1321,8 @@ mod unix {
             "def baz(): print('Version 2')"
         );
 
+        assert_eq!(case.collect_project_files(), &[baz.file()]);
+
         // Write to the symlink source.
         update_file(baz_project, "def baz(): print('Version 3')")
             .context("Failed to update bar/baz.py")?;
@@ -1295,6 +1335,7 @@ mod unix {
             source_text(case.db(), baz.file()).as_str(),
             "def baz(): print('Version 3')"
         );
+        assert_eq!(case.collect_project_files(), &[baz.file()]);
 
         Ok(())
     }
@@ -1354,6 +1395,8 @@ mod unix {
         );
         assert_eq!(baz.file().path(case.db()).as_system_path(), Some(&*bar_baz));
 
+        assert_eq!(case.collect_project_files(), &[patched_bar_baz_file]);
+
         // Write to the symlink target.
         update_file(&patched_bar_baz, "def baz(): print('Version 2')")
             .context("Failed to update bar/baz.py")?;
@@ -1389,6 +1432,7 @@ mod unix {
             bar_baz_text = bar_baz_text.as_str()
         );
 
+        assert_eq!(case.collect_project_files(), &[patched_bar_baz_file]);
         Ok(())
     }
 
@@ -1469,6 +1513,8 @@ mod unix {
             Some(&*baz_original)
         );
 
+        assert_eq!(case.collect_project_files(), &[]);
+
         // Write to the symlink target.
         update_file(&baz_original, "def baz(): print('Version 2')")
             .context("Failed to update bar/baz.py")?;
@@ -1494,6 +1540,8 @@ mod unix {
             "def baz(): print('Version 2')"
         );
 
+        assert_eq!(case.collect_project_files(), &[]);
+
         Ok(())
     }
 }

crates/red_knot_project/src/db/changes.rs

@@ -208,11 +208,12 @@ impl ProjectDatabase {
                         return WalkState::Continue;
                     }
 
-                    if entry
-                        .path()
-                        .extension()
-                        .and_then(PySourceType::try_from_extension)
-                        .is_some()
+                    if entry.path().starts_with(&project_path)
+                        && entry
+                            .path()
+                            .extension()
+                            .and_then(PySourceType::try_from_extension)
+                            .is_some()
                     {
                         let mut paths = added_paths.lock().unwrap();