mirror of https://github.com/nocodb/nocodb
Muhammed Mustafa
2 years ago
4 changed files with 172 additions and 58 deletions
@ -1,55 +0,0 @@
|
||||
import knex from 'knex'; |
||||
import fs from 'fs'; |
||||
import Project from '../../../models/Project'; |
||||
import Audit from '../../../models/Audit'; |
||||
|
||||
const config = { |
||||
client: 'mysql2', |
||||
connection: { |
||||
host: 'localhost', |
||||
port: 3306, |
||||
user: 'root', |
||||
password: 'password', |
||||
database: 'test_sakila', |
||||
multipleStatements: true, |
||||
dateStrings: true, |
||||
}, |
||||
}; |
||||
|
||||
const isMysqlSakilaToBeReset = async () => { |
||||
const sakilaProject = await Project.getByTitle('externalREST'); |
||||
|
||||
const audits = |
||||
sakilaProject && (await Audit.projectAuditList(sakilaProject.id, {})); |
||||
|
||||
return audits?.length > 0; |
||||
}; |
||||
|
||||
const resetMysqlSakila = async () => { |
||||
const knexClient = knex(config); |
||||
|
||||
try { |
||||
await knexClient.raw(`DROP DATABASE test_sakila`); |
||||
} catch (e) { |
||||
console.log('Error dropping db', e); |
||||
} |
||||
await knexClient.raw(`CREATE DATABASE test_sakila`); |
||||
|
||||
const testsDir = __dirname.replace( |
||||
'/src/lib/services/test/TestResetService', |
||||
'/tests' |
||||
); |
||||
|
||||
const schemaFile = fs |
||||
.readFileSync(`${testsDir}/mysql-sakila-db/03-test-sakila-schema.sql`) |
||||
.toString(); |
||||
const dataFile = fs |
||||
.readFileSync(`${testsDir}/mysql-sakila-db/04-test-sakila-data.sql`) |
||||
.toString(); |
||||
await knexClient.raw(schemaFile); |
||||
await knexClient.raw(dataFile); |
||||
|
||||
await knexClient.destroy(); |
||||
}; |
||||
|
||||
export { resetMysqlSakila, isMysqlSakilaToBeReset }; |
@ -0,0 +1,165 @@
|
||||
import axios from 'axios'; |
||||
import Knex from 'knex'; |
||||
|
||||
import fs from 'fs'; |
||||
import Audit from '../../../models/Audit'; |
||||
import { sakilaTableNames } from '../../../utils/globals'; |
||||
import Project from '../../../models/Project'; |
||||
|
||||
const config = { |
||||
client: 'mysql2', |
||||
connection: { |
||||
host: 'localhost', |
||||
port: 3306, |
||||
user: 'root', |
||||
password: 'password', |
||||
database: 'sakila', |
||||
multipleStatements: true, |
||||
dateStrings: true, |
||||
}, |
||||
}; |
||||
|
||||
const extMysqlProject = (title, parallelId) => ({ |
||||
title, |
||||
bases: [ |
||||
{ |
||||
type: 'mysql2', |
||||
config: { |
||||
client: 'mysql2', |
||||
connection: { |
||||
host: 'localhost', |
||||
port: '3306', |
||||
user: 'root', |
||||
password: 'password', |
||||
database: `test_sakila_${parallelId}`, |
||||
}, |
||||
}, |
||||
inflection_column: 'camelize', |
||||
inflection_table: 'camelize', |
||||
}, |
||||
], |
||||
external: true, |
||||
}); |
||||
|
||||
const mysqlSakilaSqlViews = [ |
||||
'actor_info', 'customer_list', 'film_list', 'nicer_but_slower_film_list', 'sales_by_film_category', 'sales_by_store', 'staff_list' |
||||
] |
||||
|
||||
const dropTablesAndViews = async (knex: Knex) => { |
||||
for (const view of mysqlSakilaSqlViews) { |
||||
try { |
||||
await knex.raw(`DROP VIEW ${view}`); |
||||
} catch (e) { |
||||
} |
||||
} |
||||
|
||||
for (const table of sakilaTableNames) { |
||||
try { |
||||
await knex.raw(`DROP TABLE ${table}`); |
||||
} catch (e) { |
||||
} |
||||
} |
||||
}; |
||||
|
||||
const isSakilaMysqlToBeReset = async (knex: Knex, parallelId: string, project?: Project, ) => { |
||||
const tablesInDb: Array<string> = await knex.raw( |
||||
`SELECT table_name FROM information_schema.tables WHERE table_schema = 'test_sakila_${parallelId}'` |
||||
) |
||||
|
||||
if ( |
||||
tablesInDb.length === 0 || |
||||
(tablesInDb.length > 0 && !tablesInDb.includes(`actor`)) |
||||
) { |
||||
return true; |
||||
} |
||||
|
||||
if(!project) return false; |
||||
|
||||
const audits = await Audit.projectAuditList(project.id, {}); |
||||
|
||||
return audits?.length > 0; |
||||
}; |
||||
|
||||
const resetSakilaMysql = async (knex:Knex, parallelId: string) => { |
||||
await dropTablesAndViews(knex); |
||||
|
||||
const testsDir = __dirname.replace( |
||||
'/src/lib/services/test/TestResetService', |
||||
'/tests' |
||||
); |
||||
|
||||
try { |
||||
await knex.raw(`DROP DATABASE test_sakila_${parallelId}`); |
||||
} catch (e) { |
||||
console.log('Error dropping db', e); |
||||
} |
||||
await knex.raw(`CREATE DATABASE test_sakila_${parallelId}`); |
||||
|
||||
const trx = await knex.transaction(); |
||||
|
||||
try { |
||||
const schemaFile = fs |
||||
.readFileSync( |
||||
`${testsDir}/mysql-sakila-db/03-test-sakila-schema.sql` |
||||
) |
||||
.toString() |
||||
.replace(/test_sakila/g, `test_sakila_${parallelId}`); |
||||
|
||||
const dataFile = fs |
||||
.readFileSync( |
||||
`${testsDir}/mysql-sakila-db/04-test-sakila-data.sql` |
||||
) |
||||
.toString() |
||||
.replace(/test_sakila/g, `test_sakila_${parallelId}`); |
||||
|
||||
await trx.raw(schemaFile); |
||||
await trx.raw(dataFile); |
||||
|
||||
await trx.commit(); |
||||
} catch (e) { |
||||
console.log('Error resetting mysql db', e); |
||||
await trx.rollback(e); |
||||
} |
||||
}; |
||||
|
||||
const resetMysqlSakilaProject = async ({ |
||||
token, |
||||
title, |
||||
parallelId, |
||||
oldProject, |
||||
}: { |
||||
token: string; |
||||
title: string; |
||||
parallelId: string; |
||||
oldProject?: Project | undefined; |
||||
}) => { |
||||
const knex = Knex(config); |
||||
|
||||
try{ |
||||
await knex.raw(`USE test_sakila_${parallelId}`); |
||||
} catch (e) { |
||||
await knex.raw(`CREATE DATABASE test_sakila_${parallelId}`); |
||||
await knex.raw(`USE test_sakila_${parallelId}`); |
||||
} |
||||
|
||||
if (await isSakilaMysqlToBeReset(knex, parallelId, oldProject)) { |
||||
await resetSakilaMysql(knex, parallelId); |
||||
} |
||||
|
||||
const response = await axios.post( |
||||
'http://localhost:8080/api/v1/db/meta/projects/', |
||||
extMysqlProject(title, parallelId), |
||||
{ |
||||
headers: { |
||||
'xc-auth': token, |
||||
}, |
||||
} |
||||
); |
||||
if (response.status !== 200) { |
||||
console.error('Error creating project', response.data); |
||||
} |
||||
|
||||
await knex.destroy(); |
||||
}; |
||||
|
||||
export default resetMysqlSakilaProject; |
Loading…
Reference in new issue