public void OverrideDefaultVariablesFromProvider()
        {
            // arrange
            var builder = new MigratorBuilder();

            var providerMock = new Mock <IDbProvider>();

            providerMock
            .Setup(x => x.GetDefaultVariables())
            .Returns(() => new Dictionary <string, string>
            {
                { DefaultVariables.User, ProviderUserName },
                { DefaultVariables.DbName, ProviderDbName }
            });
            var factoryMock = new Mock <IDbProviderFactory>();

            factoryMock
            .Setup(x => x.CreateDbProvider())
            .Returns(() => providerMock.Object);

            IReadOnlyDictionary <string, string> scriptVariables = null;
            var migrationsProviderMock = new Mock <IMigrationsProvider>();

            migrationsProviderMock
            .Setup(x => x.GetMigrations(
                       It.IsAny <IDbProvider>(),
                       It.IsAny <Dictionary <string, string> >(),
                       It.IsAny <ILogger>()))
            .Callback <IDbProvider, IReadOnlyDictionary <string, string>, ILogger>((provider, variables, logger) => { scriptVariables = variables; })
            .Returns <IDbProvider, IReadOnlyDictionary <string, string>, ILogger>((provider, variables, logger) => new List <IMigration>(0));

            builder.UseCustomMigrationsProvider(migrationsProviderMock.Object);
            builder.UserDbProviderFactory(factoryMock.Object);
            builder.UseVariable(DefaultVariables.User, ManualUserName);

            // act
            var migrator = builder.Build();

            // assert
            migrator.Should().NotBeNull("because we've create migrator");

            scriptVariables.Should().NotBeNull("because provider returns 2 variables");
            scriptVariables.Count.Should().Be(2, "because provider returns variable with user and db names");
            scriptVariables.ContainsKey(DefaultVariables.User).Should().BeTrue("because user name is default variable");
            scriptVariables[DefaultVariables.User].Should().BeEquivalentTo(ManualUserName, "because we set it manually to builder");
            scriptVariables.ContainsKey(DefaultVariables.DbName).Should().BeTrue("because db name is default variable");
            scriptVariables[DefaultVariables.DbName].Should().BeEquivalentTo(ProviderDbName, "because we set it manually to mock");
        }
Exemplo n.º 2
0
        /// <summary>
        /// Use provider to make migration on Postgre database.
        /// </summary>
        /// <param name="builder"></param>
        /// <param name="connectionString">Connection string to Postgre</param>
        /// <param name="migrationTableHistoryName">Migration history table name. If param <see langword="null"/> default value from DB will be used </param>
        /// <param name="databaseEncoding">Text presentation of database encoding for Postgre. If <see langword="null"/> default value from DB will be used </param>
        /// <param name="lcCollate"> String sort order for Postgre. If param <see langword="null"/> default value from DB will be used </param>
        /// <param name="lcCtype">Character classification for Postgre. If param <see langword="null"/> default value from DB will be used</param>
        /// <param name="connectionLimit">Limit of connections to Postgre. If param <see langword="null"/> default value from DB will be used </param>
        /// <param name="template">The name of the template from which to create the new database. If param <see langword="null"/> default value from DB will be used</param>
        /// <param name="tableSpace">The name of the tablespace that will be associated with the new database. If param <see langword="null"/> default value from DB will be used </param>
        /// <remarks>
        /// For detailed params description look at <see cref="PostgreDbProviderOptions"/>
        /// </remarks>
        // ReSharper disable once InconsistentNaming
        public static MigratorBuilder UsePostgreSQL(
            this MigratorBuilder builder,
            string connectionString,
            string migrationTableHistoryName = null,
            string databaseEncoding          = null,
            string lcCollate    = null,
            string lcCtype      = null,
            int?connectionLimit = null,
            string template     = null,
            string tableSpace   = null)
        {
            var options = new PostgreDbProviderOptions(
                connectionString,
                migrationTableHistoryName,
                databaseEncoding,
                lcCollate,
                lcCtype,
                connectionLimit,
                template,
                tableSpace);

            builder.UserDbProviderFactory(new PostgreDbProviderFactory(options));
            return(builder);
        }