twenty/packages/twenty-front/src/testing/mock-data/users.ts

import {
  AUTO_SELECT_FAST_MODEL_ID,
  AUTO_SELECT_SMART_MODEL_ID,
} from 'twenty-shared/constants';
import { type CurrentUserWorkspace } from '@/auth/states/currentUserWorkspaceState';
import { CUSTOM_WORKSPACE_APPLICATION_MOCK } from '@/object-metadata/hooks/__tests__/constants/CustomWorkspaceApplicationMock.test.constant';
import { type WorkspaceMember } from '@/workspace-member/types/WorkspaceMember';
import {
  OnboardingStatus,
  PermissionFlagType,
  SubscriptionInterval,
  SubscriptionStatus,
  type User,
  type Workspace,
  WorkspaceActivationStatus,
  WorkspaceMemberDateFormatEnum,
  WorkspaceMemberTimeFormatEnum,
} from '~/generated-metadata/graphql';
import { mockBillingPlans } from '~/testing/mock-data/billing-plans';
import { getTestEnrichedObjectMetadataItemsMock } from '~/testing/utils/getTestEnrichedObjectMetadataItemsMock';

type MockedUser = Pick<
  User,
  | 'id'
  | 'email'
  | 'firstName'
  | 'lastName'
  | 'canAccessFullAdminPanel'
  | 'canImpersonate'
  | '__typename'
  | 'supportUserHash'
  | 'onboardingStatus'
  | 'userVars'
  | 'availableWorkspaces'
  | 'hasPassword'
> & {
  workspaceMember: WorkspaceMember | null;
  locale: string;
  currentWorkspace: Workspace;
  workspaces: Array<{ workspace: Workspace }>;
  workspaceMembers: WorkspaceMember[];
  currentUserWorkspace: CurrentUserWorkspace;
};

export const avatarUrl =
  'data:image/jpeg;base64,/9j/4AAQSkZJRgABAQAAYABgAAD/4QCMRXhpZgAATU0AKgAAAAgABQESAAMAAAABAAEAAAEaAAUAAAABAAAASgEbAAUAAAABAAAAUgEoAAMAAAABAAIAAIdpAAQAAAABAAAAWgAAAAAAAABgAAAAAQAAAGAAAAABAAOgAQADAAAAAQABAACgAgAEAAAAAQAAABSgAwAEAAAAAQAAABQAAAAA/8AAEQgAFAAUAwEiAAIRAQMRAf/EAB8AAAEFAQEBAQEBAAAAAAAAAAABAgMEBQYHCAkKC//EALUQAAIBAwMCBAMFBQQEAAABfQECAwAEEQUSITFBBhNRYQcicRQygZGhCCNCscEVUtHwJDNicoIJChYXGBkaJSYnKCkqNDU2Nzg5OkNERUZHSElKU1RVVldYWVpjZGVmZ2hpanN0dXZ3eHl6g4SFhoeIiYqSk5SVlpeYmZqio6Slpqeoqaqys7S1tre4ubrCw8TFxsfIycrS09TV1tfY2drh4uPk5ebn6Onq8fLz9PX29/j5+v/EAB8BAAMBAQEBAQEBAQEAAAAAAAABAgMEBQYHCAkKC//EALURAAIBAgQEAwQHBQQEAAECdwABAgMRBAUhMQYSQVEHYXETIjKBCBRCkaGxwQkjM1LwFWJy0QoWJDThJfEXGBkaJicoKSo1Njc4OTpDREVGR0hJSlNUVVZXWFlaY2RlZmdoaWpzdHV2d3h5eoKDhIWGh4iJipKTlJWWl5iZmqKjpKWmp6ipqrKztLW2t7i5usLDxMXGx8jJytLT1NXW19jZ2uLj5OXm5+jp6vLz9PX29/j5+v/bAEMACwgICggHCwoJCg0MCw0RHBIRDw8RIhkaFBwpJCsqKCQnJy0yQDctMD0wJyc4TDk9Q0VISUgrNk9VTkZUQEdIRf/bAEMBDA0NEQ8RIRISIUUuJy5FRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRf/dAAQAAv/aAAwDAQACEQMRAD8Ava1q728otYY98joSCTgZrnbXWdTtrhrfVZXWLafmcAEkdgR/hVltQku9Q8+OIEBcGOT+ID0PY1ka1KH2u8ToqnPLbmIqG7u6LtbQ7RXBRec4Uck9eKXcPWsKDWVnhWSL5kYcFelSf2m3901POh8jP//QoyIAnTuKpXsY82NsksUyWPU5q/L9z8RVK++/F/uCsVsaEURwgA4HtT9x9TUcf3KfUGh//9k=';
