Allows an _id attribute of another type than string

Author: v-six

src/Jenssegers/Mongodb/Model.php

@@ -54,7 +54,11 @@ public function getIdAttribute($value)
         // Return _id as string
         if (array_key_exists('_id', $this->attributes))
         {
-            return (string) $this->attributes['_id'];
+            if ($this->attributes['_id'] instanceof MongoId) {
+                return (string) $this->attributes['_id'];
+            }
+
+            return $this->attributes['_id'];
         }
     }
 

tests/ModelTest.php

@@ -79,7 +79,7 @@ public function testUpdate()
 		$this->assertEquals(20, $check->age);
 	}
 
-	public function testManualId()
+	public function testManualStringId()
 	{
 		$user = new User;
 		$user->_id = '4af9f23d8ead0e1d32000000';
@@ -93,6 +93,35 @@ public function testManualId()
 
 		$raw = $user->getAttributes();
 		$this->assertInstanceOf('MongoId', $raw['_id']);
+
+		$user = new User;
+		$user->_id = 'customId';
+		$user->name = 'John Doe';
+		$user->title = 'admin';
+		$user->age = 35;
+		$user->save();
+
+		$this->assertEquals(true, $user->exists);
+		$this->assertEquals('customId', $user->_id);
+
+		$raw = $user->getAttributes();
+		$this->assertInternalType('string', $raw['_id']);
+	}
+
+	public function testManualIntId()
+	{
+		$user = new User;
+		$user->_id = 1;
+		$user->name = 'John Doe';
+		$user->title = 'admin';
+		$user->age = 35;
+		$user->save();
+
+		$this->assertEquals(true, $user->exists);
+		$this->assertEquals(1, $user->_id);
+
+		$raw = $user->getAttributes();
+		$this->assertInternalType('integer', $raw['_id']);
 	}
 
 	public function testDelete()