Tests: Add BillingTest.Test_01_NoActiveAreaCommitments()

Tests/HelperTests/BillingTest.cs

@@ -1,4 +1,5 @@
 using Elwig.Helpers;
+using Elwig.Helpers.Billing;
 using Microsoft.Data.Sqlite;
 using System.Reflection;
 
@@ -6,6 +7,9 @@ namespace Tests.HelperTests {
     [TestFixture]
     public class BillingTest {
 
+        private const int Year1 = 2020, Year2 = 2020;
+        private const int MgNr1 = 101, MgNr2 = 102, MgNr3 = 103, MgNr4 = 104;
+
         private SqliteConnection? Connection;
 
         [OneTimeSetUp]
@@ -22,9 +26,127 @@ namespace Tests.HelperTests {
             Connection = null;
         }
 
+        [TearDown]
+        public async Task CleanupDatabasePayment() {
+            if (Connection == null) return;
+            await AppDbContext.ExecuteBatch(Connection, """
+                DELETE FROM credit;
+                DELETE FROM delivery_part_bucket;
+                DELETE FROM payment_variant;
+                """);
+        }
+
+        private Task<Dictionary<string, AreaComBucket>> GetMemberAreaCommitmentBuckets(int year, int mgnr) {
+            var ctx = new AppDbContext();
+            return ctx.GetMemberAreaCommitmentBuckets(year, mgnr, Connection);
+        }
+
+        private Task<Dictionary<string, int>> GetMemberDeliveryBuckets(int year, int mgnr) {
+            var ctx = new AppDbContext();
+            return ctx.GetMemberDeliveryBuckets(year, mgnr, Connection);
+        }
+
+        private Task<Dictionary<string, int>> GetMemberPaymentBuckets(int year, int mgnr) {
+            var ctx = new AppDbContext();
+            return ctx.GetMemberPaymentBuckets(year, mgnr, Connection);
+        }
+
+        private async Task<Dictionary<(string, string), int>> GetMemberDeliveryPrices(int year, int mgnr) {
+            var buckets = new Dictionary<(string, string), int>();
+            using (var cmd = Connection!.CreateCommand()) {
+                cmd.CommandText = $"""
+                    SELECT lsnr || '/' ||  d.dpnr, d.sortid || b.discr, price
+                    FROM v_delivery d
+                        LEFT JOIN payment_delivery_part_bucket p ON (p.year, p.did, p.dpnr) = (d.year, d.did, d.dpnr)
+                        LEFT JOIN delivery_part_bucket b ON (b.year, b.did, b.dpnr, b.bktnr) = (p.year, p.did, p.dpnr, p.bktnr)
+                    WHERE d.year = {year} AND mgnr = {mgnr}
+                    """;
+                using var reader = await cmd.ExecuteReaderAsync();
+                while (await reader.ReadAsync()) {
+                    var lsnr = reader.GetString(0);
+                    var bucket = reader.GetString(1);
+                    buckets[(lsnr, bucket)] = reader.GetInt32(2);
+                }
+            }
+            return buckets;
+        }
+
+        private static string FormatBuckets(Dictionary<string, int> buckets) {
+            return string.Join("\r\n", buckets.OrderBy(b => b.Key));
+        }
+
+        private static string FormatBuckets(Dictionary<(string, string), int> buckets) {
+            return string.Join("\r\n", buckets.OrderBy(b => b.Key));
+        }
+
+        private static string FormatBuckets(Dictionary<string, AreaComBucket> buckets) {
+            return string.Join("\r\n", buckets.OrderBy(b => b.Key));
+        }
+
         [Test]
-        public void Test() {
-            // TODO
+        public async Task Test_01_NoActiveAreaCommitments() {
+            if (Connection == null) return;
+            var data = """
+                {
+                  "mode": "elwig",
+                  "version": 1,
+                  "payment": {
+                    "GV/": "curve:0",
+                    "GV/B": "curve:1",
+                    "GV/K": "curve:2"
+                  },
+                  "quality": {"WEI": 0.1},
+                  "curves": [{
+                    "id": 0,
+                    "mode": "oe",
+                    "data": {"15kmw": 0.5},
+                    "geb": 0.1
+                  }, {
+                    "id": 1,
+                    "mode": "oe",
+                    "data": {"15kmw": 0.54},
+                    "geb": 0.1
+                  }, {
+                    "id": 2,
+                    "mode": "oe",
+                    "data": {"15kmw": 0.61},
+                    "geb": 0.1
+                  }]
+                }
+                """;
+            await AppDbContext.ExecuteBatch(Connection, $"""
+                INSERT INTO payment_variant (year, avnr, name, date, transfer_date, test_variant, calc_time, data) VALUES
+                (2020, 1, 'Test', '2021-01-15', NULL, TRUE, NULL, '{data}');
+                """);
+
+            int mgnr = MgNr1, year = Year1;
+            var areaCom = await GetMemberAreaCommitmentBuckets(year, mgnr);
+            Assert.That(FormatBuckets(areaCom), Is.EqualTo(""));
+            var delivery = await GetMemberDeliveryBuckets(year, mgnr);
+            Assert.That(FormatBuckets(delivery), Is.EqualTo("""
+                [GV, 16000]
+                [GV_, 1000]
+                [GVB, 8000]
+                [GVK, 4000]
+                """));
+
+            BillingVariant b = new(year, 1);
+            await b.CalculateBuckets(false, false, false);
+            var payment = await GetMemberPaymentBuckets(year, mgnr);
+            Assert.That(FormatBuckets(payment), Is.EqualTo("""
+                [GV_, 17000]
+                """));
+
+            await b.Calculate();
+            var prices = await GetMemberDeliveryPrices(year, mgnr);
+            Assert.That(FormatBuckets(prices), Is.EqualTo("""
+                [(20201001X001/1, GV_), 6100]
+                [(20201001X001/2, GV_), 5000]
+                [(20201001X002/1, GV_), 5400]
+                [(20201001X002/2, GV_), 5400]
+                [(20201001X003/1, GV_), 1000]
+                [(20201001X003/2, GV_), 5000]
+                """));
         }
     }
 }