export const workspaceLogoUrl =
  'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAACXBIWXMAAA7EAAAOxAGVKw4bAAACb0lEQVR4nO2VO4taQRTHr3AblbjxEVlwCwVhg7BoqqCIjy/gAyyFWNlYBOxsfH0KuxgQGwXRUkGuL2S7i1barGAgiwbdW93SnGOc4BonPiKahf3DwXFmuP/fPM4ZlvmlTxAhCBdzHnEQWYiv7Mr4C3NeuVYhQYDPzOUUQgDLBQGcLHNhvQK8DACPx8PTxiqVyvISG43GbyaT6Qfpn06n0m63e/tPAPF4vJ1MJu8kEsnWTCkWi1yr1RKGw+GDRqPBOTfr44vFQvD7/Q/lcpmaaVQAr9fLp1IpO22c47hGOBz+MB6PH+Vy+VYDAL8qlUoGtVotzOfzq4MAgsHgE/6KojiQyWR/bKVSqbSszHFM8Pl8z1YK48JsNltCOBwOnrYLO+8AAIjb+nHbycoTiUQfDJ7tFq4YAHiVSmXBxcD41u8flQU8z7fhzO0r83atVns3Go3u9Xr9x0O/RQXo9/tsIBBg6vX606a52Wz+bZ7P5/WwG29gxSJzhKgA6XTaDoFNF+krFAocmC//4yWEcSf2wTm7mCO19xFgSsKOLI16vV7b7XY7mRNoLwA0JymJ5uQIzgIAuX5PzDElT2m+E8BqtQ4ymcx7Yq7T6a6ZE4sKgOadTucaCwkxp1UzlEKh0GDxIXOwDWHAdi6Xe3swQDQa/Q7mywoolUpvsaptymazDWKxmBHTlWXZm405BFZoNpuGgwEmk4mE2SGtVivii4f1AO7J3ZopkQCQj7Ar1FeRChCJRJzVapX6DKNIfSc1Ax+wtQWQ55h6bH8FWDfYV4fO3wlwDr0C/BcADYiTPCxHqIEA2QsCZAkAKnRGkMbKN/sTX5YHPQ1e7SkAAAAASUVORK5CYII=';

// Extract Pro monthly base product from mockBillingPlans to use in workspace billing mocks
const PRO_PLAN = mockBillingPlans.listPlans.find((p) => p.planKey === 'PRO')!;
const PRO_BASE_LICENSED_PRODUCT = PRO_PLAN?.licensedProducts?.[0]!;
const PRO_BASE_MONTHLY_PRICE = PRO_BASE_LICENSED_PRODUCT?.prices?.find(
  (pr) => pr.recurringInterval === 'Month',
)!;
const PRO_METERED_PRODUCT = PRO_PLAN?.meteredProducts?.[0]!;
const PRO_METERED_MONTHLY_PRICE = PRO_METERED_PRODUCT?.prices?.find(
  (pr) => pr.recurringInterval === 'Month',
)!;

