fix: fix that withModuleSelection() dropped all fields after it dropp…

…ed one
AEB-labs · Aug 6, 2024 · 8efee14 · 8efee14
1 parent 08631db
commit 8efee14
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/spec/project/select-modules-in-sources.spec.ts b/spec/project/select-modules-in-sources.spec.ts
@@ -15,6 +15,7 @@ describe('selectModulesInProjectSource', () => {
                         two: String @modules(in: "module2")
                         extra1: String @modules(in: ["extra1", "module2 && extra2"])
                         extra2: String @modules(in: ["extra2"])
+                        one: String @modules(in: "module1")
                     }
 
                     type Two @rootEntity @modules(in: ["module2", "module3"]) {
@@ -35,6 +36,7 @@ describe('selectModulesInProjectSource', () => {
                         two: String @modules(in: ["module2"])
                         extra1: String @modules(in: ["extra1"])
                         
+                        one: String @modules(in: "module1")
                     }
 
                     type Two @rootEntity @modules(in: ["module2"]) {

diff --git a/src/project/select-modules-in-sources.ts b/src/project/select-modules-in-sources.ts
@@ -240,7 +240,7 @@ function selectModulesInGraphQLSource({
                         }
                         // remove field
                         changes.push({ location: fieldDef.loc });
-                        break;
+                        continue;
                     }
 
                     const moduleChange = changeModuleDirectiveIfNecessary(fieldDef, {