Merge pull request #12 from bendavies/sqlite-support

support SQLite for functional testing purposes

Author: fre5h

DBAL/Types/AbstractEnumType.php

@@ -11,6 +11,7 @@
 namespace Fresh\Bundle\DoctrineEnumBundle\DBAL\Types;
 
 use Doctrine\DBAL\Platforms\AbstractPlatform;
+use Doctrine\DBAL\Platforms\SqlitePlatform;
 use Doctrine\DBAL\Types\Type;
 
 /**
@@ -19,6 +20,7 @@
  * Provides support of MySQL ENUM type for Doctrine in Symfony applications
  *
  * @author Artem Genvald <genvaldartem@gmail.com>
+ * @author Ben Davies <ben.davies@gmail.com>
  */
 abstract class AbstractEnumType extends Type
 {
@@ -54,14 +56,18 @@ public function convertToDatabaseValue($value, AbstractPlatform $platform)
      */
     public function getSqlDeclaration(array $fieldDeclaration, AbstractPlatform $platform)
     {
-        $values = array_map(
+        $values = implode(', ', array_map(
             function ($value) {
                 return "'{$value}'";
             },
             $this->getValues()
-        );
+        ));
 
-        return 'ENUM(' . implode(', ', $values) . ')';
+        if ($platform instanceof SqlitePlatform) {
+            return sprintf('TEXT CHECK(%s IN (%s))', $fieldDeclaration['name'], $values);
+        }
+
+        return sprintf('ENUM(%s)', $values);
     }
 
     /**

Tests/DBAL/Types/AbstractEnumTypeTest.php

@@ -0,0 +1,71 @@
+<?php
+/*
+ * This file is part of the FreshDoctrineEnumBundle
+ *
+ * (c) Artem Genvald <genvaldartem@gmail.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+namespace Fresh\Bundle\DoctrineEnumBundle\Tests\DBAL\Types;
+
+use Doctrine\DBAL\Platforms\AbstractPlatform;
+use Doctrine\DBAL\Platforms\MySqlPlatform;
+use Doctrine\DBAL\Platforms\SqlitePlatform;
+
+/**
+ * AbstractEnumTypeTest
+ *
+ * @author Ben Davies <ben.davies@gmail.com>
+ *
+ * @coversDefaultClass \Fresh\Bundle\DoctrineEnumBundle\DBAL\Types\AbstractEnumType
+ */
+class AbstractEnumTypeTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @var \Fresh\Bundle\DoctrineEnumBundle\DBAL\Types\AbstractEnumType
+     */
+    private $type;
+
+    /**
+     * Set up EnumType
+     */
+    public function setUp()
+    {
+        $this->type = $this->getMockBuilder('Fresh\Bundle\DoctrineEnumBundle\DBAL\Types\AbstractEnumType')
+            ->disableOriginalConstructor()
+            ->setMethods(array('getValues'))
+            ->getMockForAbstractClass();
+
+        $this->type->staticExpects($this->any())
+            ->method('getValues')
+            ->will($this->returnValue(array('M', 'F')));
+    }
+
+    /**
+     * Test that the SQL declaration is the correct for the platform
+     *
+     * @param array            $fieldDeclaration The Field Declaration
+     * @param AbstractPlatform $platform         The DBAL Platform
+     * @param string           $expected         Expected Sql Declaration
+     *
+     * @test
+     * @covers ::getSqlDeclaration
+     * @dataProvider platformProvider
+     */
+    public function getSqlDeclaration(array $fieldDeclaration, AbstractPlatform $platform, $expected)
+    {
+        $this->assertEquals($expected, $this->type->getSqlDeclaration($fieldDeclaration, $platform));
+    }
+
+    /**
+     * @return array
+     */
+    public function platformProvider()
+    {
+        return array(
+            array(array('name' => 'sex'), new MySqlPlatform(), "ENUM('M', 'F')"),
+            array(array('name' => 'sex'), new SqlitePlatform(), "TEXT CHECK(sex IN ('M', 'F'))")
+        );
+    }
+}