Tweaked tests

Author: jenssegers

tests/QueryBuilderTest.php

@@ -234,10 +234,10 @@ public function testPull()
 	public function testDistinct()
 	{
 		DB::collection('items')->insert(array(
-			array('name' => 'knife', 'type' => 'sharp', 'amount' => 34),
-			array('name' => 'fork',  'type' => 'sharp', 'amount' => 20),
-			array('name' => 'spoon', 'type' => 'round', 'amount' => 3),
-			array('name' => 'spoon', 'type' => 'round', 'amount' => 14)
+			array('name' => 'knife', 'type' => 'sharp',),
+			array('name' => 'fork',  'type' => 'sharp'),
+			array('name' => 'spoon', 'type' => 'round'),
+			array('name' => 'spoon', 'type' => 'round')
 		));
 
 		$items = DB::collection('items')->distinct('name')->get();
@@ -260,6 +260,14 @@ public function testCustomId()
 
 		$item = DB::collection('items')->where('_id', 'fork')->first();
 		$this->assertEquals('fork', $item['_id']);
+
+		DB::collection('users')->insert(array(
+			array('_id' => 1, 'name' => 'Jane Doe'),
+			array('_id' => 2, 'name' => 'John Doe')
+		));
+
+		$item = DB::collection('users')->find(1);
+		$this->assertEquals(1, $item['_id']);
 	}
 
 	public function testTake()

tests/QueryTest.php

@@ -3,7 +3,7 @@
 
 class QueryTest extends PHPUnit_Framework_TestCase {
 
-	public function setUp()
+	public static function setUpBeforeClass()
 	{
 		// test data
 		User::create(array('name' => 'John Doe', 'age' => 35, 'title' => 'admin'));
@@ -17,7 +17,7 @@ public function setUp()
 		User::create(array('name' => 'Error', 'age' => null, 'title' => null));
 	}
 
-	public function tearDown()
+	public static function tearDownAfterClass()
 	{
 		User::truncate();
 	}