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.
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>.
20 namespace Doctrine\Tests\DBAL\Sharding\SQLAzure;
22 use Doctrine\DBAL\Platforms\SQLAzurePlatform;
23 use Doctrine\DBAL\Schema\Schema;
24 use Doctrine\DBAL\Sharding\SQLAzure\Schema\MultiTenantVisitor;
26 class MultiTenantVisitorTest extends \PHPUnit_Framework_TestCase
28 public function testMultiTenantPrimaryKey()
30 $platform = new SQLAzurePlatform();
31 $visitor = new MultiTenantVisitor();
33 $schema = new Schema();
34 $foo = $schema->createTable('foo');
35 $foo->addColumn('id', 'string');
36 $foo->setPrimaryKey(array('id'));
37 $schema->visit($visitor);
39 $this->assertEquals(array('id', 'tenant_id'), $foo->getPrimaryKey()->getColumns());
40 $this->assertTrue($foo->hasColumn('tenant_id'));
43 public function testMultiTenantNonPrimaryKey()
45 $platform = new SQLAzurePlatform();
46 $visitor = new MultiTenantVisitor();
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');
55 $foo->getPrimaryKey()->addFlag('nonclustered');
56 $foo->getIndex('idx')->addFlag('clustered');
58 $schema->visit($visitor);
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());