miniSql

创建
zgc123@gmail.com authored at 11/19/2023 1:40:15 AM
2.20 KiB
TemplateModel_tests.cs
#region License
// Copyright 2005-2019 Paul Kohler (https://github.com/paulkohler/minisqlquery). All rights reserved.
// This source code is made available under the terms of the GNU Lesser General Public License v3.0
// https://github.com/paulkohler/minisqlquery/blob/master/LICENSE
#endregion
using System;
using System.Collections.Generic;
using System.IO;
using MiniSqlQuery.Core;
using MiniSqlQuery.Core.Template;
using MiniSqlQuery.PlugIns.TemplateViewer;
using NUnit.Framework;
using Rhino.Mocks;

// ReSharper disable InconsistentNaming

namespace MiniSqlQuery.Tests.Templates
{
    [TestFixture]
    public class TemplateModel_tests
    {
        TemplateModel _model;
        IApplicationServices _services;
        IDatabaseInspector _databaseInspector;

        [SetUp]
        public void TestSetUp()
        {
            _services = MockRepository.GenerateStub<IApplicationServices>();
            _databaseInspector = MockRepository.GenerateStub<IDatabaseInspector>();
            _services.Expect(x => x.Resolve<TemplateHost>()).Return(new TemplateHost(_services, _databaseInspector));
            _model = new TemplateModel(_services, new NVelocityWrapper());
        }

        [Test]
        public void ModelData_parameters_are_precessed()
        {
            var items = new Dictionary<string, object>();
            string processedtext = _model.ProcessTemplate("create new ${Host.Date(\"yyyy\")}", items).Text;
            Assert.That(processedtext, Is.EqualTo("create new " + DateTime.Now.Year));
        }

        [Test]
        public void If_a_file_extension_is_set_this_defaults_the_TemplateResult()
        {
            var filename = Path.Combine(TestContext.CurrentContext.TestDirectory, @"Templates\foo.cs.mt");
            var result = _model.ProcessTemplateFile(filename, null);
            Assert.That(result.Extension, Is.EqualTo("cs"));
        }

        [Test]
        public void If_no_file_extension_is_set_this_defaults_the_TemplateResult_to_SQL()
        {
            var filename = Path.Combine(TestContext.CurrentContext.TestDirectory, @"Templates\bar.mt");
            var result = _model.ProcessTemplateFile(filename, null);
            Assert.That(result.Extension, Is.EqualTo("sql"));
        }
    }
}