export const mockCurrentWorkspace = {
  workspaceCustomApplication: CUSTOM_WORKSPACE_APPLICATION_MOCK,
  workspaceCustomApplicationId: CUSTOM_WORKSPACE_APPLICATION_MOCK.id,
  subdomain: 'acme.twenty.com',
  id: '7dfbc3f7-6e5e-4128-957e-8d86808cdf6w',
  displayName: 'Twenty',
  inviteHash: 'twenty.com-invite-hash',
  logo: workspaceLogoUrl,
  isPublicInviteLinkEnabled: true,
  allowImpersonation: true,
  activationStatus: WorkspaceActivationStatus.ACTIVE,
  hasValidEnterpriseKey: false,
  hasValidSignedEnterpriseKey: false,
  hasValidEnterpriseValidityToken: false,
  isGoogleAuthEnabled: true,
  isPasswordAuthEnabled: true,
  isMicrosoftAuthEnabled: false,
  isCustomDomainEnabled: false,
  isPasswordAuthBypassEnabled: false,
  isGoogleAuthBypassEnabled: false,
  isMicrosoftAuthBypassEnabled: false,
  workspaceUrls: {
    customUrl: undefined,
    subdomainUrl: 'twenty.twenty.com',
  },
  featureFlags: [],
  createdAt: '2023-04-26T10:23:42.33625+00:00',
  updatedAt: '2023-04-26T10:23:42.33625+00:00',
  metadataVersion: 1,
  trashRetentionDays: 14,
  fastModel: AUTO_SELECT_FAST_MODEL_ID,
  smartModel: AUTO_SELECT_SMART_MODEL_ID,
  routerModel: 'auto',
  enabledAiModelIds: [],
  useRecommendedModels: true,
  currentBillingSubscription: {
    __typename: 'BillingSubscription',
    id: '7efbc3f7-6e5e-4128-957e-8d86808cdf6a',
    interval: SubscriptionInterval.Month,
    status: SubscriptionStatus.Active,
    currentPeriodEnd: new Date().toString(),
    metadata: {
      plan: PRO_PLAN.planKey,
    },
    phases: [],
    billingSubscriptionItems: [
      {
        __typename: 'BillingSubscriptionItem',
        id: '11111111-1111-4111-8111-111111111111',
        hasReachedCurrentPeriodCap: false,
        quantity: 1,
        stripePriceId: PRO_BASE_MONTHLY_PRICE.stripePriceId,
        billingProduct: {
          name: PRO_BASE_LICENSED_PRODUCT.name,
          description: PRO_BASE_LICENSED_PRODUCT.description,
          images: PRO_BASE_LICENSED_PRODUCT.images,
          metadata: PRO_BASE_LICENSED_PRODUCT.metadata,
        },
      },
      {
        __typename: 'BillingSubscriptionItem',
        id: '11111111-1111-4111-8111-111111111112',
        hasReachedCurrentPeriodCap: false,
        quantity: null,
        stripePriceId: PRO_METERED_MONTHLY_PRICE.stripePriceId,
        billingProduct: {
          name: PRO_METERED_PRODUCT.name,
          description: PRO_METERED_PRODUCT.description,
          images: PRO_METERED_PRODUCT.images,
          metadata: PRO_METERED_PRODUCT.metadata,
        },
      },
    ],
  },
  billingEntitlements: [],
  billingSubscriptions: [
    {
      __typename: 'BillingSubscription',
      id: '7efbc3f7-6e5e-4128-957e-8d86808cdf6a',
      status: SubscriptionStatus.Active,
      metadata: {},
      phases: [],
      billingSubscriptionItems: [
        {
          __typename: 'BillingSubscriptionItem',
          id: '22222222-2222-4222-8222-222222222222',
          hasReachedCurrentPeriodCap: false,
          quantity: 1,
          stripePriceId: PRO_BASE_MONTHLY_PRICE.stripePriceId,
          billingProduct: {
            name: PRO_BASE_LICENSED_PRODUCT.name,
            description: PRO_BASE_LICENSED_PRODUCT.description,
            images: PRO_BASE_LICENSED_PRODUCT.images,
            metadata: PRO_BASE_LICENSED_PRODUCT.metadata,
          },
        },
      ],
    },
  ],
  workspaceMembersCount: 1,
  databaseSchema: '',
  databaseUrl: '',
  isTwoFactorAuthenticationEnforced: false,
  eventLogRetentionDays: 90,
  __typename: 'Workspace',
} as const satisfies Workspace;

export const mockedWorkspaceMemberData: WorkspaceMember = {
  __typename: 'WorkspaceMember',
  id: '7dfbc3f7-6e5e-4128-957e-8d86808cdf6b',
  colorScheme: 'Light',
  avatarUrl,
  locale: 'en',
  name: {
    firstName: 'Charles',
    lastName: 'Test',
  },
  createdAt: '2023-04-26T10:23:42.33625+00:00',
  updatedAt: '2023-04-26T10:23:42.33625+00:00',
  userId: '2603c1f9-0172-4ea6-986c-eeaccdf7f4cf',
  userEmail: 'charles@test.com',
  dateFormat: WorkspaceMemberDateFormatEnum.DAY_FIRST,
  timeFormat: WorkspaceMemberTimeFormatEnum.HOUR_24,
  timeZone: 'America/New_York',
};

