Skip to content

Commit 008c035

Browse files
author
Amir Moualem
authored
Merge pull request #271 from snyk/chore/stop-referencing-homebase
fix: removing more references to Homebase
2 parents ab91fb8 + ffad43d commit 008c035

File tree

5 files changed

+18
-17
lines changed

5 files changed

+18
-17
lines changed

src/kube-scanner/index.ts

+5-5
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import logger = require('../common/logger');
22
import { pullImages, removePulledImages, getImagesWithFileSystemPath } from '../images';
33
import { scanImages } from './image';
4-
import { deleteHomebaseWorkload, sendDepGraph } from '../transmitter';
5-
import { constructHomebaseDeleteWorkloadPayload, constructDepGraph } from '../transmitter/payload';
4+
import { deleteWorkload, sendDepGraph } from '../transmitter';
5+
import { constructDeleteWorkloadPayload, constructDepGraph } from '../transmitter/payload';
66
import { IWorkload, ILocalWorkloadLocator } from '../transmitter/types';
77
import { IPullableImage } from '../images/types';
88

@@ -35,10 +35,10 @@ export = class WorkloadWorker {
3535
}
3636

3737
public async delete(localWorkloadLocator: ILocalWorkloadLocator): Promise<void> {
38-
const deletePayload = constructHomebaseDeleteWorkloadPayload(localWorkloadLocator);
38+
const deletePayload = constructDeleteWorkloadPayload(localWorkloadLocator);
3939
logger.info({workloadName: this.name, workload: localWorkloadLocator},
40-
'removing workloads from homebase');
41-
await deleteHomebaseWorkload(deletePayload);
40+
'removing workloads from upstream');
41+
await deleteWorkload(deletePayload);
4242
}
4343

4444
private async scanImagesAndSendResults(

src/kube-scanner/watchers/handlers/pod.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import logger = require('../../../common/logger');
55
import WorkloadWorker = require('../../../kube-scanner');
66
import { sendWorkloadMetadata } from '../../../transmitter';
77
import { IWorkload } from '../../../transmitter/types';
8-
import { constructHomebaseWorkloadMetadataPayload } from '../../../transmitter/payload';
8+
import { constructWorkloadMetadataPayload } from '../../../transmitter/payload';
99
import { buildMetadataForWorkload } from '../../metadata-extractor';
1010
import { PodPhase } from '../types';
1111
import state = require('../../../state');
@@ -97,7 +97,7 @@ export async function podWatchHandler(pod: V1Pod): Promise<void> {
9797

9898
// every element contains the workload information, so we can get it from the first one
9999
const workloadMember = workloadMetadata[0];
100-
const workloadMetadataPayload = constructHomebaseWorkloadMetadataPayload(workloadMember);
100+
const workloadMetadataPayload = constructWorkloadMetadataPayload(workloadMember);
101101
const workloadLocator = workloadMetadataPayload.workloadLocator;
102102
const workloadKey = `${workloadLocator.namespace}/${workloadLocator.type}/${workloadLocator.name}`;
103103
const workloadRevision = workloadMember.revision ? workloadMember.revision.toString() : ''; // this is actually the observed generation

src/transmitter/index.ts

+6-5
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ export async function sendDepGraph(...payloads: IDepGraphPayload[]): Promise<voi
2020
logger.info({payload: payloadWithoutDepGraph, attempt}, 'dependency graph sent upstream successfully');
2121
}
2222
} catch (error) {
23-
logger.error({error, payload: payloadWithoutDepGraph}, 'could not send the dependency scan result to Homebase');
23+
logger.error({error, payload: payloadWithoutDepGraph}, 'could not send the dependency scan result upstream');
2424
}
2525
}
2626
}
@@ -36,15 +36,16 @@ export async function sendWorkloadMetadata(payload: IWorkloadMetadataPayload): P
3636
logger.info({workloadLocator: payload.workloadLocator, attempt}, 'workload metadata sent upstream successfully');
3737
}
3838
} catch (error) {
39-
logger.error({error, workloadLocator: payload.workloadLocator}, 'could not send workload metadata to Homebase');
39+
logger.error({error, workloadLocator: payload.workloadLocator}, 'could not send workload metadata upstream');
4040
}
4141
}
4242

