vendor/doctrine/dbal/tests/Doctrine/Tests/DBAL/Sharding/SQLAzure/MultiTenantVisitorTest.php

   1 <?php
   2 /*
   3  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   4  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   5  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   6  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
   7  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   8  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   9  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  10  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  11  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  12  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  13  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  14  *
  15  * This software consists of voluntary contributions made by many individuals
  16  * and is licensed under the LGPL. For more information, see
  17  * <http://www.doctrine-project.org>.
  18  */
  19
  20 namespace Doctrine\Tests\DBAL\Sharding\SQLAzure;
  21
  22 use Doctrine\DBAL\Platforms\SQLAzurePlatform;
  23 use Doctrine\DBAL\Schema\Schema;
  24 use Doctrine\DBAL\Sharding\SQLAzure\Schema\MultiTenantVisitor;
  25
  26 class MultiTenantVisitorTest extends \PHPUnit_Framework_TestCase
  27 {
  28     public function testMultiTenantPrimaryKey()
  29     {
  30         $platform = new SQLAzurePlatform();
  31         $visitor = new MultiTenantVisitor();
  32
  33         $schema = new Schema();
  34         $foo = $schema->createTable('foo');
  35         $foo->addColumn('id', 'string');
  36         $foo->setPrimaryKey(array('id'));
  37         $schema->visit($visitor);
  38
  39         $this->assertEquals(array('id', 'tenant_id'), $foo->getPrimaryKey()->getColumns());
  40         $this->assertTrue($foo->hasColumn('tenant_id'));
  41     }
  42
  43     public function testMultiTenantNonPrimaryKey()
  44     {
  45         $platform = new SQLAzurePlatform();
  46         $visitor = new MultiTenantVisitor();
  47
  48         $schema = new Schema();
  49         $foo = $schema->createTable('foo');
  50         $foo->addColumn('id', 'string');
  51         $foo->addColumn('created', 'datetime');
  52         $foo->setPrimaryKey(array('id'));
  53         $foo->addIndex(array('created'), 'idx');
  54
  55         $foo->getPrimaryKey()->addFlag('nonclustered');
  56         $foo->getIndex('idx')->addFlag('clustered');
  57
  58         $schema->visit($visitor);
  59
  60         $this->assertEquals(array('id'), $foo->getPrimaryKey()->getColumns());
  61         $this->assertTrue($foo->hasColumn('tenant_id'));
  62         $this->assertEquals(array('created', 'tenant_id'), $foo->getIndex('idx')->getColumns());
  63     }
  64 }
  65