export const mockedUserData: MockedUser = {
  id: '7dfbc3f7-6e5e-4128-957e-8d86808cdf6d',
  __typename: 'User',
  email: 'charles@test.com',
  firstName: 'Charles',
  lastName: 'Test',
  canAccessFullAdminPanel: false,
  canImpersonate: false,
  hasPassword: true,
  supportUserHash:
    'a95afad9ff6f0b364e2a3fd3e246a1a852c22b6e55a3ca33745a86c201f9c10d',
  workspaceMember: mockedWorkspaceMemberData,
  currentWorkspace: mockCurrentWorkspace,
  currentUserWorkspace: {
    permissionFlags: [
      PermissionFlagType.WORKSPACE_MEMBERS,
      PermissionFlagType.CONNECTED_ACCOUNTS,
    ],
    twoFactorAuthenticationMethodSummary: [],
    objectsPermissions: getTestEnrichedObjectMetadataItemsMock().map(
      (item) => ({
        objectMetadataId: item.id,
        canReadObjectRecords: true,
        canUpdateObjectRecords: true,
        canSoftDeleteObjectRecords: true,
        canDestroyObjectRecords: true,
        restrictedFields: {},
        rowLevelPermissionPredicates: [],
        rowLevelPermissionPredicateGroups: [],
      }),
    ),
  },
  locale: 'en',
  workspaces: [{ workspace: mockCurrentWorkspace }],
  workspaceMembers: [mockedWorkspaceMemberData],
  onboardingStatus: OnboardingStatus.COMPLETED,
  availableWorkspaces: {
    availableWorkspacesForSignIn: [],
    availableWorkspacesForSignUp: [],
  },
  userVars: {},
};

export const mockedLimitedPermissionsUserData: MockedUser = {
  ...mockedUserData,
  currentUserWorkspace: {
    ...mockedUserData.currentUserWorkspace,
    objectsPermissions: getTestEnrichedObjectMetadataItemsMock()
      .filter(
        (objectMetadata) =>
          objectMetadata.nameSingular !== 'task' &&
          objectMetadata.nameSingular !== 'opportunity',
      )
      .map((item) => ({
        objectMetadataId: item.id,
        canReadObjectRecords: true,
        canUpdateObjectRecords: true,
        canSoftDeleteObjectRecords: true,
        canDestroyObjectRecords: true,
        restrictedFields: {},
        rowLevelPermissionPredicates: [],
        rowLevelPermissionPredicateGroups: [],
      })),
  },
};

export const mockedOnboardingUserData = (
  onboardingStatus?: OnboardingStatus,
) => {
  return {
    id: '7dfbc3f7-6e5e-4128-957e-8d86808cdf6d',
    __typename: 'User',
    email: 'workspace-onboarding@test.com',
    firstName: '',
    lastName: '',
    hasPassword: false,
    canAccessFullAdminPanel: false,
    canImpersonate: false,
    supportUserHash:
      '4fb61d34ed3a4aeda2476d4b308b5162db9e1809b2b8277e6fdc6efc4a609254',
    workspaceMember: null,
    workspaceMembers: [],
    currentWorkspace: mockCurrentWorkspace,
    currentUserWorkspace: {
      permissionFlags: [PermissionFlagType.WORKSPACE_MEMBERS],
      objectPermissions: getTestEnrichedObjectMetadataItemsMock().map(
        (item) => ({
          objectMetadataId: item.id,
          canReadObjectRecords: true,
          canUpdateObjectRecords: true,
          canSoftDeleteObjectRecords: true,
          canDestroyObjectRecords: true,
          restrictedFields: {},
          rowLevelPermissionPredicates: [],
          rowLevelPermissionPredicateGroups: [],
        }),
      ),
    },
    locale: 'en',
    workspaces: [{ workspace: mockCurrentWorkspace }],
    onboardingStatus: onboardingStatus || null,
    userVars: {},
    availableWorkspaces: {
      availableWorkspacesForSignIn: [],
      availableWorkspacesForSignUp: [],
    },
  };
};