yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-09-09 08:39:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

use dropTableIfExists

Browse Source
This commit is contained in:
Pierre Noël 2021-03-29 10:11:19 +02:00
parent 2149493137
commit f3500ecea5
2 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
packages/strapi-plugin-i18n/config/functions/migrations/__tests__/content-type.test.js
View File

@ -145,8 +145,7 @@ describe('i18n - Migration - enable/disable localization on a CT', () => {
const definition = { client: 'pg' }; const definition = { client: 'pg' };
const defaultLocaleRows = [{ value: '"fr"' }]; const defaultLocaleRows = [{ value: '"fr"' }];
const deleteRelations = jest.fn(); const deleteRelations = jest.fn();
const hasTable = jest.fn(() => Promise.resolve(true)); const dropTableIfExists = jest.fn();
const dropTable = jest.fn();
const table = jest.fn(); const table = jest.fn();
const model = { orm: 'bookshelf', deleteRelations }; const model = { orm: 'bookshelf', deleteRelations };
const where = jest.fn(() => Promise.resolve(defaultLocaleRows)); const where = jest.fn(() => Promise.resolve(defaultLocaleRows));
@ -162,7 +161,7 @@ describe('i18n - Migration - enable/disable localization on a CT', () => {
trx.offset = jest.fn(() => trx); trx.offset = jest.fn(() => trx);
trx.limit = jest.fn(() => Promise.resolve([{ id: 1 }, { id: 2 }])); trx.limit = jest.fn(() => Promise.resolve([{ id: 1 }, { id: 2 }]));
const transaction = jest.fn(() => Promise.resolve(trx)); const transaction = jest.fn(() => Promise.resolve(trx));
const knex = { where, transaction, schema: { hasTable, dropTable, table } }; const knex = { where, transaction, schema: { dropTableIfExists, table } };
knex.select = jest.fn(() => knex); knex.select = jest.fn(() => knex);
knex.from = jest.fn(() => knex); knex.from = jest.fn(() => knex);
const ORM = { knex }; const ORM = { knex };
@ -177,7 +176,7 @@ describe('i18n - Migration - enable/disable localization on a CT', () => {
expect(transaction).toHaveBeenCalled(); expect(transaction).toHaveBeenCalled();
expect(trx.commit).toHaveBeenCalled(); expect(trx.commit).toHaveBeenCalled();
expect(table).toHaveBeenCalled(); expect(table).toHaveBeenCalled();
expect(dropTable).toHaveBeenCalledWith('countries__localizations'); expect(dropTableIfExists).toHaveBeenCalledWith('countries__localizations');
}); });
test('i18n => non i18n - sqlite', async () => { test('i18n => non i18n - sqlite', async () => {
const previousDefinition = { const previousDefinition = {
@ -187,8 +186,7 @@ describe('i18n - Migration - enable/disable localization on a CT', () => {
const definition = { client: 'sqlite3' }; const definition = { client: 'sqlite3' };
const defaultLocaleRows = [{ value: '"fr"' }]; const defaultLocaleRows = [{ value: '"fr"' }];
const deleteRelations = jest.fn(); const deleteRelations = jest.fn();
const hasTable = jest.fn(() => Promise.resolve(true)); const dropTableIfExists = jest.fn();
const dropTable = jest.fn();
const table = jest.fn(); const table = jest.fn();
const model = { orm: 'bookshelf', deleteRelations }; const model = { orm: 'bookshelf', deleteRelations };
const where = jest.fn(() => Promise.resolve(defaultLocaleRows)); const where = jest.fn(() => Promise.resolve(defaultLocaleRows));
@ -204,7 +202,7 @@ describe('i18n - Migration - enable/disable localization on a CT', () => {
trx.offset = jest.fn(() => trx); trx.offset = jest.fn(() => trx);
trx.limit = jest.fn(() => Promise.resolve([{ id: 1 }, { id: 2 }])); trx.limit = jest.fn(() => Promise.resolve([{ id: 1 }, { id: 2 }]));
const transaction = jest.fn(() => Promise.resolve(trx)); const transaction = jest.fn(() => Promise.resolve(trx));
const knex = { where, transaction, schema: { hasTable, dropTable, table } }; const knex = { where, transaction, schema: { dropTableIfExists, table } };
knex.select = jest.fn(() => knex); knex.select = jest.fn(() => knex);
knex.from = jest.fn(() => knex); knex.from = jest.fn(() => knex);
const ORM = { knex }; const ORM = { knex };
@ -221,7 +219,7 @@ describe('i18n - Migration - enable/disable localization on a CT', () => {
expect(trx.commit).toHaveBeenCalled(); expect(trx.commit).toHaveBeenCalled();
expect(table).not.toHaveBeenCalled(); expect(table).not.toHaveBeenCalled();
expect(context).toEqual({ recreateSqliteTable: true }); expect(context).toEqual({ recreateSqliteTable: true });
expect(dropTable).toHaveBeenCalledWith('countries__localizations'); expect(dropTableIfExists).toHaveBeenCalledWith('countries__localizations');
}); });
}); });
describe('mongoose', () => { describe('mongoose', () => {

5
packages/strapi-plugin-i18n/config/functions/migrations/content-type/disable/migrate-for-bookshelf.js
View File

@ -52,10 +52,7 @@ const migrateForBookshelf = async (
}); });
} }
const tableExists = await ORM.knex.schema.hasTable(localizationsTable); await ORM.knex.schema.dropTableIfExists(localizationsTable);
if (tableExists) {
await ORM.knex.schema.dropTable(localizationsTable);
}
}; };
module.exports = migrateForBookshelf; module.exports = migrateForBookshelf;