ContextBasedTest.cs

Go to the documentation of this file.

// 
//  Copyright 2010-2014 Deveel
// 
//    Licensed under the Apache License, Version 2.0 (the "License");
//    you may not use this file except in compliance with the License.
//    You may obtain a copy of the License at
// 
//        http://www.apache.org/licenses/LICENSE-2.0
// 
//    Unless required by applicable law or agreed to in writing, software
//    distributed under the License is distributed on an "AS IS" BASIS,
//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//    See the License for the specific language governing permissions and
//    limitations under the License.
using System;

using Deveel.Data.Configuration;
using Deveel.Data.Services;
using Deveel.Data.Transactions;

using NUnit.Framework;

namespace Deveel.Data {
    [TestFixture]
    public abstract class ContextBasedTest {
        protected const string AdminUserName = "SA";
        protected const string AdminPassword = "1234567890";
        protected const string DatabaseName = "testdb";

        protected virtual bool SingleContext {
            get { return false; }
        }

        protected IQuery Query { get; private set; }

        protected ISystem System { get; private set; }

        protected IDatabase Database { get; private set; }

        protected ISession Session { get; private set; }

        protected virtual void RegisterServices(ServiceContainer container) {
        }

        protected virtual ISystem CreateSystem() {
            var builder = new TestSystemBuilder(this);
            return builder.BuildSystem();
        }

        protected virtual IDatabase CreateDatabase(ISystem system, IConfiguration configuration) {
            return system.CreateDatabase(configuration, AdminUserName, AdminPassword);
        }

        protected virtual ISession CreateAdminSession(IDatabase database) {
            var user = database.Authenticate(AdminUserName, AdminPassword);
            var transaction = database.CreateTransaction(IsolationLevel.Serializable);
            return new Session(transaction, user);
        }

        protected virtual IQuery CreateQuery(ISession session) {
            return session.CreateQuery();
        }

        protected ISession CreateUserSession(string userName, string password) {
            return Database.CreateUserSession(userName, password);
        }

        protected virtual void OnSetUp(string testName) {
            
        }

        protected virtual void OnTearDown() {

        }

        [SetUp]
        public void TestSetUp() {
            if (!SingleContext)
                CreateContext();

            var testName = TestContext.CurrentContext.Test.Name;
            OnSetUp(testName);
        }

        [TestFixtureSetUp]
        public void TestFixtureSetUp() {
            if (SingleContext)
                CreateContext();
        }

        private void CreateContext() {
            System = CreateSystem();

            var dbConfig = new Configuration.Configuration();
            dbConfig.SetValue("database.name", DatabaseName);

            Database = CreateDatabase(System, dbConfig);
            Session = CreateAdminSession(Database);
            Query = CreateQuery(Session);
        }

        private void DisposeContext() {
            if (Query != null)
                Query.Dispose();

            if (Database != null)
                Database.Dispose();

            if (System != null)
                System.Dispose();

            Query = null;
            Database = null;
            Database = null;
            System = null;
        }

        [TearDown]
        public void TestTearDown() {
            OnTearDown();

            if (!SingleContext)
                DisposeContext();
        }

        [TestFixtureTearDown]
        public void TestFixtureTearDown() {
            if (SingleContext)
                DisposeContext();
        }

        private class TestSystemBuilder : SystemBuilder {
            private ContextBasedTest test;

            public TestSystemBuilder(ContextBasedTest test) {
                this.test = test;
            }

            protected override void OnServiceRegistration(ServiceContainer container) {
                test.RegisterServices(container);
            }
        }
    }
}