sub2api/backend/migrations/channel_pricing_multipliers_migration_test.go

package migrations

import (
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestChannelPricingMultipliersMigration(t *testing.T) {
	content, err := FS.ReadFile("228_channel_pricing_multipliers.sql")
	require.NoError(t, err)

	sql := strings.Join(strings.Fields(string(content)), " ")
	for _, column := range []string{
		"fast_multiplier NUMERIC(12,6)",
		"flex_multiplier NUMERIC(12,6)",
		"input_multiplier NUMERIC(12,6)",
		"output_multiplier NUMERIC(12,6)",
		"cache_write_multiplier NUMERIC(12,6)",
		"cache_read_multiplier NUMERIC(12,6)",
	} {
		require.Contains(t, sql, "ADD COLUMN IF NOT EXISTS "+column)
	}

	constraints := []struct {
		table  string
		name   string
		column string
	}{
		{"channel_model_pricing", "channel_model_pricing_fast_multiplier_positive", "fast_multiplier"},
		{"channel_model_pricing", "channel_model_pricing_flex_multiplier_positive", "flex_multiplier"},
		{"channel_pricing_intervals", "channel_pricing_intervals_input_multiplier_positive", "input_multiplier"},
		{"channel_pricing_intervals", "channel_pricing_intervals_output_multiplier_positive", "output_multiplier"},
		{"channel_pricing_intervals", "channel_pricing_intervals_cache_write_multiplier_positive", "cache_write_multiplier"},
		{"channel_pricing_intervals", "channel_pricing_intervals_cache_read_multiplier_positive", "cache_read_multiplier"},
	}
	for _, constraint := range constraints {
		require.Contains(t, sql, "conname = '"+constraint.name+"' AND conrelid = '"+constraint.table+"'::regclass")
		require.Contains(t, sql, "ALTER TABLE "+constraint.table+" ADD CONSTRAINT "+constraint.name+
			" CHECK ("+constraint.column+" IS NULL OR "+constraint.column+" > 0)")
	}
}