From cb3211cf4602eb47d5a2aa113ac1d7b56ca8cf42 Mon Sep 17 00:00:00 2001
From: Lucien Greathouse <me@lpghatguy.com>
Date: Fri, 4 Oct 2019 18:10:45 -0700
Subject: [PATCH] Mark project node as instigating source for instances from
 project

---
 src/snapshot_middleware/project.rs                        | 7 ++++++-
 .../snapshots/test__project_with_children.snap            | 8 ++++++++
 .../test__project_with_path_to_project_with_children.snap | 8 ++++++++
 3 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/src/snapshot_middleware/project.rs b/src/snapshot_middleware/project.rs
index a0b5d6aa..4ea7b267 100644
--- a/src/snapshot_middleware/project.rs
+++ b/src/snapshot_middleware/project.rs
@@ -6,7 +6,7 @@ use rbx_reflection::try_resolve_value;
 use crate::{
     imfs::{FsErrorKind, Imfs, ImfsEntry, ImfsFetcher},
     project::{Project, ProjectNode},
-    snapshot::{InstanceMetadata, InstanceSnapshot},
+    snapshot::{InstanceMetadata, InstanceSnapshot, InstigatingSource},
 };
 
 use super::{
@@ -174,6 +174,11 @@ fn snapshot_project_node<F: ImfsFetcher>(
         metadata.ignore_unknown_instances = true;
     }
 
+    metadata.instigating_source = Some(InstigatingSource::ProjectNode(
+        instance_name.to_string(),
+        node.clone(),
+    ));
+
     Ok(Some(InstanceSnapshot {
         snapshot_id: None,
         name,
diff --git a/src/snapshot_middleware/snapshots/test__project_with_children.snap b/src/snapshot_middleware/snapshots/test__project_with_children.snap
index b6329a43..1893e242 100644
--- a/src/snapshot_middleware/snapshots/test__project_with_children.snap
+++ b/src/snapshot_middleware/snapshots/test__project_with_children.snap
@@ -16,6 +16,14 @@ children:
   - snapshot_id: ~
     metadata:
       ignore_unknown_instances: true
+      instigating_source:
+        ProjectNode:
+          - Child
+          - class_name: Model
+            children: {}
+            properties: {}
+            ignore_unknown_instances: ~
+            path: ~
       relevant_paths: []
     name: Child
     class_name: Model
diff --git a/src/snapshot_middleware/snapshots/test__project_with_path_to_project_with_children.snap b/src/snapshot_middleware/snapshots/test__project_with_path_to_project_with_children.snap
index e0ffbefd..06fca8b6 100644
--- a/src/snapshot_middleware/snapshots/test__project_with_path_to_project_with_children.snap
+++ b/src/snapshot_middleware/snapshots/test__project_with_path_to_project_with_children.snap
@@ -17,6 +17,14 @@ children:
   - snapshot_id: ~
     metadata:
       ignore_unknown_instances: true
+      instigating_source:
+        ProjectNode:
+          - SomeChild
+          - class_name: Model
+            children: {}
+            properties: {}
+            ignore_unknown_instances: ~
+            path: ~
       relevant_paths: []
     name: SomeChild
     class_name: Model