mirror of
https://github.com/ToolJet/ToolJet
synced 2026-04-21 13:37:28 +00:00
* SSO for GitHub Enterprise self hosted * changes * test cases * fixes * label fix * fixes * readme changes
1076 lines
38 KiB
TypeScript
1076 lines
38 KiB
TypeScript
import * as request from 'supertest';
|
|
import { INestApplication } from '@nestjs/common';
|
|
import { clearDB, createUser, createNestAppInstanceWithEnvMock } from '../../test.helper';
|
|
import { mocked } from 'ts-jest/utils';
|
|
import got from 'got';
|
|
import { Organization } from 'src/entities/organization.entity';
|
|
import { Repository } from 'typeorm';
|
|
|
|
jest.mock('got');
|
|
const mockedGot = mocked(got);
|
|
|
|
describe('oauth controller', () => {
|
|
let app: INestApplication;
|
|
let orgRepository: Repository<Organization>;
|
|
let mockConfig;
|
|
|
|
const authResponseKeys = [
|
|
'id',
|
|
'email',
|
|
'first_name',
|
|
'last_name',
|
|
'auth_token',
|
|
'admin',
|
|
'organization_id',
|
|
'organization',
|
|
'group_permissions',
|
|
'app_group_permissions',
|
|
].sort();
|
|
|
|
const groupPermissionsKeys = [
|
|
'id',
|
|
'organization_id',
|
|
'group',
|
|
'app_create',
|
|
'app_delete',
|
|
'updated_at',
|
|
'created_at',
|
|
'folder_create',
|
|
'folder_update',
|
|
'folder_delete',
|
|
'org_environment_variable_create',
|
|
'org_environment_variable_delete',
|
|
'org_environment_variable_update',
|
|
].sort();
|
|
|
|
beforeEach(async () => {
|
|
await clearDB();
|
|
});
|
|
|
|
beforeAll(async () => {
|
|
({ app, mockConfig } = await createNestAppInstanceWithEnvMock());
|
|
orgRepository = app.get('OrganizationRepository');
|
|
});
|
|
|
|
afterEach(() => {
|
|
jest.resetAllMocks();
|
|
jest.clearAllMocks();
|
|
});
|
|
|
|
describe('SSO Login', () => {
|
|
let current_organization: Organization;
|
|
beforeEach(async () => {
|
|
const { organization } = await createUser(app, {
|
|
email: 'anotherUser@tooljet.io',
|
|
});
|
|
current_organization = organization;
|
|
});
|
|
|
|
describe('Multi-Workspace instance level SSO', () => {
|
|
beforeEach(() => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
});
|
|
describe('sign in via Git OAuth', () => {
|
|
const token = 'some-Token';
|
|
|
|
it('Workspace Login - should return 401 when the user does not exist and sign up is disabled', async () => {
|
|
await orgRepository.update(current_organization.id, { enableSignUp: false });
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id })
|
|
.expect(401);
|
|
});
|
|
|
|
it('Workspace Login - should return 401 when inherit SSO is disabled', async () => {
|
|
await orgRepository.update(current_organization.id, { inheritSSO: false });
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id })
|
|
.expect(401);
|
|
});
|
|
|
|
it('Common Login - should return 401 when the user does not exist and sign up is disabled', async () => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'SSO_DISABLE_SIGNUPS':
|
|
return 'true';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token }).expect(401);
|
|
});
|
|
|
|
it('Common Login - should return 401 when the user does not exist domain mismatch', async () => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'SSO_ACCEPTED_DOMAINS':
|
|
return 'tooljet.io,tooljet.com';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljett.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token }).expect(401);
|
|
});
|
|
|
|
it('Workspace Login - should return 401 when the user does not exist domain mismatch', async () => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'SSO_ACCEPTED_DOMAINS':
|
|
return 'tooljett.io,tooljet.com';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
await orgRepository.update(current_organization.id, { domain: 'tooljet.io,tooljet.com' });
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljett.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id })
|
|
.expect(401);
|
|
});
|
|
|
|
it('Common Login - should return login info when the user does not exist and domain matches and sign up is enabled', async () => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'SSO_ACCEPTED_DOMAINS':
|
|
return 'tooljet.io,tooljet.com';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('ssoUserGit@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('UserGit');
|
|
expect(admin).toBeTruthy();
|
|
expect(organization_id).not.toBe(current_organization.id);
|
|
expect(organization).toBe('Untitled workspace');
|
|
expect(group_permissions).toHaveLength(2);
|
|
expect([group_permissions[0].group, group_permissions[1].group]).toContain('all_users');
|
|
expect([group_permissions[0].group, group_permissions[1].group]).toContain('admin');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(Object.keys(group_permissions[1]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Workspace Login - should return login info when the user does not exist and domain matches and sign up is enabled', async () => {
|
|
await orgRepository.update(current_organization.id, { domain: 'tooljet.io,tooljet.com', enableSignUp: true });
|
|
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('ssoUserGit@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('UserGit');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Common Login - should return login info when the user does not exist and sign up is enabled', async () => {
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('ssoUserGit@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('UserGit');
|
|
expect(admin).toBeTruthy();
|
|
expect(organization_id).not.toBe(current_organization.id);
|
|
expect(organization).toBe('Untitled workspace');
|
|
expect(group_permissions).toHaveLength(2);
|
|
expect([group_permissions[0].group, group_permissions[1].group]).toContain('all_users');
|
|
expect([group_permissions[0].group, group_permissions[1].group]).toContain('admin');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(Object.keys(group_permissions[1]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Workspace Login - should return login info when the user does not exist and domain includes space matches and sign up is enabled', async () => {
|
|
await orgRepository.update(current_organization.id, {
|
|
enableSignUp: true,
|
|
domain: ' tooljet.io , tooljet.com, , , gmail.com',
|
|
});
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('ssoUserGit@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('UserGit');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Workspace Login - should return login info when the user does not exist and sign up is enabled', async () => {
|
|
await orgRepository.update(current_organization.id, {
|
|
enableSignUp: true,
|
|
});
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO UserGit',
|
|
email: 'ssoUserGit@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('ssoUserGit@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('UserGit');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Common Login - should return login info when the user exist', async () => {
|
|
await createUser(app, {
|
|
firstName: 'SSO',
|
|
lastName: 'userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
groups: ['all_users'],
|
|
organization: current_organization,
|
|
status: 'active',
|
|
});
|
|
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('anotherUser1@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('userExist');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Workspace Login - should return login info when the user exist', async () => {
|
|
await createUser(app, {
|
|
firstName: 'SSO',
|
|
lastName: 'userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
groups: ['all_users'],
|
|
organization: current_organization,
|
|
status: 'active',
|
|
});
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('anotherUser1@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('userExist');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
});
|
|
|
|
it('Common Login - should return login info when the user exist but invited status', async () => {
|
|
const { orgUser } = await createUser(app, {
|
|
firstName: 'SSO',
|
|
lastName: 'userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
groups: ['all_users'],
|
|
organization: current_organization,
|
|
status: 'invited',
|
|
});
|
|
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('anotherUser1@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('userExist');
|
|
expect(admin).toBeTruthy();
|
|
expect(organization_id).not.toBe(current_organization.id);
|
|
expect(organization).toBe('Untitled workspace');
|
|
expect(group_permissions).toHaveLength(2);
|
|
expect([group_permissions[0].group, group_permissions[1].group]).toContain('all_users');
|
|
expect([group_permissions[0].group, group_permissions[1].group]).toContain('admin');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(Object.keys(group_permissions[1]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
await orgUser.reload();
|
|
expect(orgUser.status).toEqual('invited');
|
|
});
|
|
|
|
it('Workspace Login - should return login info when the user exist but invited status', async () => {
|
|
const { orgUser } = await createUser(app, {
|
|
firstName: 'SSO',
|
|
lastName: 'userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
groups: ['all_users'],
|
|
organization: current_organization,
|
|
status: 'invited',
|
|
});
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('anotherUser1@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('userExist');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
await orgUser.reload();
|
|
expect(orgUser.status).toEqual('active');
|
|
});
|
|
it('Common login - should return login info when the user exist and hostname exist in configs', async () => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'SSO_GIT_OAUTH2_HOST':
|
|
return 'https://github.host.com';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
|
|
const { orgUser } = await createUser(app, {
|
|
firstName: 'SSO',
|
|
lastName: 'userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
groups: ['all_users'],
|
|
organization: current_organization,
|
|
});
|
|
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer()).post('/api/oauth/sign-in/common/git').send({ token });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
|
|
expect(gitAuthResponse).toBeCalledWith('https://github.host.com/login/oauth/access_token', expect.anything());
|
|
expect(gitGetUserResponse).toBeCalledWith('https://github.host.com/api/v3/user', expect.anything());
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('anotherUser1@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('userExist');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
await orgUser.reload();
|
|
expect(orgUser.status).toEqual('active');
|
|
});
|
|
it('Workspace login - should return login info when the user exist and hostname exist in configs', async () => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'SSO_GIT_OAUTH2_HOST':
|
|
return 'https://github.host.com';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
|
|
const { orgUser } = await createUser(app, {
|
|
firstName: 'SSO',
|
|
lastName: 'userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
groups: ['all_users'],
|
|
organization: current_organization,
|
|
});
|
|
|
|
const gitAuthResponse = jest.fn();
|
|
gitAuthResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
access_token: 'some-access-token',
|
|
scope: 'scope',
|
|
token_type: 'bearer',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
const gitGetUserResponse = jest.fn();
|
|
gitGetUserResponse.mockImplementation(() => {
|
|
return {
|
|
json: () => {
|
|
return {
|
|
name: 'SSO userExist',
|
|
email: 'anotherUser1@tooljet.io',
|
|
};
|
|
},
|
|
};
|
|
});
|
|
|
|
mockedGot.mockImplementationOnce(gitAuthResponse);
|
|
mockedGot.mockImplementationOnce(gitGetUserResponse);
|
|
|
|
const response = await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token, organizationId: current_organization.id });
|
|
|
|
expect(response.statusCode).toBe(201);
|
|
|
|
expect(gitAuthResponse).toBeCalledWith('https://github.host.com/login/oauth/access_token', expect.anything());
|
|
expect(gitGetUserResponse).toBeCalledWith('https://github.host.com/api/v3/user', expect.anything());
|
|
expect(Object.keys(response.body).sort()).toEqual(authResponseKeys);
|
|
|
|
const {
|
|
email,
|
|
first_name,
|
|
last_name,
|
|
admin,
|
|
group_permissions,
|
|
app_group_permissions,
|
|
organization_id,
|
|
organization,
|
|
} = response.body;
|
|
|
|
expect(email).toEqual('anotherUser1@tooljet.io');
|
|
expect(first_name).toEqual('SSO');
|
|
expect(last_name).toEqual('userExist');
|
|
expect(admin).toBeFalsy();
|
|
expect(organization_id).toBe(current_organization.id);
|
|
expect(organization).toBe(current_organization.name);
|
|
expect(group_permissions).toHaveLength(1);
|
|
expect(group_permissions[0].group).toEqual('all_users');
|
|
expect(Object.keys(group_permissions[0]).sort()).toEqual(groupPermissionsKeys);
|
|
expect(app_group_permissions).toHaveLength(0);
|
|
await orgUser.reload();
|
|
expect(orgUser.status).toEqual('active');
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('Multi-Workspace Disabled', () => {
|
|
beforeEach(() => {
|
|
jest.spyOn(mockConfig, 'get').mockImplementation((key: string) => {
|
|
switch (key) {
|
|
case 'SSO_GOOGLE_OAUTH2_CLIENT_ID':
|
|
return 'google-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_ID':
|
|
return 'git-client-id';
|
|
case 'SSO_GIT_OAUTH2_CLIENT_SECRET':
|
|
return 'git-secret';
|
|
case 'DISABLE_MULTI_WORKSPACE':
|
|
return 'true';
|
|
default:
|
|
return process.env[key];
|
|
}
|
|
});
|
|
});
|
|
describe('sign in via Git OAuth', () => {
|
|
it('should return 403', async () => {
|
|
await request(app.getHttpServer())
|
|
.post('/api/oauth/sign-in/common/git')
|
|
.send({ token: 'some-token' })
|
|
.expect(403);
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
afterAll(async () => {
|
|
await app.close();
|
|
});
|
|
});
|