Kristaps Fabians Geikins
136 lines
4.0 KiB
TypeScript
136 lines
4.0 KiB
TypeScript
import { db } from '@/db/knex'
|
|
import { saveActivityFactory } from '@/modules/activitystream/repositories'
|
|
import { addCommitCreatedActivityFactory } from '@/modules/activitystream/services/commitActivity'
|
|
import { VersionsEmitter } from '@/modules/core/events/versionsEmitter'
|
|
import {
|
|
getBranchByIdFactory,
|
|
getStreamBranchByNameFactory,
|
|
markCommitBranchUpdatedFactory
|
|
} from '@/modules/core/repositories/branches'
|
|
import {
|
|
createCommitFactory,
|
|
insertBranchCommitsFactory,
|
|
insertStreamCommitsFactory
|
|
} from '@/modules/core/repositories/commits'
|
|
import { getObjectFactory } from '@/modules/core/repositories/objects'
|
|
import { markCommitStreamUpdatedFactory } from '@/modules/core/repositories/streams'
|
|
import {
|
|
createCommitByBranchIdFactory,
|
|
createCommitByBranchNameFactory
|
|
} from '@/modules/core/services/commit/management'
|
|
import { createObject } from '@/modules/core/services/objects'
|
|
import { publish } from '@/modules/shared/utils/subscriptions'
|
|
import { BasicTestUser } from '@/test/authHelper'
|
|
import { BasicTestStream } from '@/test/speckle-helpers/streamHelper'
|
|
|
|
const markCommitStreamUpdated = markCommitStreamUpdatedFactory({ db })
|
|
const getObject = getObjectFactory({ db })
|
|
const createCommitByBranchId = createCommitByBranchIdFactory({
|
|
createCommit: createCommitFactory({ db }),
|
|
getObject,
|
|
getBranchById: getBranchByIdFactory({ db }),
|
|
insertStreamCommits: insertStreamCommitsFactory({ db }),
|
|
insertBranchCommits: insertBranchCommitsFactory({ db }),
|
|
markCommitStreamUpdated,
|
|
markCommitBranchUpdated: markCommitBranchUpdatedFactory({ db }),
|
|
versionsEventEmitter: VersionsEmitter.emit,
|
|
addCommitCreatedActivity: addCommitCreatedActivityFactory({
|
|
saveActivity: saveActivityFactory({ db }),
|
|
publish
|
|
})
|
|
})
|
|
|
|
const createCommitByBranchName = createCommitByBranchNameFactory({
|
|
createCommitByBranchId,
|
|
getStreamBranchByName: getStreamBranchByNameFactory({ db }),
|
|
getBranchById: getBranchByIdFactory({ db })
|
|
})
|
|
|
|
export type BasicTestCommit = {
|
|
/**
|
|
* Can be left empty, will be filled on creation
|
|
*/
|
|
id: string
|
|
/**
|
|
* Can be left empty, will be filled on creation
|
|
*/
|
|
objectId: string
|
|
/**
|
|
* Can be left empty, will be filled on creation if stream passed in
|
|
*/
|
|
streamId: string
|
|
/**
|
|
* Can be left empty, will be filled on creation if owner passed in
|
|
*/
|
|
authorId: string
|
|
/**
|
|
* Defaults to 'main'
|
|
*/
|
|
branchName?: string
|
|
/**
|
|
* Auto-generated, if empty
|
|
*/
|
|
message?: string
|
|
|
|
/**
|
|
* Empty array by default
|
|
*/
|
|
parents?: string[]
|
|
}
|
|
|
|
export async function createTestObject(params: { projectId: string }) {
|
|
return await createObject({ streamId: params.projectId, object: { foo: 'bar' } })
|
|
}
|
|
|
|
/**
|
|
* Ensure all commits have objectId set
|
|
*/
|
|
async function ensureObjects(commits: BasicTestCommit[]) {
|
|
const commitsWithoutObjects = commits.filter((c) => !c.objectId)
|
|
await Promise.all(
|
|
commitsWithoutObjects.map((c) =>
|
|
createObject({ streamId: c.streamId, object: { foo: 'bar' } }).then(
|
|
(oid) => (c.objectId = oid)
|
|
)
|
|
)
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Create test commits
|
|
*/
|
|
export async function createTestCommits(
|
|
commits: BasicTestCommit[],
|
|
options?: Partial<{ owner: BasicTestUser; stream: BasicTestStream }>
|
|
) {
|
|
const { owner, stream } = options || {}
|
|
|
|
commits.forEach((c) => {
|
|
if (owner) c.authorId = owner.id
|
|
if (stream) c.streamId = stream.id
|
|
})
|
|
|
|
await ensureObjects(commits)
|
|
await Promise.all(
|
|
commits.map((c) =>
|
|
createCommitByBranchName({
|
|
streamId: c.streamId,
|
|
branchName: c.branchName || 'main',
|
|
message: c.message || 'this message is auto generated',
|
|
sourceApplication: 'tests',
|
|
objectId: c.objectId,
|
|
authorId: c.authorId,
|
|
totalChildrenCount: 0,
|
|
parents: c.parents || []
|
|
}).then((newCommit) => (c.id = newCommit.id))
|
|
)
|
|
)
|
|
}
|
|
|
|
export async function createTestCommit(
|
|
commit: BasicTestCommit,
|
|
options?: Partial<{ owner: BasicTestUser; stream: BasicTestStream }>
|
|
) {
|
|
await createTestCommits([commit], options)
|
|
}
|