mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-09 17:40:09 +00:00
8209f5a108
Integrates Stripe's pricing table web component into the subscription dialog when the subscription_tiers_enabled feature flag is active. The implementation includes a new StripePricingTable component that loads Stripe's pricing table script and renders the table with proper error handling and loading states. The subscription dialog now displays the Stripe pricing table with contact us and enterprise links, using a 1100px width that balances multi-column layout with visual design. Configuration supports environment variables, remote config, and window config for the Stripe publishable key and pricing table ID. ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-7288-feat-add-Stripe-pricing-table-integration-for-subscription-dialog-conditional-on-featur-2c46d73d365081fa9d93c213df118996) by [Unito](https://www.unito.io)
114 lines
3.2 KiB
TypeScript
114 lines
3.2 KiB
TypeScript
import { mount, flushPromises } from '@vue/test-utils'
|
|
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
import { createI18n } from 'vue-i18n'
|
|
import { ref } from 'vue'
|
|
|
|
import enMessages from '@/locales/en/main.json' with { type: 'json' }
|
|
import StripePricingTable from '@/platform/cloud/subscription/components/StripePricingTable.vue'
|
|
|
|
const mockLoadStripeScript = vi.fn()
|
|
let currentConfig = {
|
|
publishableKey: 'pk_test_123',
|
|
pricingTableId: 'prctbl_123'
|
|
}
|
|
let hasConfig = true
|
|
|
|
vi.mock('@/config/stripePricingTableConfig', () => ({
|
|
getStripePricingTableConfig: () => currentConfig,
|
|
hasStripePricingTableConfig: () => hasConfig
|
|
}))
|
|
|
|
const mockIsLoaded = ref(false)
|
|
const mockIsLoading = ref(false)
|
|
const mockError = ref(null)
|
|
|
|
vi.mock(
|
|
'@/platform/cloud/subscription/composables/useStripePricingTableLoader',
|
|
() => ({
|
|
useStripePricingTableLoader: () => ({
|
|
loadScript: mockLoadStripeScript,
|
|
isLoaded: mockIsLoaded,
|
|
isLoading: mockIsLoading,
|
|
error: mockError
|
|
})
|
|
})
|
|
)
|
|
|
|
const i18n = createI18n({
|
|
legacy: false,
|
|
locale: 'en',
|
|
messages: { en: enMessages }
|
|
})
|
|
|
|
const mountComponent = () =>
|
|
mount(StripePricingTable, {
|
|
global: {
|
|
plugins: [i18n]
|
|
}
|
|
})
|
|
|
|
describe('StripePricingTable', () => {
|
|
beforeEach(() => {
|
|
currentConfig = {
|
|
publishableKey: 'pk_test_123',
|
|
pricingTableId: 'prctbl_123'
|
|
}
|
|
hasConfig = true
|
|
mockLoadStripeScript.mockReset().mockResolvedValue(undefined)
|
|
mockIsLoaded.value = false
|
|
mockIsLoading.value = false
|
|
mockError.value = null
|
|
})
|
|
|
|
it('renders the Stripe pricing table when config is available', async () => {
|
|
const wrapper = mountComponent()
|
|
|
|
await flushPromises()
|
|
|
|
expect(mockLoadStripeScript).toHaveBeenCalled()
|
|
|
|
const stripePricingTable = wrapper.find('stripe-pricing-table')
|
|
expect(stripePricingTable.exists()).toBe(true)
|
|
expect(stripePricingTable.attributes('publishable-key')).toBe('pk_test_123')
|
|
expect(stripePricingTable.attributes('pricing-table-id')).toBe('prctbl_123')
|
|
})
|
|
|
|
it('shows missing config message when credentials are absent', () => {
|
|
hasConfig = false
|
|
currentConfig = { publishableKey: '', pricingTableId: '' }
|
|
|
|
const wrapper = mountComponent()
|
|
|
|
expect(
|
|
wrapper.find('[data-testid="stripe-table-missing-config"]').exists()
|
|
).toBe(true)
|
|
expect(mockLoadStripeScript).not.toHaveBeenCalled()
|
|
})
|
|
|
|
it('shows loading indicator when script is loading', async () => {
|
|
// Mock loadScript to never resolve, simulating loading state
|
|
mockLoadStripeScript.mockImplementation(() => new Promise(() => {}))
|
|
|
|
const wrapper = mountComponent()
|
|
await flushPromises()
|
|
|
|
expect(wrapper.find('[data-testid="stripe-table-loading"]').exists()).toBe(
|
|
true
|
|
)
|
|
expect(wrapper.find('stripe-pricing-table').exists()).toBe(false)
|
|
})
|
|
|
|
it('shows error indicator when script fails to load', async () => {
|
|
// Mock loadScript to reject, simulating error state
|
|
mockLoadStripeScript.mockRejectedValue(new Error('Script failed to load'))
|
|
|
|
const wrapper = mountComponent()
|
|
await flushPromises()
|
|
|
|
expect(wrapper.find('[data-testid="stripe-table-error"]').exists()).toBe(
|
|
true
|
|
)
|
|
expect(wrapper.find('stripe-pricing-table').exists()).toBe(false)
|
|
})
|
|
})
|