From 92e4e81788b176e8a15ea4613f2a6d5ccd1f147f Mon Sep 17 00:00:00 2001
From: Taras Machyshyn <taras@espocrm.com>
Date: Thu, 4 Mar 2021 13:23:11 +0200
Subject: [PATCH] Test fixes

---
 .../Espo/Core/Utils/Database/HelperTest.php     | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/tests/integration/Espo/Core/Utils/Database/HelperTest.php b/tests/integration/Espo/Core/Utils/Database/HelperTest.php
index b2253b4846..043d6ffafb 100644
--- a/tests/integration/Espo/Core/Utils/Database/HelperTest.php
+++ b/tests/integration/Espo/Core/Utils/Database/HelperTest.php
@@ -196,7 +196,22 @@ class HelperTest extends \tests\integration\Core\BaseTestCase
     {
         $this->initTest();
 
-        $this->assertEquals('MySQL', $this->helper->getDatabaseType());
+        $databaseInfo = $this->getDatabaseInfo();
+        if (empty($databaseInfo)) {
+            return;
+        }
+
+        switch ($databaseInfo['type']) {
+            case 'mysql':
+                $this->assertEquals('MySQL', $this->helper->getDatabaseType());
+                break;
+
+            case 'mariadb':
+                $this->assertEquals('MariaDB', $this->helper->getDatabaseType());
+                break;
+        }
+
+        $this->assertEquals('MySQL', $this->helper->getDatabaseType('MySQL'));
         $this->assertEquals('MariaDB', $this->helper->getDatabaseType('MariaDB'));
     }
 }