fix: update association with foreign key (#5756)

This commit is contained in:
ChengLei Shao 2024-11-30 08:59:07 +08:00 committed by GitHub
parent bff82856e7
commit 9ddb4c179e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 51 additions and 1 deletions

View File

@ -66,6 +66,48 @@ describe('update associations', () => {
expect(profile1['userId']).toBe(user.id);
});
it('should update has many association with foreign key', async () => {
const User = db.collection({
name: 'users',
fields: [
{ type: 'string', name: 'name' },
{ type: 'hasMany', name: 'profiles', foreignKey: 'userId', target: 'profiles' },
],
});
const Profile = db.collection({
name: 'profiles',
fields: [
{ type: 'string', name: 'name' },
{ type: 'bigInt', name: 'userId' },
{ type: 'belongsTo', name: 'user', foreignKey: 'userId' },
],
});
await db.sync();
// create user
const user = await User.repository.create({ values: { name: 'user1' } });
const profile = await Profile.repository.create({ values: { name: 'profile1' } });
const profileData = profile.toJSON();
await User.repository.update({
filterByTk: user.id,
values: {
profiles: [
{
...profileData,
userId: null,
},
],
},
updateAssociationValues: ['profiles'],
});
const profile1 = await Profile.repository.findOne({ filterByTk: profile.id });
expect(profile1['userId']).toBe(user.id);
});
test('update belongs to with foreign key and object', async () => {
const throughAB = db.collection({
name: 'throughAB',

View File

@ -351,7 +351,11 @@ export async function updateSingleAssociation(
if (updateAssociationValues.includes(key)) {
const updateValues = { ...value };
if (association.associationType === 'HasOne') {
delete updateValues[association.foreignKey];
}
await instance.update(updateValues, { ...options, transaction });
}
@ -536,6 +540,10 @@ export async function updateMultipleAssociation(
continue;
}
if (updateAssociationValues.includes(key)) {
if (association.associationType === 'HasMany') {
delete item[association.foreignKey];
}
await instance.update(item, { ...options, transaction });
}
await updateAssociations(instance, item, {