cleanup and add comment

kolide · Dec 19, 2024 · 2d74749 · 2d74749
1 parent b9e1918
commit 2d74749
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pkg/osquery/runtime/runner.go b/pkg/osquery/runtime/runner.go
@@ -369,6 +369,8 @@ func (r *Runner) InstanceStatuses() map[string]types.InstanceStatus {
 	return instanceStatuses
 }
 
+// UpdateRegistrationIDs detects any changes between the new and stored registration IDs,
+// and resets the runner instances for the new registrationIDs if required
 func (r *Runner) UpdateRegistrationIDs(newRegistrationIDs []string) error {
 	slices.Sort(newRegistrationIDs)
 	existingRegistrationIDs := r.registrationIds

diff --git a/pkg/osquery/runtime/runtime_test.go b/pkg/osquery/runtime/runtime_test.go
@@ -674,7 +674,7 @@ func TestMultipleInstancesWithUpdatedRegistrationIDs(t *testing.T) {
 
 	waitShutdown(t, runner, logBytes)
 
-	// Confirm both instances exited
+	// Confirm instance exited
 	require.Contains(t, runner.instances, types.DefaultRegistrationID)
 	require.NotNil(t, runner.instances[types.DefaultRegistrationID].stats)
 	require.NotEmpty(t, runner.instances[types.DefaultRegistrationID].stats.ExitTime, "exit time should be added to default instance stats on shutdown")