mirror of https://github.com/nocodb/nocodb
Muhammed Mustafa
2 years ago
15 changed files with 365 additions and 34 deletions
@ -0,0 +1,13 @@
|
||||
import { Request, Router } from 'express'; |
||||
import { TestResetService } from '../../services/test/TestResetService'; |
||||
|
||||
export async function reset(_: Request<any, any>, res) { |
||||
const service = new TestResetService(); |
||||
|
||||
res.json(await service.process()); |
||||
} |
||||
|
||||
const router = Router({ mergeParams: true }); |
||||
|
||||
router.get('/api/v1/meta/test/reset', reset); |
||||
export default router; |
||||
@ -0,0 +1,65 @@
|
||||
import axios from 'axios'; |
||||
|
||||
const extPgProject = { |
||||
title: 'pgExtREST', |
||||
bases: [ |
||||
{ |
||||
type: 'pg', |
||||
config: { |
||||
client: 'pg', |
||||
connection: { |
||||
host: 'localhost', |
||||
port: '5432', |
||||
user: 'postgres', |
||||
password: 'password', |
||||
database: 'postgres', |
||||
}, |
||||
searchPath: ['public'], |
||||
}, |
||||
inflection_column: 'camelize', |
||||
inflection_table: 'camelize', |
||||
}, |
||||
], |
||||
external: true, |
||||
}; |
||||
|
||||
// const extMysqlProject = {
|
||||
// title: 'externalREST',
|
||||
// bases: [
|
||||
// {
|
||||
// type: 'mysql2',
|
||||
// config: {
|
||||
// client: 'mysql2',
|
||||
// connection: {
|
||||
// host: 'localhost',
|
||||
// port: '5432',
|
||||
// user: 'root',
|
||||
// password: 'password',
|
||||
// database: 'sakila',
|
||||
// },
|
||||
// },
|
||||
// inflection_column: 'camelize',
|
||||
// inflection_table: 'camelize',
|
||||
// },
|
||||
// ],
|
||||
// external: true,
|
||||
// };
|
||||
|
||||
const createProjects = async (token) => { |
||||
return await Promise.all( |
||||
[extPgProject].map(async (projectAttr) => { |
||||
const response = await axios.post( |
||||
'http://localhost:8080/api/v1/db/meta/projects/', |
||||
projectAttr, |
||||
{ |
||||
headers: { |
||||
'xc-auth': token, |
||||
}, |
||||
} |
||||
); |
||||
return response.data; |
||||
}) |
||||
); |
||||
}; |
||||
|
||||
export default createProjects; |
||||
@ -0,0 +1,17 @@
|
||||
import axios from 'axios'; |
||||
|
||||
const defaultUserArgs = { |
||||
email: 'user@nocodb.com', |
||||
password: 'Password123.', |
||||
}; |
||||
|
||||
const createUser = async () => { |
||||
const response = await axios.post( |
||||
'http://localhost:8080/api/v1/auth/user/signup', |
||||
defaultUserArgs |
||||
); |
||||
|
||||
return { token: response.data.token }; |
||||
}; |
||||
|
||||
export default createUser; |
||||
@ -0,0 +1,39 @@
|
||||
import Noco from '../../../Noco'; |
||||
|
||||
import Knex from 'knex'; |
||||
import NocoCache from '../../../cache/NocoCache'; |
||||
import NcConnectionMgrv2 from '../../../utils/common/NcConnectionMgrv2'; |
||||
import createProjects from './createProjects'; |
||||
import { isPgSakilaToBeReset, resetPgSakila } from './resetPgSakila'; |
||||
import createUser from './createUser'; |
||||
import resetMeta from './resetMeta'; |
||||
|
||||
export class TestResetService { |
||||
private knex: Knex | null = null; |
||||
|
||||
constructor() { |
||||
this.knex = Noco.ncMeta.knex; |
||||
} |
||||
|
||||
async process() { |
||||
try { |
||||
await NcConnectionMgrv2.destroyAll(); |
||||
|
||||
if (await isPgSakilaToBeReset()) { |
||||
await resetPgSakila(); |
||||
} |
||||
|
||||
await resetMeta(this.knex); |
||||
|
||||
await NocoCache.destroy(); |
||||
|
||||
const { token } = await createUser(); |
||||
const projects = await createProjects(token); |
||||
|
||||
return { token, projects }; |
||||
} catch (e) { |
||||
console.error('cleanupMeta', e); |
||||
return { error: e }; |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,59 @@
|
||||
import Model from '../../../models/Model'; |
||||
import Project from '../../../models/Project'; |
||||
import { orderedMetaTables } from '../../../utils/globals'; |
||||
|
||||
const disableForeignKeyChecks = async (knex) => { |
||||
await knex.raw('PRAGMA foreign_keys = OFF'); |
||||
// await this.knex.raw(`SET FOREIGN_KEY_CHECKS = 0`);
|
||||
}; |
||||
|
||||
const enableForeignKeyChecks = async (knex) => { |
||||
await knex.raw(`PRAGMA foreign_keys = ON;`); |
||||
// await this.knex.raw(`SET FOREIGN_KEY_CHECKS = 1`);
|
||||
}; |
||||
|
||||
const dropTablesAllNonExternalProjects = async (knex) => { |
||||
const projects = await Project.list({}); |
||||
const userCreatedTableNames: string[] = []; |
||||
await Promise.all( |
||||
projects |
||||
.filter((project) => project.is_meta) |
||||
.map(async (project) => { |
||||
await project.getBases(); |
||||
const base = project.bases && project.bases[0]; |
||||
if (!base) return; |
||||
|
||||
const models = await Model.list({ |
||||
project_id: project.id, |
||||
base_id: base.id!, |
||||
}); |
||||
models.forEach((model) => { |
||||
userCreatedTableNames.push(model.table_name); |
||||
}); |
||||
}) |
||||
); |
||||
|
||||
await disableForeignKeyChecks(knex); |
||||
|
||||
for (const tableName of userCreatedTableNames) { |
||||
await knex.raw(`DROP TABLE ${tableName}`); |
||||
} |
||||
|
||||
await enableForeignKeyChecks(knex); |
||||
}; |
||||
|
||||
const resetMeta = async (knex) => { |
||||
await dropTablesAllNonExternalProjects(knex); |
||||
|
||||
await disableForeignKeyChecks(knex); |
||||
for (const tableName of orderedMetaTables) { |
||||
try { |
||||
await knex.raw(`DELETE FROM ${tableName}`); |
||||
} catch (e) { |
||||
console.error('cleanupMetaTables', e); |
||||
} |
||||
} |
||||
await enableForeignKeyChecks(knex); |
||||
}; |
||||
|
||||
export default resetMeta; |
||||
@ -0,0 +1,56 @@
|
||||
import knex from 'knex'; |
||||
import fs from 'fs'; |
||||
import Project from '../../../models/Project'; |
||||
import Audit from '../../../models/Audit'; |
||||
|
||||
const config = { |
||||
client: 'pg', |
||||
connection: { |
||||
host: 'localhost', |
||||
port: 5432, |
||||
user: 'postgres', |
||||
password: 'password', |
||||
database: 'postgres', |
||||
}, |
||||
searchPath: ['public'], |
||||
meta: { dbtype: '' }, |
||||
pool: { min: 0, max: 5 }, |
||||
}; |
||||
|
||||
const isPgSakilaToBeReset = async () => { |
||||
const sakilaProject = await Project.getByTitle('pgExtREST'); |
||||
|
||||
const audits = |
||||
sakilaProject && (await Audit.projectAuditList(sakilaProject.id, {})); |
||||
|
||||
return audits.length > 0; |
||||
}; |
||||
|
||||
const resetPgSakila = async () => { |
||||
const knexClient = knex(config); |
||||
|
||||
try { |
||||
await knexClient.raw(`DROP SCHEMA public CASCADE`); |
||||
} catch (e) { |
||||
console.log('Error dropping pg schema', e); |
||||
} |
||||
await knexClient.raw(`CREATE SCHEMA public`); |
||||
|
||||
const testsDir = __dirname.replace( |
||||
'/src/lib/services/test/TestResetService', |
||||
'/tests' |
||||
); |
||||
|
||||
const schemaFile = fs |
||||
.readFileSync(`${testsDir}/pg-sakila-db/01-postgres-sakila-schema.sql`) |
||||
.toString(); |
||||
const dataFile = fs |
||||
.readFileSync(`${testsDir}/pg-sakila-db/02-postgres-sakila-insert-data.sql`) |
||||
.toString(); |
||||
await knexClient.raw(schemaFile); |
||||
await knexClient.raw(dataFile); |
||||
|
||||
await knexClient.destroy(); |
||||
}; |
||||
|
||||
export { resetPgSakila, isPgSakilaToBeReset }; |
||||
@ -0,0 +1,51 @@
|
||||
// let t0 = require("./explicitLogin");
|
||||
// let t01 = require("../common/00_pre_configurations");
|
||||
let t6b = require("../common/6b_downloadCsv"); |
||||
// let t6c = require("../common/6c_swagger_api");
|
||||
// let t6d = require("../common/6d_language_validation");
|
||||
// let t6e = require("../common/6e_project_operations");
|
||||
let t6f = require("../common/6f_attachments"); |
||||
// let t6g = require("../common/6g_base_share");
|
||||
// let t7a = require("../common/7a_create_project_from_excel");
|
||||
const { |
||||
setCurrentMode, |
||||
} = require("../../support/page_objects/projectConstants"); |
||||
// const t8a = require("../common/8a_webhook");
|
||||
const t9b = require("../common/9b_ERD"); |
||||
|
||||
const nocoTestSuite = (apiType, dbType) => { |
||||
setCurrentMode(apiType, dbType); |
||||
|
||||
// Download CSV
|
||||
t6b.genTest(apiType, dbType); |
||||
|
||||
// Attachment cell
|
||||
t6f.genTest(apiType, dbType); |
||||
|
||||
// ERD:
|
||||
t9b.genTest(apiType, dbType); |
||||
}; |
||||
|
||||
nocoTestSuite("rest", "postgres"); |
||||
|
||||
/** |
||||
* @copyright Copyright (c) 2021, Xgene Cloud Ltd |
||||
* |
||||
* @author Raju Udava <sivadstala@gmail.com> |
||||
* |
||||
* @license GNU AGPL version 3 or any later version |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as |
||||
* published by the Free Software Foundation, either version 3 of the |
||||
* License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
*/ |
||||
Loading…
Reference in new issue