Skip to content

Commit

Permalink
Consolidating $useDbConfig to $this->connection.
Browse files Browse the repository at this point in the history 
Updating test cases.
  • Loading branch information
@markstory
markstory committed May 5, 2009
1 parent f4dc4bc commit bb2f6b2
Show file tree
Hide file tree
Showing 2 changed files with 56 additions and 34 deletions.
81 changes: 48 additions & 33 deletions cake/console/libs/tasks/model.php
View file
Expand Up @@ -39,6 +39,13 @@ class ModelTask extends Shell {
* @access public
*/
var $plugin = null;
/**
* Name of the db connection used.
*
* @var string
* @access public
*/
var $connection = null;
/**
* path to MODELS directory
*
Expand Down Expand Up @@ -87,9 +94,8 @@ function execute() {
* @return void
**/
function all() {
$ds = 'default';
if (isset($this->params['connection'])) {
$ds = $this->params['connection'];
if (!isset($this->params['connection'])) {
$this->connection = 'default';
}
$this->listAll($ds, false);
$this->interactive = false;
Expand Down Expand Up @@ -121,20 +127,21 @@ function __interactive() {

$useTable = null;
$primaryKey = 'id';
$validate = array();
$associations = array('belongsTo'=> array(), 'hasOne'=> array(), 'hasMany' => array(), 'hasAndBelongsToMany'=> array());

$useDbConfig = $this->DbConfig->getConfig();
$currentModelName = $this->getName($useDbConfig);
$useTable = $this->getTable($currentModelName, $useDbConfig);
$db =& ConnectionManager::getDataSource($useDbConfig);
$validate = $associations = array();

if (empty($this->connection)) {
$this->connection = $this->DbConfig->getConfig();
}
$currentModelName = $this->getName();
$useTable = $this->getTable($currentModelName);
$db =& ConnectionManager::getDataSource($this->connection);
$fullTableName = $db->fullTableName($useTable);

$wannaDoValidation = $this->in(__('Would you like to supply validation criteria for the fields in your model?', true), array('y','n'), 'y');

if (in_array($useTable, $this->__tables)) {
App::import('Model');
$tempModel = new Model(array('name' => $currentModelName, 'table' => $useTable, 'ds' => $useDbConfig));
$tempModel = new Model(array('name' => $currentModelName, 'table' => $useTable, 'ds' => $this->connection));

$fields = $tempModel->schema();
if (!array_key_exists('id', $fields)) {
Expand Down Expand Up @@ -162,7 +169,7 @@ function __interactive() {
$this->hr();
$this->out("Name: " . $currentModelName);

if ($useDbConfig !== 'default') {
if ($this->connection !== 'default') {
$this->out("DB Config: " . $useDbConfig);
}
if ($fullTableName !== Inflector::tableize($currentModelName)) {
Expand Down Expand Up @@ -205,7 +212,7 @@ function __interactive() {
$looksGood = $this->in(__('Look okay?', true), array('y','n'), 'y');

if (low($looksGood) == 'y' || low($looksGood) == 'yes') {
if ($this->bake($currentModelName, $associations, $validate, $primaryKey, $useTable, $useDbConfig)) {
if ($this->bake($currentModelName, $associations, $validate, $primaryKey, $useTable, $this->connection)) {
if ($this->_checkUnitTest()) {
$this->bakeTest($currentModelName, $useTable, $associations);
}
Expand Down Expand Up @@ -233,7 +240,6 @@ function doValidation(&$model, $interactive = true) {
}

$validate = array();

$options = array();

if (class_exists('Validation')) {
Expand All @@ -242,29 +248,32 @@ function doValidation(&$model, $interactive = true) {
}

foreach ($fields as $fieldName => $field) {
$prompt = 'Field: ' . $fieldName . "\n";
$prompt .= 'Type: ' . $field['type'] . "\n";
$prompt .= '---------------------------------------------------------------'."\n";
$prompt .= 'Please select one of the following validation options:'."\n";
$prompt .= '---------------------------------------------------------------'."\n";
if ($this->interactive) {
$this->out('');
$this->out(sprintf(__('Field: %s', true), $fieldName));
$this->out(sprintf(__('Type: %s', true), $field['type']));
$this->hr();
$this->out(__('Please select one of the following validation options:', true));
$this->hr();
}

sort($options);

$skip = 1;
$prompt = '';
$default = 1;
foreach ($options as $key => $option) {
if ($option{0} != '_' && strtolower($option) != 'getinstance') {
$prompt .= "{$skip} - {$option}\n";
$choices[$skip] = strtolower($option);
$skip++;
$prompt .= "{$default} - {$option}\n";
$choices[$default] = strtolower($option);
$default++;
}
}

$methods = array_flip($choices);

$prompt .= "{$skip} - Do not do any validation on this field.\n";
$prompt .= "... or enter in a valid regex validation string.\n";
$prompt .= sprintf(__("%s - Do not do any validation on this field.\n", true), $default);
$prompt .= __("... or enter in a valid regex validation string.\n", true);

$guess = $skip;
$guess = $default;
if ($field['null'] != 1 && $fieldName != $model->primaryKey && !in_array($fieldName, array('created', 'modified', 'updated'))) {
if ($fieldName == 'email') {
$guess = $methods['email'];
Expand All @@ -280,12 +289,11 @@ function doValidation(&$model, $interactive = true) {
}

if ($interactive === true) {
$this->out('');
$choice = $this->in($prompt, null, $guess);
} else {
$choice = $guess;
}
if ($choice != $skip) {
if ($choice != $default) {
if (is_numeric($choice) && isset($choices[$choice])) {
$validate[$fieldName] = $choices[$choice];
} else {
Expand All @@ -305,7 +313,6 @@ function doValidation(&$model, $interactive = true) {
* @access public
*/
function doAssociations(&$model, $interactive = true) {

if (!is_object($model)) {
return false;
}
Expand Down Expand Up @@ -735,13 +742,17 @@ function bakeTest($className, $useTable = null, $associations = array()) {
$content = "<?php \n/* SVN FILE: $header$ */\n/* ". $className ." Test cases generated on: " . date('Y-m-d H:m:s') . " : ". time() . "*/\n{$out}?>";
return $this->createFile($path . $filename, $content);
}

/**
* outputs the a list of possible models or controllers from database
*
* @param string $useDbConfig Database configuration name
* @access public
*/
function listAll($useDbConfig = 'default', $interactive = true) {
function listAll($useDbConfig = null, $interactive = true) {
if (!isset($useDbConfig)) {
$useDbConfig = $this->connection;
}
$db =& ConnectionManager::getDataSource($useDbConfig);
$usePrefix = empty($db->config['prefix']) ? '' : $db->config['prefix'];
if ($usePrefix) {
Expand Down Expand Up @@ -779,7 +790,10 @@ function listAll($useDbConfig = 'default', $interactive = true) {
* @param string $useDbConfig Name of the database config you want to get tables from.
* @return void
**/
function getTable($modelName, $useDbConfig) {
function getTable($modelName, $useDbConfig = null) {
if (!isset($useDbConfig)) {
$useDbConfig = $this->connection;
}
$db =& ConnectionManager::getDataSource($useDbConfig);
$useTable = Inflector::tableize($modelName);
$fullTableName = $db->fullTableName($useTable, false);
Expand All @@ -801,7 +815,7 @@ function getTable($modelName, $useDbConfig) {
* @return string the model name
* @access public
*/
function getName($useDbConfig) {
function getName($useDbConfig = null) {
$this->listAll($useDbConfig);

$enteredModel = '';
Expand Down Expand Up @@ -848,6 +862,7 @@ function help() {
* @return null.
**/
function fixture($className, $useTable = null) {
$this->Fixture->connection = $this->connection;
$this->Fixture->bake($className, $useTable);
}
}
Expand Down
9 changes: 8 additions & 1 deletion cake/tests/cases/console/libs/tasks/model.test.php
View file
Expand Up @@ -63,6 +63,7 @@ class ModelTaskTest extends CakeTestCase {
* @var array
**/
var $fixtures = array('core.article', 'core.comment');

/**
* setUp method
*
Expand Down Expand Up @@ -92,12 +93,18 @@ function endTest() {
* @return void
**/
function testListAll() {
$this->Task->expectCallCount('out', 3);
$this->Task->expectAt(1, 'out', array('1. Article'));
$this->Task->expectAt(2, 'out', array('2. Comment'));
$result = $this->Task->listAll('test_suite');
$expected = array('articles', 'comments');
$this->assertEqual($result, $expected);

$this->Task->expectAt(4, 'out', array('1. Article'));
$this->Task->expectAt(5, 'out', array('2. Comment'));
$this->Task->connection = 'test_suite';
$result = $this->Task->listAll();
$expected = array('articles', 'comments');
$this->assertEqual($result, $expected);
}

/**
Expand Down

0 comments on commit bb2f6b2

Please sign in to comment.