43-
export async function deleteHomebaseWorkload(payload: IDeleteWorkloadPayload): Promise<void> {
43+
export async function deleteWorkload(payload: IDeleteWorkloadPayload): Promise<void> {
4444
try {
4545
const {response, attempt} = await retryRequest('delete', `${upstreamUrl}/api/v1/workload`, payload);
4646
if (response.statusCode === 404) {
47-
const msg = 'attempted to delete a workload Homebase could not find, maybe we are still building it?';
47+
// TODO: maybe we're still building it?
48+
const msg = 'attempted to delete a workload the Upstream service could not find';
4849
logger.info({payload}, msg);
4950
return;
5051
}
@@ -54,7 +55,7 @@ export async function deleteHomebaseWorkload(payload: IDeleteWorkloadPayload): P
5455
logger.info({workloadLocator: payload.workloadLocator, attempt}, 'workload deleted successfully');
5556
}
5657
} catch (error) {
57-
logger.error({error, payload}, 'could not send workload to delete to Homebase');
58+
logger.error({error, payload}, 'could not send delete a workload from the upstream');
5859
}
5960
}
6061

src/transmitter/payload.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ export function constructDepGraph(
5050
return results;
5151
}
5252

53-
export function constructHomebaseWorkloadMetadataPayload(workload: IWorkload): IWorkloadMetadataPayload {
53+
export function constructWorkloadMetadataPayload(workload: IWorkload): IWorkloadMetadataPayload {
5454
if (!workload) {
5555
throw new Error('can\'t build workload metadata payload for undefined workload');
5656
}
@@ -73,7 +73,7 @@ export function constructHomebaseWorkloadMetadataPayload(workload: IWorkload): I
7373
return { workloadLocator, agentId: config.AGENT_ID, workloadMetadata };
7474
}
7575

76-
export function constructHomebaseDeleteWorkloadPayload(
76+
export function constructDeleteWorkloadPayload(
7777
localWorkloadLocator: ILocalWorkloadLocator,
7878
): IDeleteWorkloadPayload {
7979
return {

test/unit/transmitter-payload.test.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ tap.test('constructDepGraph happy flow', async (t) => {
8282

8383
const payloads = payload.constructDepGraph(scannedImages, workloadMetadata);
8484

85-
t.equals(payloads.length, 1, 'one payload to send to Homebase');
85+
t.equals(payloads.length, 1, 'one payload to send upstream');
8686
const firstPayload = payloads[0];
8787
t.equals(firstPayload.dependencyGraph, JSON.stringify('whatever1'), 'dependency graph present in payload');
8888
t.equals(firstPayload.imageLocator.cluster, 'grapefruit', 'cluster present in payload');
@@ -104,7 +104,7 @@ tap.test('constructDepGraph happy flow', async (t) => {
104104
config.MONITOR_VERSION = backups.version;
105105
});
106106

107-
tap.test('constructHomebaseWorkloadMetadataPayload happy flow', async (t) => {
107+
tap.test('constructWorkloadMetadataPayload happy flow', async (t) => {
108108
const workloadWithImages: transmitterTypes.IWorkload = {
109109
type: 'type',
110110
name: 'workloadName',
@@ -122,7 +122,7 @@ tap.test('constructHomebaseWorkloadMetadataPayload happy flow', async (t) => {
122122
podSpec: podSpecFixture,
123123
};
124124

125-
const workloadMetadataPayload = payload.constructHomebaseWorkloadMetadataPayload(workloadWithImages);
125+
const workloadMetadataPayload = payload.constructWorkloadMetadataPayload(workloadWithImages);
126126

127127
t.equals(workloadMetadataPayload.workloadLocator.cluster, 'grapefruit', 'cluster present in payload');
128128
t.equals(workloadMetadataPayload.workloadLocator.namespace, 'spacename', 'image ID present in payload');

0 commit comments

Comments
 (0)