diff --git a/servers/user-api/src/config/index.ts b/servers/user-api/src/config/index.ts index 8a1d3f294..1192a66d6 100644 --- a/servers/user-api/src/config/index.ts +++ b/servers/user-api/src/config/index.ts @@ -71,6 +71,11 @@ export default { user_id: [...tables.tablesWithSensitivePii], created_by_user_id: ['channels'], }, + // For test seeds + userPIITestSeedTables: { + user_id: [...tables.tablesWithSensitivePii, 'user_firefox_account'], + created_by_user_id: ['channels'], + }, }, sentry: { dsn: process.env.SENTRY_DSN || '', diff --git a/servers/user-api/src/test/functional/seeds.ts b/servers/user-api/src/test/functional/seeds.ts index cffeed9cb..0a7d201dd 100644 --- a/servers/user-api/src/test/functional/seeds.ts +++ b/servers/user-api/src/test/functional/seeds.ts @@ -4,9 +4,9 @@ import { writeClient as conn } from '../../database/client'; import config from '../../config'; export async function truncatePiiTables() { - const allTables = Object.entries(config.database.userPIITables).flatMap( - ([_, tables]) => tables, - ); + const allTables = Object.entries( + config.database.userPIITestSeedTables, + ).flatMap(([_, tables]) => tables); await Promise.all( allTables.map((table: string) => { return conn()(table).truncate(); @@ -26,15 +26,17 @@ export async function PiiTableSeed(user_id: number, firefox_uid: string) { }; await Promise.all( - Object.entries(config.database.userPIITables).flatMap(([key, tables]) => { - return tables.map((tableName) => { - if (!(tableName in specialTablesSeeds)) { - return conn()(tableName).insert({ [key]: user_id }); - } else { - return conn()(tableName).insert(specialTablesSeeds[tableName]); - } - }); - }), + Object.entries(config.database.userPIITestSeedTables).flatMap( + ([key, tables]) => { + return tables.map((tableName) => { + if (!(tableName in specialTablesSeeds)) { + return conn()(tableName).insert({ [key]: user_id }); + } else { + return conn()(tableName).insert(specialTablesSeeds[tableName]); + } + }); + }, + ), ); }