-
Notifications
You must be signed in to change notification settings - Fork 15
/
synchronization.ts
91 lines (82 loc) · 2.54 KB
/
synchronization.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import { Polly } from '@pollyjs/core';
import {
SynchronizationJob,
SynchronizationJobStatus,
} from '@jupiterone/integration-sdk-core';
interface SetupOptions {
baseUrl: string;
polly: Polly;
job: SynchronizationJob;
}
export function setupSynchronizerApi({ polly, job, baseUrl }: SetupOptions) {
polly.server
.post(`${baseUrl}/persister/synchronization/jobs`)
.intercept((req, res) => {
allowCrossOrigin(req, res);
res.status(200).json({ job });
});
polly.server
.post(`${baseUrl}/persister/synchronization/jobs/${job.id}/entities`)
.intercept((req, res) => {
allowCrossOrigin(req, res);
job.numEntitiesUploaded += JSON.parse(req.body!).entities.length;
res.status(200).json({ job });
});
polly.server
.post(`${baseUrl}/persister/synchronization/jobs/${job.id}/events`)
.intercept((req, res) => {
allowCrossOrigin(req, res);
res.status(200).json({});
});
polly.server
.post(`${baseUrl}/persister/synchronization/jobs/${job.id}/relationships`)
.intercept((req, res) => {
allowCrossOrigin(req, res);
job.numRelationshipsUploaded += JSON.parse(
req.body!,
).relationships.length;
res.status(200).json({ job });
});
polly.server
.post(`${baseUrl}/persister/synchronization/jobs/${job.id}/finalize`)
.intercept((req, res) => {
allowCrossOrigin(req, res);
job.status = SynchronizationJobStatus.FINALIZE_PENDING;
res.status(200).json({ job });
});
polly.server
.post(`${baseUrl}/persister/synchronization/jobs/${job.id}/abort`)
.intercept((req, res) => {
allowCrossOrigin(req, res);
job.status = SynchronizationJobStatus.ABORTED;
res.status(200).json({ job });
});
}
function allowCrossOrigin(req, res) {
res.setHeaders({
'Access-Control-Allow-Origin': req.getHeader('origin'),
'Access-Control-Allow-Method': req.getHeader(
'access-control-request-method',
),
'Access-Control-Allow-Headers': req.getHeader(
'access-control-request-headers',
),
});
}
export function generateSynchronizationJob(): SynchronizationJob {
return {
id: 'test',
integrationJobId: 'test-job-id',
integrationInstanceId: 'test-instance-id',
status: SynchronizationJobStatus.AWAITING_UPLOADS,
startTimestamp: Date.now(),
numEntitiesUploaded: 0,
numEntitiesCreated: 0,
numEntitiesUpdated: 0,
numEntitiesDeleted: 0,
numRelationshipsUploaded: 0,
numRelationshipsCreated: 0,
numRelationshipsUpdated: 0,
numRelationshipsDeleted: 0,
};
}