Skip to content

Commit 7543af3

Browse files
test: add compute job with multiple datasets
1 parent e9451d9 commit 7543af3

2 files changed

Lines changed: 18 additions & 0 deletions

File tree

test/helpers/assets.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ interface Assets {
88
algorithm2Did: string
99
dataset1Did: string
1010
dataset2Did: string
11+
dataset3Did: string
1112
}
1213

1314
export async function initializeAssets() {
@@ -25,6 +26,7 @@ export async function initializeAssets() {
2526
const algorithm2Did = await PublishAsset(ALGORITHM2_SPEC)
2627
const dataset1Did = await PublishAsset(DATASET1_SPEC)
2728
const dataset2Did = await PublishAsset(DATASET2_SPEC)
29+
const dataset3Did = await PublishAsset(DATASET2_SPEC)
2830

2931
fs.writeFileSync(
3032
STATE_FILE,
@@ -34,6 +36,7 @@ export async function initializeAssets() {
3436
algorithm2Did,
3537
dataset1Did,
3638
dataset2Did,
39+
dataset3Did,
3740
},
3841
null,
3942
2,

test/write/asset.test.ts

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ describe('Asset Commands', () => {
7979
describe('Compute Asset Commands', () => {
8080
const algorithmDid = getAssets().algorithm1Did
8181
const datasetDid = getAssets().dataset2Did
82+
const dataset2Did = getAssets().dataset3Did
8283

8384
test(
8485
'compute job works',
@@ -93,6 +94,20 @@ describe('Compute Asset Commands', () => {
9394
}),
9495
60_000,
9596
)
97+
98+
test(
99+
'compute job with multiple datasets works',
100+
withLogin(async () => {
101+
const computeJob = await Compute.run([algorithmDid, '-d', datasetDid, '-d', dataset2Did, '-y'])
102+
expect(computeJob).toBeDefined()
103+
const firstJob = Array.isArray(computeJob) ? computeJob[0] : computeJob
104+
const jobDid = firstJob?.jobId
105+
expect(jobDid).toBeDefined()
106+
await ComputeResults.run([jobDid as string])
107+
await ComputeStatus.run([jobDid as string])
108+
}),
109+
60_000,
110+
)
96111
})
97112

98113
export const concurrent = false

0 commit comments

Comments
 (0)