fix mock db tests for multi-org prebuilds

SasSwart · SasSwart · commit 05c28d141473 · 2025-05-27T14:58:50.000Z
diff --git a/enterprise/coderd/prebuilds/membership_test.go b/enterprise/coderd/prebuilds/membership_test.go
@@ -57,6 +57,14 @@ func TestReconcileAll(t *testing.T) {
 			backgroundOrg := dbgen.Organization(t, db, database.Organization{})
 			targetOrg := dbgen.Organization(t, db, database.Organization{})
 
+			if !dbtestutil.WillUsePostgres() {
+				// dbmem doesn't ensure membership to the default organization
+				dbgen.OrganizationMember(t, db, database.OrganizationMember{
+					OrganizationID: defaultOrg.ID,
+					UserID:         agplprebuilds.SystemUserID,
+				})
+			}
+
 			dbgen.OrganizationMember(t, db, database.OrganizationMember{OrganizationID: backgroundOrg.ID, UserID: agplprebuilds.SystemUserID})
 			if tc.preExistingMembership {
 				// System user already a member of both orgs.
