dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

test/Npgsql.Tests/PgPassFileTests.cs

58 строк2 KB

DoraemonDesu

Add PgPassFile connection string parameter

25 июл 2017, 15:38

25 июл 2017, 15:387282bbc

using System;
using System.IO;
using System.Linq;
using NUnit.Framework;

namespace Npgsql.Tests
{
    [TestFixture]
    public class PgPassFileTests
    {
        [Test]
        public void ShouldParseAllEntries()
        {
            var file = PgPassFile.Load(_pgpassFile);
            var entries = file.Entries.ToList();
            Assert.That(entries.Count, Is.EqualTo(3));
        }

        [Test]
        public void ShouldFindFirstEntryWhenMultipleMatch()
        {
            var file = PgPassFile.Load(_pgpassFile);
            var entry = file.GetFirstMatchingEntry("testhost");
            Assert.That(entry.Password, Is.EqualTo("testpass"));
        }

        [Test]
        public void ShouldFindDefaultForNoMatches()
        {
            var file = PgPassFile.Load(_pgpassFile);
            var entry = file.GetFirstMatchingEntry("notarealhost");
            Assert.That(entry.Password, Is.EqualTo("defaultpass"));
        }

        string _pgpassFile;

        [OneTimeSetUp]
        public void CreateTestFile()
        {
            // set up pgpass file with fake content that can be used for this test
            const string content = @"testhost:1234:testdatabase:testuser:testpass
testhost:*:*:*:testdefaultpass
# helpful comment goes here
*:*:*:*:defaultpass";

            _pgpassFile = Path.GetTempFileName();

            File.WriteAllText(_pgpassFile, content);
        }

        [OneTimeTearDown]
        public void DeleteTestFile()
        {
            if (File.Exists(_pgpassFile))
                File.Delete(_pgpassFile);
        }
    }
}