mirror of
https://github.com/xuthus83/pigallery2.git
synced 2024-11-03 21:04:03 +08:00
81 lines
2.5 KiB
TypeScript
81 lines
2.5 KiB
TypeScript
import {expect} from 'chai';
|
|
import {PersonManager} from '../../../../../src/backend/model/database/PersonManager';
|
|
import {DBTestHelper} from '../../../DBTestHelper';
|
|
import {TestHelper} from '../../../../TestHelper';
|
|
import {PhotoDTO} from '../../../../../src/common/entities/PhotoDTO';
|
|
import {Utils} from '../../../../../src/common/Utils';
|
|
import {ParentDirectoryDTO} from '../../../../../src/common/entities/DirectoryDTO';
|
|
import {VideoDTO} from '../../../../../src/common/entities/VideoDTO';
|
|
import {SQLConnection} from '../../../../../src/backend/model/database/SQLConnection';
|
|
import {PersonEntry} from '../../../../../src/backend/model/database/enitites/PersonEntry';
|
|
|
|
|
|
// to help WebStorm to handle the test cases
|
|
declare let describe: any;
|
|
declare const after: any;
|
|
declare const before: any;
|
|
declare const it: any;
|
|
|
|
|
|
// eslint-disable-next-line prefer-const
|
|
describe = DBTestHelper.describe();
|
|
|
|
describe('PersonManager', (sqlHelper: DBTestHelper) => {
|
|
|
|
|
|
let dir: ParentDirectoryDTO;
|
|
|
|
let v: VideoDTO;
|
|
let p: PhotoDTO;
|
|
let p2: PhotoDTO;
|
|
let pFaceLess: PhotoDTO;
|
|
|
|
let savedPerson: PersonEntry[] = [];
|
|
|
|
const setUpSqlDB = async () => {
|
|
await sqlHelper.initDB();
|
|
const directory: ParentDirectoryDTO = TestHelper.getDirectoryEntry();
|
|
p = TestHelper.getPhotoEntry1(directory);
|
|
p2 = TestHelper.getPhotoEntry2(directory);
|
|
const pFaceLessTmp = TestHelper.getPhotoEntry3(directory);
|
|
delete pFaceLessTmp.metadata.faces;
|
|
v = TestHelper.getVideoEntry1(directory);
|
|
|
|
dir = await DBTestHelper.persistTestDir(directory);
|
|
p = (dir.media.filter(m => m.name === p.name)[0] as any);
|
|
p2 = (dir.media.filter(m => m.name === p2.name)[0] as any);
|
|
pFaceLess = (dir.media[2] as any);
|
|
v = (dir.media.filter(m => m.name === v.name)[0] as any);
|
|
savedPerson = await (await SQLConnection.getConnection()).getRepository(PersonEntry).find({
|
|
relations: ['sampleRegion',
|
|
'sampleRegion.media',
|
|
'sampleRegion.media.directory']
|
|
});
|
|
};
|
|
|
|
|
|
before(async () => {
|
|
await setUpSqlDB();
|
|
});
|
|
|
|
after(async () => {
|
|
await sqlHelper.clearDB();
|
|
});
|
|
|
|
|
|
it('should get person', async () => {
|
|
const pm = new PersonManager();
|
|
const person = Utils.clone(savedPerson[0]);
|
|
|
|
const selected = Utils.clone(await pm.get('Boba Fett'));
|
|
delete selected.sampleRegion;
|
|
delete person.sampleRegion;
|
|
person.count = 1;
|
|
expect(selected).to.deep.equal(person);
|
|
|
|
expect((await pm.get('Boba Fett') as PersonEntry).sampleRegion.media.name).to.deep.equal(p.name);
|
|
});
|
|
|
|
|
|
});
|