// @flow import { v4 as uuidv4 } from "uuid"; import { Share, Team, User, Event, Document, Collection, Group, GroupUser, Attachment, IntegrationAuthentication, Integration, AuthenticationProvider, FileOperation, } from "../models"; let count = 1; export async function buildShare(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildUser({ teamId: overrides.teamId }); overrides.userId = user.id; } if (!overrides.documentId) { const document = await buildDocument({ createdById: overrides.userId, teamId: overrides.teamId, }); overrides.documentId = document.id; } return Share.create({ published: true, ...overrides, }); } export function buildTeam(overrides: Object = {}) { count++; return Team.create( { name: `Team ${count}`, authenticationProviders: [ { name: "slack", providerId: uuidv4(), }, ], ...overrides, }, { include: "authenticationProviders", } ); } export function buildEvent(overrides: Object = {}) { return Event.create({ name: "documents.publish", ip: "127.0.0.1", ...overrides, }); } export async function buildGuestUser(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } count++; return User.create({ email: `user${count}@example.com`, name: `User ${count}`, createdAt: new Date("2018-01-01T00:00:00.000Z"), lastActiveAt: new Date("2018-01-01T00:00:00.000Z"), ...overrides, }); } export async function buildUser(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } const authenticationProvider = await AuthenticationProvider.findOne({ where: { teamId: overrides.teamId, }, }); count++; return User.create( { email: `user${count}@example.com`, name: `User ${count}`, username: `user${count}`, createdAt: new Date("2018-01-01T00:00:00.000Z"), lastActiveAt: new Date("2018-01-01T00:00:00.000Z"), authentications: [ { authenticationProviderId: authenticationProvider.id, providerId: uuidv4(), }, ], ...overrides, }, { include: "authentications", } ); } export async function buildAdmin(overrides: Object = {}) { return buildUser({ ...overrides, isAdmin: true }); } export async function buildInvite(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } count++; return User.create({ email: `user${count}@example.com`, name: `User ${count}`, createdAt: new Date("2018-01-01T00:00:00.000Z"), ...overrides, }); } export async function buildIntegration(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } const user = await buildUser({ teamId: overrides.teamId }); const authentication = await IntegrationAuthentication.create({ service: "slack", userId: user.id, teamId: user.teamId, token: "fake-access-token", scopes: ["example", "scopes", "here"], }); return Integration.create({ type: "post", service: "slack", settings: { serviceTeamId: "slack_team_id", }, authenticationId: authentication.id, ...overrides, }); } export async function buildCollection(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildUser({ teamId: overrides.teamId }); overrides.userId = user.id; } count++; return Collection.create({ name: `Test Collection ${count}`, description: "Test collection description", createdById: overrides.userId, permission: "read_write", ...overrides, }); } export async function buildGroup(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildUser({ teamId: overrides.teamId }); overrides.userId = user.id; } count++; return Group.create({ name: `Test Group ${count}`, createdById: overrides.userId, ...overrides, }); } export async function buildGroupUser(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildUser({ teamId: overrides.teamId }); overrides.userId = user.id; } count++; return GroupUser.create({ createdById: overrides.userId, ...overrides, }); } export async function buildDocument(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildUser(); overrides.userId = user.id; } if (!overrides.collectionId) { const collection = await buildCollection(overrides); overrides.collectionId = collection.id; } count++; return Document.create({ title: `Document ${count}`, text: "This is the text in an example document", publishedAt: new Date(), lastModifiedById: overrides.userId, createdById: overrides.userId, ...overrides, }); } export async function buildFileOperation(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildAdmin({ teamId: overrides.teamId }); overrides.userId = user.id; } if (!overrides.collectionId) { const collection = await buildCollection(overrides); overrides.collectionId = collection.id; } return FileOperation.create({ state: "creating", size: 0, key: "key/to/aws/file.zip", url: "https://www.urltos3file.com/file.zip", ...overrides, }); } export async function buildAttachment(overrides: Object = {}) { if (!overrides.teamId) { const team = await buildTeam(); overrides.teamId = team.id; } if (!overrides.userId) { const user = await buildUser({ teamId: overrides.teamId }); overrides.userId = user.id; } if (!overrides.documentId) { const document = await buildDocument(overrides); overrides.documentId = document.id; } count++; return Attachment.create({ key: `uploads/key/to/file ${count}.png`, url: `https://redirect.url.com/uploads/key/to/file ${count}.png`, contentType: "image/png", size: 100, acl: "public-read", createdAt: new Date("2018-01-02T00:00:00.000Z"), updatedAt: new Date("2018-01-02T00:00:00.000Z"), ...overrides, }); }