PHP中MongoDb应用类库

<?php /** * Test cases for the Cakephp mongoDB datasource. * * This datasource uses Pecl Mongo (http://php.net/mongo) * and is thus dependent on PHP 5.0 and greater. * * Copyright 2010, Yasushi Ichikawa http://github.com/ichikaway/ * * Licensed under The MIT License * Redistributions of files must retain the above copyright notice. * * @copyright Copyright (c) 2010, Yasushi Ichikawa http://github.com/ichikaway/ * @package mongodb * @subpackage mongodb.tests.cases.datasources * @license http://www.opensource.org/licenses/mit-license.php The MIT License */ /** * Import relevant classes for testing */ App::import('Model', 'Mongodb.MongodbSource'); /** * Generate Mock Model */ Mock::generate('AppModel', 'MockPost'); /** * Post Model for the test * * @package app * @subpackage app.model.post */ class Post extends AppModel { public $useDbConfig = 'mongo_test'; /** * mongoSchema property * * @public array * @access public */ public $primaryKey='_id'; public $validate = array( 'uniquefield1' => array( 'rule' => 'isUnique', 'required' => false ), 'uniquefield2' => array( 'rule' => 'manualUniqueValidation', 'required' => false ), ); public $mongoSchema = array( 'title' => array('type' => 'string'), 'body' => array('type' => 'string'), 'text' => array('type' => 'text'), 'uniquefield1' => array('type' => 'text'), 'uniquefield2' => array('type' => 'text'), 'count' => array('type' => 'integer'), 'created' => array('type' => 'datetime'), 'modified' => array('type' => 'datetime'), ); function manualUniqueValidation($check) { $c = $this->find('count', array( 'conditions' => array( 'uniquefield2' => $check['uniquefield2'] ) )); if ($c === 0) { return true; } return false; } } /** * MongoArticle class * * @uses AppModel * @package mongodb * @subpackage mongodb.tests.cases.datasources */ class MongoArticle extends AppModel { public $useDbConfig = 'mongo_test'; } /** * MongoDB Source test class * * @package app * @subpackage app.model.datasources */ class MongodbSourceTest extends CakeTestCase { /** * Database Instance * * @var resource * @access public */ public $mongodb; /** * Base Config * * @var array * @access public * */ protected $_config = array( 'datasource' => 'mongodb', 'host' => 'localhost', 'login' => '', 'password' => '', 'database' => 'test_mongo', 'port' => 27017, 'prefix' => '', 'persistent' => true, ); /** * Sets up the environment for each test method * * @return void * @access public */ public function startTest() { $connections = ConnectionManager::enumConnectionObjects(); if (!empty($connections['test']['classname']) && $connections['test']['classname'] === 'mongodbSource') { $config = new DATABASE_CONFIG(); $this->_config = $config->test; } ConnectionManager::create('mongo_test', $this->_config); $this->Mongo = new MongodbSource($this->_config); $this->Post = ClassRegistry::init('Post'); $this->Post->setDataSource('mongo_test'); $this->mongodb =& ConnectionManager::getDataSource($this->Post->useDbConfig); $this->mongodb->connect(); $this->dropData(); } /** * Destroys the environment after each test method is run * * @return void * @access public */ public function endTest() { $this->dropData(); unset($this->Post); unset($this->Mongo); unset($this->mongodb); ClassRegistry::flush(); } /** * get Mongod server version * * @return numeric * @access public */ public function getMongodVersion() { $mongo = $this->Post->getDataSource(); return $mongo->execute('db.version()'); } /** * Insert data method for mongodb. * * @param array insert data * @return void * @access public */ public function insertData($data) { try { $this->mongodb ->connection ->selectDB($this->_config['database']) ->selectCollection($this->Post->table) ->insert($data, true); } catch (MongoException $e) { trigger_error($e->getMessage()); } } /** * Drop database * * @return void * @access public */ public function dropData() { try { $db = $this->mongodb ->connection ->selectDB($this->_config['database']); foreach($db->listCollections() as $collection) { $collection->drop(); } } catch (MongoException $e) { trigger_error($e->getMessage()); } } /** * testCreateConnectionName * * @return void * @access public */ public function testCreateConnectionName() { $config = array( 'datasource' => 'mongodb', 'host' => 'localhost', 'login' => '', 'password' => '', 'database' => 'test_mongo', 'port' => 27017, 'prefix' => '', 'persistent' => false, ); $version = '1.2.2'; $expect = 'mongodb://localhost:27017'; $host = $this->mongodb->createConnectionName($config, $version); $this->assertIdentical($expect, $host); $config = array( 'datasource' => 'mongodb', 'host' => 'localhost', 'login' => 'user', 'password' => 'pass', 'database' => 'test_mongo', 'port' => 27017, 'prefix' => '', 'persistent' => false, ); $version = '1.2.2'; $expect = 'mongodb://user:pass@localhost:27017/test_mongo'; $host = $this->mongodb->createConnectionName($config, $version); $this->assertIdentical($expect, $host); $config = array( 'datasource' => 'mongodb', 'host' => 'localhost', 'login' => 'user', 'password' => 'pass', 'database' => 'test_mongo', 'port' => 27017, 'prefix' => '', 'persistent' => false, ); $version = '1.0.0'; $expect = 'user:pass@localhost:27017/test_mongo'; $host = $this->mongodb->createConnectionName($config, $version); $this->assertIdentical($expect, $host); } /** * Tests connection * * @return void * @access public */ public function testConnect() { $result = $this->Mongo->connect(); $this->assertTrue($result); $this->assertTrue($this->Mongo->connected); $this->assertTrue($this->Mongo->isConnected()); } /** * Tests the disconnect method of the Mongodb DataSource * * @return void * @access public */ public function testDisconnect() { $result = $this->Mongo->disconnect(); $this->assertTrue($result); $this->assertFalse($this->Mongo->connected); } /** * Tests the listSources method of the Mongodb DataSource * * @return void * @access public */ public function testListSources() { $this->assertTrue(is_array($this->mongodb->listSources())); } /** * Tests the getMongoDb method of the Mongodb DataSource * * @return void * @access public */ public function testGetMongoDb() { $obj = $this->mongodb->getMongoDb(); $this->assertTrue(is_object($obj)); $objName = get_class($obj); $this->assertEqual('MongoDB', $objName); } /** * Tests the Model::getMongoDb() call MongodbSource::getMongoDb * * @return void * @access public */ public function testGetMongoDbFromModel() { $obj = $this->Post->getMongoDb(); $this->assertTrue(is_object($obj)); $objName = get_class($obj); $this->assertEqual('MongoDB', $objName); } /** * Tests the getMongoCollection method of the Mongodb DataSource * * @return void * @access public */ public function testGetMongoCollection() { $obj = $this->mongodb->getMongoCollection($this->Post); $this->assertTrue(is_object($obj)); $objName = get_class($obj); $this->assertEqual('MongoCollection', $objName); } /** * Tests the describe method of the Mongodb DataSource * * @return void * @access public */ public function testDescribe() { $mockObj = new MockPost();