add PgArrayToMany LazyLoding test #1145

2025-06-19 04:18:16 +08:00 · 2022-06-06 15:17:46 +08:00
parent a50c14e77a
commit b93a487e2f
4 changed files with 134 additions and 17 deletions
--- a/FreeSql.Tests/FreeSql.Tests/PostgreSQL/PostgreSQLPgArrayToManyTest
+++ b/FreeSql.Tests/FreeSql.Tests/PostgreSQL/PostgreSQLPgArrayToManyTest
@ -0,0 +1,120 @@
+using FreeSql.DataAnnotations;
+using System;
+using System.Collections.Generic;
+using System.Reflection;
+using System.Text;
+using Xunit;
+
+namespace FreeSql.Tests.PostgreSQL
+{
+    public class PostgreSQLPgArrayToManyTest
+    {
+
+        [Table(Name = "pgarray_tomany_user_lazyloading")]
+        public class UserLazyLoading
+        {
+            public int Id { get; set; }
+            public int[] RoleIds { get; set; }
+            public string UserName { get; set; }
+
+            [Navigate(nameof(RoleIds))]
+            public virtual List<RoleLazyLoading> Roles { get; set; }
+        }
+
+        [Table(Name = "pgarray_tomany_role_lazyloading")]
+        public class RoleLazyLoading
+        {
+            public int Id { get; set; }
+            public string RoleName { get; set; }
+
+            [Navigate(nameof(UserLazyLoading.RoleIds))]
+            public virtual List<UserLazyLoading> Users { get; set; }
+        }
+
+        [Fact]
+        public void LazyLoading()
+        {
+            var fsql = g.pgsql;
+            fsql.Delete<UserLazyLoading>().Where("1=1").ExecuteAffrows();
+            fsql.Delete<RoleLazyLoading>().Where("1=1").ExecuteAffrows();
+
+            var roles = new[]
+            {
+                new RoleLazyLoading { Id = 1, RoleName = "role1" },
+                new RoleLazyLoading { Id = 2, RoleName = "role2" },
+                new RoleLazyLoading { Id = 3, RoleName = "role3" }
+            };
+            Assert.Equal(3, fsql.Insert(roles).ExecuteAffrows());
+            var users = new[]
+            {
+                new UserLazyLoading { Id = 11, RoleIds = new [] { 1,2 }, UserName = "user1" },
+                new UserLazyLoading { Id = 12, RoleIds = new [] { 1,2,3 }, UserName = "user2" },
+                new UserLazyLoading { Id = 13, RoleIds = new [] { 1,3 }, UserName = "user3" },
+                new UserLazyLoading { Id = 14, RoleIds = new [] { 3,2,1 }, UserName = "user4" },
+                new UserLazyLoading { Id = 15, RoleIds = null, UserName = "user5" },
+                new UserLazyLoading { Id = 16, RoleIds = new int[0], UserName = "user6" },
+            };
+            Assert.Equal(6, fsql.Insert(users).ExecuteAffrows());
+
+            var role = fsql.Select<RoleLazyLoading>().Where(a => a.Id == 1).First();
+            Assert.IsNotType<RoleLazyLoading>(role);
+
+            var users2 = role.Users;
+            Assert.Equal(4, users2.Count);
+            Assert.Equal(11, users2[0].Id);
+            Assert.Equal("user1", users2[0].UserName);
+            Assert.Equal("1,2", string.Join(",", users2[0].RoleIds));
+            Assert.Equal(12, users2[1].Id);
+            Assert.Equal("user2", users2[1].UserName);
+            Assert.Equal("1,2,3", string.Join(",", users2[1].RoleIds));
+            Assert.Equal(13, users2[2].Id);
+            Assert.Equal("user3", users2[2].UserName);
+            Assert.Equal("1,3", string.Join(",", users2[2].RoleIds));
+            Assert.Equal(14, users2[3].Id);
+            Assert.Equal("user4", users2[3].UserName);
+            Assert.Equal("3,2,1", string.Join(",", users2[3].RoleIds));
+
+            var roles2 = users2[0].Roles;
+            Assert.Equal(2, roles2.Count);
+            Assert.Equal(1, roles2[0].Id);
+            Assert.Equal("role1", roles2[0].RoleName);
+            Assert.Equal(2, roles2[1].Id);
+            Assert.Equal("role2", roles2[1].RoleName);
+
+            roles2 = users2[1].Roles;
+            Assert.Equal(3, roles2.Count);
+            Assert.Equal(1, roles2[0].Id);
+            Assert.Equal("role1", roles2[0].RoleName);
+            Assert.Equal(2, roles2[1].Id);
+            Assert.Equal("role2", roles2[1].RoleName);
+            Assert.Equal(3, roles2[2].Id);
+            Assert.Equal("role3", roles2[2].RoleName);
+
+            roles2 = users2[2].Roles;
+            Assert.Equal(2, roles2.Count);
+            Assert.Equal(1, roles2[0].Id);
+            Assert.Equal("role1", roles2[0].RoleName);
+            Assert.Equal(3, roles2[1].Id);
+            Assert.Equal("role3", roles2[1].RoleName);
+
+            roles2 = users2[3].Roles;
+            Assert.Equal(3, roles2.Count);
+            Assert.Equal(3, roles2[0].Id);
+            Assert.Equal("role3", roles2[0].RoleName);
+            Assert.Equal(2, roles2[1].Id);
+            Assert.Equal("role2", roles2[1].RoleName);
+            Assert.Equal(1, roles2[2].Id);
+            Assert.Equal("role1", roles2[2].RoleName);
+
+            var user = fsql.Select<UserLazyLoading>().Where(a => a.Id == 11).First();
+            Assert.IsNotType<UserLazyLoading>(user);
+
+            roles2 = user.Roles;
+            Assert.Equal(2, roles2.Count);
+            Assert.Equal(1, roles2[0].Id);
+            Assert.Equal("role1", roles2[0].RoleName);
+            Assert.Equal(2, roles2[1].Id);
+            Assert.Equal("role2", roles2[1].RoleName);
+        }
+    }
+}