started to work on command to set number of available custom variables

1 files changed, 32 insertions, 0 deletions

diff --git a/tests/PHPUnit/Integration/Core/DbTest.php b/tests/PHPUnit/Integration/Core/DbTest.php
new file mode 100644
index 0000000000..3a8c31e7e8
--- /dev/null
+++ b/tests/PHPUnit/Integration/Core/DbTest.php
@@ -0,0 +1,32 @@
+<?php
+/**
+ * Piwik - Open source web analytics
+ *
+ * @link http://piwik.org
+ * @license http://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
+ */
+use Piwik\Db;
+use Piwik\Common;
+
+/**
+ * Class Core_DbTest
+ *
+ * @group Core
+ */
+class Core_DbTest extends DatabaseTestCase
+{
+
+    public function test_getColumnNamesFromTable()
+    {
+        $this->assertColumnNames('access', array('login', 'idsite', 'access'));
+        $this->assertColumnNames('option', array('option_name', 'option_value', 'autoload'));
+    }
+
+    private function assertColumnNames($tableName, $expectedColumnNames)
+    {
+        $colmuns = Db::getColumnNamesFromTable(Common::prefixTable($tableName));
+
+        $this->assertEquals($expectedColumnNames, $colmuns);
+    }
+
+}
\ No newline at end of file