@humanlayer/sdk
Version:
typescript client for humanlayer.dev
220 lines (219 loc) • 7.04 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const approval_1 = require("./approval");
test('Humanlayer#requireApproval()', async () => {
const mockBackend = {
functions: jest.fn(),
contacts: jest.fn(),
};
const functions = {
add: jest.fn(),
get: jest.fn(),
};
mockBackend.functions.mockReturnValue(functions);
functions.add.mockReturnValue(null);
const mockFunction = jest.fn();
mockFunction.mockReturnValue('bosh');
Object.defineProperty(mockFunction, 'name', { value: '_fn_', writable: false });
const hl = new approval_1.HumanLayer({
backend: mockBackend,
genid: (x) => 'generated-id',
sleep: (x) => Promise.resolve(),
});
const wrapped = hl.requireApproval()(mockFunction);
const returnValue = {
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
},
status: {
requested_at: new Date(),
approved: true,
},
};
functions.get.mockReturnValue(returnValue);
const ret = await wrapped({ bar: 'baz' });
expect(functions.add).toHaveBeenCalledWith({
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
},
});
expect(functions.get).toHaveBeenCalledWith('generated-id');
expect(mockFunction).toHaveBeenCalledWith({ bar: 'baz' });
});
test('Humanlayer(contactChannel)#requireApproval()', async () => {
const mockBackend = {
functions: jest.fn(),
contacts: jest.fn(),
};
const functions = {
add: jest.fn(),
get: jest.fn(),
};
mockBackend.functions.mockReturnValue(functions);
functions.add.mockReturnValue(null);
const mockFunction = jest.fn();
mockFunction.mockReturnValue('bosh');
Object.defineProperty(mockFunction, 'name', { value: '_fn_', writable: false });
const hl = new approval_1.HumanLayer({
contactChannel: {
slack: {
channel_or_user_id: 'U8675309',
context_about_channel_or_user: 'a dm with the librarian',
},
},
backend: mockBackend,
genid: (x) => 'generated-id',
sleep: (x) => Promise.resolve(),
});
const wrapped = hl.requireApproval()(mockFunction);
const returnValue = {
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
channel: {
slack: {
channel_or_user_id: 'U8675309',
context_about_channel_or_user: 'a dm with the librarian',
},
},
},
status: {
requested_at: new Date(),
approved: true,
},
};
functions.get.mockReturnValue(returnValue);
const ret = await wrapped({ bar: 'baz' });
expect(functions.add).toHaveBeenCalledWith({
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
channel: {
slack: {
channel_or_user_id: 'U8675309',
context_about_channel_or_user: 'a dm with the librarian',
},
},
},
});
expect(functions.get).toHaveBeenCalledWith('generated-id');
expect(mockFunction).toHaveBeenCalledWith({ bar: 'baz' });
});
test('Humanlayer()#requireApproval(contactChannel)', async () => {
const mockBackend = {
functions: jest.fn(),
contacts: jest.fn(),
};
const functions = {
add: jest.fn(),
get: jest.fn(),
};
mockBackend.functions.mockReturnValue(functions);
functions.add.mockReturnValue(null);
const mockFunction = jest.fn();
mockFunction.mockReturnValue('bosh');
Object.defineProperty(mockFunction, 'name', { value: '_fn_', writable: false });
const hl = new approval_1.HumanLayer({
backend: mockBackend,
genid: (x) => 'generated-id',
sleep: (x) => Promise.resolve(),
});
const wrapped = hl.requireApproval({
slack: {
channel_or_user_id: 'U8675309',
context_about_channel_or_user: 'a dm with the librarian',
},
})(mockFunction);
const returnValue = {
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
channel: {
slack: {
channel_or_user_id: 'U8675309',
context_about_channel_or_user: 'a dm with the librarian',
},
},
},
status: {
requested_at: new Date(),
approved: true,
},
};
functions.get.mockReturnValue(returnValue);
const ret = await wrapped({ bar: 'baz' });
expect(functions.add).toHaveBeenCalledWith({
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
channel: {
slack: {
channel_or_user_id: 'U8675309',
context_about_channel_or_user: 'a dm with the librarian',
},
},
},
});
expect(functions.get).toHaveBeenCalledWith('generated-id');
expect(mockFunction).toHaveBeenCalledWith({ bar: 'baz' });
});
test('Humanlayer()#requireApproval() with deny', async () => {
const mockBackend = {
functions: jest.fn(),
contacts: jest.fn(),
};
const functions = {
add: jest.fn(),
get: jest.fn(),
};
mockBackend.functions.mockReturnValue(functions);
functions.add.mockReturnValue(null);
const mockFunction = jest.fn();
mockFunction.mockReturnValue('bosh');
Object.defineProperty(mockFunction, 'name', { value: '_fn_', writable: false });
const hl = new approval_1.HumanLayer({
backend: mockBackend,
genid: (x) => 'generated-id',
sleep: (x) => Promise.resolve(),
});
const wrapped = hl.requireApproval()(mockFunction);
const returnValue = {
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
},
status: {
requested_at: new Date(),
approved: false,
comment: 'nope',
},
};
functions.get.mockReturnValue(returnValue);
const ret = await wrapped({ bar: 'baz' });
expect(functions.add).toHaveBeenCalledWith({
run_id: 'generated-id',
call_id: 'generated-id',
spec: {
fn: '_fn_',
kwargs: { bar: 'baz' },
},
});
expect(functions.get).toHaveBeenCalledWith('generated-id');
expect(ret).toBe('User denied function _fn_ with comment: nope');
});