BUGFIX: Pass correct class to allowPrimaryKeyEditing in yaml fixture

git-svn-id: svn://svn.silverstripe.com/silverstripe/open/modules/sapphire/branches/2.4@101171 467b73ca-7a2a-4603-9d3b-597d59a354a9
silverstripe · Feb 2, 2011 · f9836b0 · f9836b0
1 parent 5253803
commit f9836b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/model/YamlFixture.php b/core/model/YamlFixture.php
@@ -178,9 +178,9 @@ protected function writeDataObject($dataClass, $items) {
 
 				// The database needs to allow inserting values into the foreign key column (ID in our case)
 				$conn = DB::getConn();
-				if(method_exists($conn, 'allowPrimaryKeyEditing')) $conn->allowPrimaryKeyEditing($dataClass, true);
+				if(method_exists($conn, 'allowPrimaryKeyEditing')) $conn->allowPrimaryKeyEditing(ClassInfo::baseDataClass($dataClass), true);
 				$obj->write(false, true);
-				if(method_exists($conn, 'allowPrimaryKeyEditing')) $conn->allowPrimaryKeyEditing($dataClass, false);
+				if(method_exists($conn, 'allowPrimaryKeyEditing')) $conn->allowPrimaryKeyEditing(ClassInfo::baseDataClass($dataClass), false);
 			}
 
 			// Populate the dictionary with the ID