From 90609b7f5deec07ea575177f4da0d41f18131aec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maxim=20Darr=C3=A9?=
 <33755271+maxmousse@users.noreply.github.com>
Date: Mon, 19 Dec 2022 17:27:16 -0500
Subject: [PATCH] fix: update unit-test to handle ownership system fields
 (#540)

---
 hapify-models.json                            |  2 +-
 .../__kebab__.controller.spec.ts.hpf          | 24 ++++++++++++++-----
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/hapify-models.json b/hapify-models.json
index 77fd96b6e..b2eb9cfaa 100644
--- a/hapify-models.json
+++ b/hapify-models.json
@@ -28,7 +28,7 @@
             "onUpdate": "Cascade",
             "ownerStringPath": "user.id"
           },
-          "name": "user",
+          "name": "owner",
           "notes": "User that owns the answer",
           "properties": ["internal", "ownership"],
           "subtype": "oneMany",
diff --git a/libs/hapify/templates/nestjs-models-rest/hapify/generated/nestjs-models-rest/src/__kebab__/controllers/__kebab__.controller.spec.ts.hpf b/libs/hapify/templates/nestjs-models-rest/hapify/generated/nestjs-models-rest/src/__kebab__/controllers/__kebab__.controller.spec.ts.hpf
index 9a2229e0e..eb8119d83 100644
--- a/libs/hapify/templates/nestjs-models-rest/hapify/generated/nestjs-models-rest/src/__kebab__/controllers/__kebab__.controller.spec.ts.hpf
+++ b/libs/hapify/templates/nestjs-models-rest/hapify/generated/nestjs-models-rest/src/__kebab__/controllers/__kebab__.controller.spec.ts.hpf
@@ -139,7 +139,9 @@ describe('<<Model pascal>>Controller', () => {
       <<< if (root.names.pascal !== 'User') { >>>const mockUser = mockUserFactory();<<< }>>>
       mockReqUser.mockReturnValue(mockUser);
       (mock<<Model pascal>>Service.getDefaultOwnerships as jest.Mock).mockReturnValue({
-        user: { connect: mockUser },
+        <<for Fields ownership and not primary and internal field>>
+          <<field camel>>: { connect: mockUser },
+        <<endfor>>
       });
       <<endif>>
 
@@ -182,7 +184,9 @@ describe('<<Model pascal>>Controller', () => {
       <<< if (root.names.pascal !== 'User') { >>>const mockUser = mockUserFactory();<<< }>>>
       mockReqUser.mockReturnValue(mockUser);
       (mock<<Model pascal>>Service.getDefaultOwnerships as jest.Mock).mockReturnValue({
-        user: { connect: mockUser },
+        <<for Fields ownership and not primary and internal field>>
+          <<field camel>>: { connect: mockUser },
+        <<endfor>>
       });
       <<endif>>
 
@@ -706,7 +710,9 @@ describe('<<Model pascal>>Controller', () => {
       <<< if (root.names.pascal !== 'User') { >>>const mockUser = mockUserFactory();<<< }>>>
       mockReqUser.mockReturnValue(mockUser);
       (mock<<Model pascal>>Service.getDefaultOwnerships as jest.Mock).mockReturnValue({
-        user: { connect: mockUser },
+        <<for Fields ownership and not primary and internal field>>
+          <<field camel>>: { connect: mockUser },
+        <<endfor>>
       });
 
       const { can, build } = new AbilityBuilder(AppAbility);
@@ -772,7 +778,9 @@ describe('<<Model pascal>>Controller', () => {
       <<< if (root.names.pascal !== 'User') { >>>const mockUser = mockUserFactory();<<< }>>>
       mockReqUser.mockReturnValue(mockUser);
       (mock<<Model pascal>>Service.getDefaultOwnerships as jest.Mock).mockReturnValue({
-        user: { connect: mockUser },
+        <<for Fields ownership and not primary and internal field>>
+          <<field camel>>: { connect: mockUser },
+        <<endfor>>
       });
 
       const { can, build } = new AbilityBuilder(AppAbility);
@@ -838,7 +846,9 @@ describe('<<Model pascal>>Controller', () => {
       <<< if (root.names.pascal !== 'User') { >>>const mockUser = mockUserFactory();<<< }>>>
       mockReqUser.mockReturnValue(mockUser);
       (mock<<Model pascal>>Service.getDefaultOwnerships as jest.Mock).mockReturnValue({
-        user: { connect: mockUser },
+        <<for Fields ownership and not primary and internal field>>
+          <<field camel>>: { connect: mockUser },
+        <<endfor>>
       });
 
       const { can, build } = new AbilityBuilder(AppAbility);
@@ -904,7 +914,9 @@ describe('<<Model pascal>>Controller', () => {
       <<< if (root.names.pascal !== 'User') { >>>const mockUser = mockUserFactory();<<< }>>>
       mockReqUser.mockReturnValue(mockUser);
       (mock<<Model pascal>>Service.getDefaultOwnerships as jest.Mock).mockReturnValue({
-        user: { connect: mockUser },
+        <<for Fields ownership and not primary and internal field>>
+          <<field camel>>: { connect: mockUser },
+        <<endfor>>
       });
 
       const { can, build } = new AbilityBuilder(AppAbility);