added test for adding device late

Author: ritamerkl

Assets/Tests/InputSystem/Plugins/XRTests.cs

@@ -1,6 +1,7 @@
 // ENABLE_VR is not defined on Game Core but the assembly is available with limited features when the XR module is enabled.
 #if ENABLE_VR || UNITY_GAMECORE
 using System;
+using System.Collections;
 using System.Collections.Generic;
 using System.Runtime.InteropServices;
 using NUnit.Framework;
@@ -11,6 +12,7 @@
 using UnityEngine.InputSystem.LowLevel;
 using UnityEngine.InputSystem.Utilities;
 using UnityEngine.InputSystem.XR;
+using UnityEngine.TestTools;
 using UnityEngine.XR;
 
 using Usages = UnityEngine.InputSystem.CommonUsages;
@@ -629,6 +631,57 @@ public void Components_TrackedPoseDriver_DoesNotEnableOrDisableReferenceActions(
         Assert.That(trackingStateInput.action.enabled, Is.False);
     }
 
+    [UnityTest]
+    [Category("Components")]
+    public IEnumerator LateAddedXRControllerCanLinkTrackedPoseDriver()
+    {
+        var go = new GameObject();
+        var tpd = go.AddComponent<TrackedPoseDriver>();
+        tpd.updateType = TrackedPoseDriver.UpdateType.UpdateAndBeforeRender;
+        tpd.trackingType = TrackedPoseDriver.TrackingType.RotationAndPosition;
+        tpd.ignoreTrackingState = false;
+        var transform = tpd.transform;
+
+        var positionAction = new InputAction(binding: "<XRController>/devicePosition");
+        var rotationAction = new InputAction(binding: "<XRController>/deviceRotation");
+        var trackingStateAction = new InputAction(binding: "<XRController>/trackingState");
+        tpd.positionInput = new InputActionProperty(positionAction);
+        tpd.rotationInput = new InputActionProperty(rotationAction);
+        tpd.trackingStateInput = new InputActionProperty(trackingStateAction);
+
+        yield return null;
+
+        Assert.That(positionAction.controls.Count, Is.EqualTo(0));
+        Assert.That(rotationAction.controls.Count, Is.EqualTo(0));
+
+        var device = InputSystem.AddDevice<XRController>();
+        InputSystem.AddDeviceUsage(device, "RightHand");
+
+        yield return null;
+
+        Assert.That(positionAction.controls.Count, Is.EqualTo(1));
+        Assert.That(rotationAction.controls.Count, Is.EqualTo(1));
+
+        var position = new Vector3(1f, 2f, 3f);
+        var rotation = new Quaternion(0.09853293f, 0.09853293f, 0.09853293f, 0.9853293f);
+        using (StateEvent.From(device, out var stateEvent))
+        {
+            device.devicePosition.WriteValueIntoEvent(position, stateEvent);
+            device.deviceRotation.WriteValueIntoEvent(rotation, stateEvent);
+            device.trackingState.WriteValueIntoEvent((int)(InputTrackingState.Position | InputTrackingState.Rotation), stateEvent);
+
+            transform.position = Vector3.zero;
+            transform.rotation = Quaternion.identity;
+            InputSystem.QueueEvent(stateEvent);
+            InputSystem.Update(InputUpdateType.Dynamic);
+        }
+
+        yield return null;
+
+        Assert.That(transform.position, Is.EqualTo(position));
+        Assert.That(transform.rotation, Is.EqualTo(rotation));
+    }
+
     [Test]
     [Category("Components")]
     public void Components_TrackedPoseDriver_RequiresResolvedTrackingStateBindings()