// Copyright (C) Stichting Deltares 2016. All rights reserved.
//
// This file is part of Ringtoets.
//
// Ringtoets is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// All names, logos, and references to "Deltares" are registered trademarks of
// Stichting Deltares and remain full property of Stichting Deltares at all times.
// All rights reserved.

using System;
using System.Data;
using System.Data.SQLite;
using System.IO;
using Core.Common.IO.Readers;
using Core.Common.TestUtil;
using Migration.Scripts.Data;
using Migration.Scripts.Data.Exceptions;
using NUnit.Framework;
using Rhino.Mocks;

namespace Application.Ringtoets.Migration.Test
{
    [TestFixture]
    public class RingtoetsUpgradeScriptTest
    {
        [Test]
        [TestCase("")]
        [TestCase(null)]
        public void Constructor_InvalidFromVersion_ThrowsArgumentException(string fromVersion)
        {
            // Setup
            const string query = "Valid query";
            const string toVersion = "toVersion";

            // Call
            TestDelegate call = () => new RingtoetsUpgradeScript(fromVersion, toVersion, query);

            // Assert
            string paramName = Assert.Throws<ArgumentException>(call).ParamName;
            Assert.AreEqual("fromVersion", paramName);
        }

        [Test]
        [TestCase("")]
        [TestCase(null)]
        public void Constructor_InvalidToVersion_ThrowsArgumentException(string toVersion)
        {
            // Setup
            const string fromVersion = "fromVersion";
            const string query = "Valid query";

            // Call
            TestDelegate call = () => new RingtoetsUpgradeScript(fromVersion, toVersion, query);

            // Assert
            string paramName = Assert.Throws<ArgumentException>(call).ParamName;
            Assert.AreEqual("toVersion", paramName);
        }

        [Test]
        [TestCase("")]
        [TestCase("   ")]
        [TestCase(null)]
        public void Constructor_InvalidQuery_ThrowsArgumentException(string query)
        {
            // Setup
            const string fromVersion = "fromVersion";
            const string toVersion = "toVersion";

            // Call
            TestDelegate call = () => new RingtoetsUpgradeScript(fromVersion, toVersion, query);

            // Assert
            string paramName = Assert.Throws<ArgumentException>(call).ParamName;
            Assert.AreEqual("query", paramName);
        }

        [Test]
        public void Constructor_ValidParameters_ReturnsExpectedValues()
        {
            // Setup
            const string fromVersion = "fromVersion";
            const string toVersion = "toVersion";
            const string query = ";";

            // Call
            var upgradeScript = new RingtoetsUpgradeScript(fromVersion, toVersion, query);

            // Assert
            Assert.IsInstanceOf<UpgradeScript>(upgradeScript);
            Assert.AreEqual(fromVersion, upgradeScript.FromVersion());
            Assert.AreEqual(toVersion, upgradeScript.ToVersion());
        }

        [Test]
        public void Upgrade_ActualQueries_ExpectedProperties()
        {
            // Setup
            var createScript = new RingtoetsCreateScript("2", "CREATE TABLE 'MigrationScript' ('Field' TEXT NOT NULL);");
            var upgradeScript = new RingtoetsUpgradeScript("1", "2", "INSERT INTO 'MigrationScript' SELECT '{0}';");
            var migrationScript = new FileMigrationScript(createScript, upgradeScript);
            var versionedFile = new RingtoetsVersionedFile("c:\\file.ext");

            // Call
            IVersionedFile upgradedFile = migrationScript.Upgrade(versionedFile);

            // Assert
            Assert.IsNotNull(upgradedFile);

            Assert.IsTrue(File.Exists(upgradedFile.Location));
            using (var msdr = new MigrationScriptDatabaseReader(upgradedFile.Location))
            {
                Assert.IsTrue(msdr.IsValueInserted(versionedFile.Location));
            }
            using (new FileDisposeHelper(upgradedFile.Location)) {}
        }

        [Test]
        public void Upgrade_UpgradeFails_ThrowsCriticalMigrationException()
        {
            // Setup
            string filename = Path.GetRandomFileName();
            string filePath = TestHelper.GetTestDataPath(TestDataPath.Application.Ringtoets.Migration, filename);

            var sourceVersionedFile = new RingtoetsVersionedFile("c:\\file.ext");
            var targetVersionedFile = new RingtoetsVersionedFile("c:\\file.ext");
            var upgradeScript = new RingtoetsUpgradeScript("1", "2", "THIS WILL FAIL");

            // Call
            TestDelegate call = () => upgradeScript.Upgrade(sourceVersionedFile, targetVersionedFile);

            // Assert
            using (new FileDisposeHelper(filePath))
            {
                CriticalMigrationException exception = Assert.Throws<CriticalMigrationException>(call);
                Assert.AreEqual("Het upgraden van het Ringtoets bestand versie 1 naar 2 is mislukt.",
                                exception.Message);
                Assert.IsInstanceOf<SQLiteException>(exception.InnerException);
            }
        }

        [Test]
        public void Upgrade_ValidParameters_ExpectedProperties()
        {
            // Setup
            string filename = Path.GetRandomFileName();
            string filePath = TestHelper.GetTestDataPath(TestDataPath.Application.Ringtoets.Migration, filename);

            var mockRepository = new MockRepository();
            var sourceVersionedFile = mockRepository.Stub<IVersionedFile>();
            sourceVersionedFile.Expect(tvf => tvf.Location).Return("c:\\file.ext");
            var targetVersionedFile = mockRepository.Stub<IVersionedFile>();
            targetVersionedFile.Expect(tvf => tvf.Location).Return(filePath);
            mockRepository.ReplayAll();

            var upgradeScript = new RingtoetsUpgradeScript("1", "2", ";");

            // Call
            upgradeScript.Upgrade(sourceVersionedFile, targetVersionedFile);

            // Assert
            Assert.IsNotNull(targetVersionedFile);
            using (new FileDisposeHelper(filePath))
            {
                Assert.IsTrue(File.Exists(filePath));
            }
            mockRepository.VerifyAll();
        }

        private class MigrationScriptDatabaseReader : SqLiteDatabaseReaderBase
        {
            public MigrationScriptDatabaseReader(string filePath) : base(filePath) {}

            public bool IsValueInserted(string value)
            {
                const string query = "SELECT FIELD FROM 'MigrationScript';";
                try
                {
                    using (IDataReader dataReader = CreateDataReader(query, null))
                    {
                        if (!dataReader.Read())
                        {
                            return false;
                        }

                        return dataReader.Read<string>("Field").Equals(value);
                    }
                }
                catch
                {
                    return false;
                }
            }
        }
    }
}