|
|
@@ -10,6 +10,7 @@ import { ActivityEntity } from "./modules/activities/activity.entity"
|
|
|
import { RouteEntity } from "./modules/routes/route.entity"
|
|
|
import { AreaEntity } from "./modules/areas/area.entity"
|
|
|
import { LocationEntity } from "./modules/locations/location.entity"
|
|
|
+import { Passenger } from "./modules/passengers/passenger.entity"
|
|
|
|
|
|
// 在测试环境下使用测试数据库配置
|
|
|
const isTestEnv = process.env.NODE_ENV === 'test';
|
|
|
@@ -20,7 +21,7 @@ const dataSource = isTestEnv && testDatabaseUrl
|
|
|
? new DataSource({
|
|
|
type: "postgres",
|
|
|
url: testDatabaseUrl,
|
|
|
- entities: [User, Role, File, ActivityEntity, RouteEntity, AreaEntity, LocationEntity],
|
|
|
+ entities: [User, Role, File, ActivityEntity, RouteEntity, AreaEntity, LocationEntity, Passenger],
|
|
|
migrations: [],
|
|
|
synchronize: true, // 测试环境总是同步schema
|
|
|
dropSchema: true, // 测试环境每次重新创建schema
|
|
|
@@ -33,7 +34,7 @@ const dataSource = isTestEnv && testDatabaseUrl
|
|
|
username: process.env.DB_USERNAME || "postgres",
|
|
|
password: process.env.DB_PASSWORD || "",
|
|
|
database: process.env.DB_DATABASE || "postgres",
|
|
|
- entities: [User, Role, File, ActivityEntity, RouteEntity, AreaEntity, LocationEntity],
|
|
|
+ entities: [User, Role, File, ActivityEntity, RouteEntity, AreaEntity, LocationEntity, Passenger],
|
|
|
migrations: [],
|
|
|
synchronize: process.env.DB_SYNCHRONIZE !== "false",
|
|
|
logging: process.env.DB_LOGGING === "true",
|