Merge pull request #193 from Unity-Technologies/Uni-29079_Unit_test_for_MapNameToSourceRecursive

Uni-29079 [ADDED] unit test for the function

Author: AustinJubrey

Assets/FbxExporters/Editor/ConvertToModel.cs

@@ -294,7 +294,7 @@ public static void UpdateFromSourceRecursive(GameObject dest, GameObject source)
             /// <returns>Dictionary containing the name to source game object.</returns>
             /// <param name="dest">Destination GameObject.</param>
             /// <param name="source">Source GameObject.</param>
-            private static Dictionary<string,GameObject> MapNameToSourceRecursive(GameObject dest, GameObject source){
+            public static Dictionary<string,GameObject> MapNameToSourceRecursive(GameObject dest, GameObject source){
                 var nameToGO = new Dictionary<string,GameObject> ();
 
                 var q = new Queue<Transform> ();

Assets/FbxExporters/Editor/UnitTests/ConvertToModelTest.cs

@@ -182,5 +182,56 @@ public void SkinnedMeshTest()
             // In the future we'll want to assert the opposite!
             Assert.That(cubeInstance.GetComponentsInChildren<SkinnedMeshRenderer>(), Is.Empty);
         }
+
+        [Test]
+        public void MapNameToSourceTest()
+        {
+            //Create a cube with 3 children game objects
+            var cube = GameObject.CreatePrimitive(PrimitiveType.Cube);
+            var capsule = GameObject.CreatePrimitive(PrimitiveType.Capsule);
+            var sphere = GameObject.CreatePrimitive(PrimitiveType.Sphere);
+            var quad = GameObject.CreatePrimitive(PrimitiveType.Quad);
+
+            capsule.transform.parent = cube.transform;
+            sphere.transform.parent = cube.transform;
+            quad.transform.parent = cube.transform;
+            capsule.transform.SetSiblingIndex(0);
+
+            //Create a similar Heirarchy that we can use as our phony "exported" hierarchy.
+            var cube2 = GameObject.CreatePrimitive(PrimitiveType.Cube);
+            var capsule2 = GameObject.CreatePrimitive(PrimitiveType.Capsule);
+            var sphere2 = GameObject.CreatePrimitive(PrimitiveType.Sphere);
+            var quad2 = GameObject.CreatePrimitive(PrimitiveType.Quad);
+
+            capsule2.transform.parent = cube2.transform;
+            sphere2.transform.parent = cube2.transform;
+            quad2.transform.parent = cube2.transform;
+            capsule.transform.SetSiblingIndex(1);
+
+            var dictionary = ConvertToModel.MapNameToSourceRecursive(cube, cube2);
+
+            //We expect these to pass because we've given it an identical game object, as it would have after a normal export.
+            Assert.AreSame(capsule2, dictionary[capsule.name]);
+            Assert.AreSame(sphere2, dictionary[sphere.name]);
+            Assert.AreSame(quad2, dictionary[quad.name]);
+            Assert.True(dictionary.Count == 4);
+
+            //Create a broken hierarchy, one that is missing a primitive
+            var cube3 = GameObject.CreatePrimitive(PrimitiveType.Cube);
+            var capsule3 = GameObject.CreatePrimitive(PrimitiveType.Capsule);
+            var sphere3 = GameObject.CreatePrimitive(PrimitiveType.Sphere);
+
+            capsule3.transform.parent = cube3.transform;
+            sphere3.transform.parent = cube3.transform;
+
+            var dictionaryBroken = ConvertToModel.MapNameToSourceRecursive(cube, cube3);
+
+            //the dictionary size should be equal to the amount of children + the parent
+            Assert.True(dictionaryBroken.Count == 4);
+
+            Assert.IsNull(dictionaryBroken[quad.name]);
+            Assert.AreSame(capsule3, dictionaryBroken[capsule.name]);
+            Assert.AreSame(sphere3, dictionaryBroken[sphere.name]);
+        }
     }
 }