@@ -115,6 +115,26 @@ describe('useNodePricing', () => {
115115 expect ( price ) . toBe ( '$1.40/Run' )
116116 } )
117117
118+ it ( 'should return low price for kling-v2-turbo model' , ( ) => {
119+ const { getNodeDisplayPrice } = useNodePricing ( )
120+ const node = createMockNode ( 'KlingTextToVideoNode' , [
121+ { name : 'mode' , value : 'pro / 5s / v2-5-turbo' }
122+ ] )
123+
124+ const price = getNodeDisplayPrice ( node )
125+ expect ( price ) . toBe ( '$0.35/Run' )
126+ } )
127+
128+ it ( 'should return high price for kling-v2-turbo model' , ( ) => {
129+ const { getNodeDisplayPrice } = useNodePricing ( )
130+ const node = createMockNode ( 'KlingTextToVideoNode' , [
131+ { name : 'mode' , value : 'pro / 10s / v2-5-turbo' }
132+ ] )
133+
134+ const price = getNodeDisplayPrice ( node )
135+ expect ( price ) . toBe ( '$0.70/Run' )
136+ } )
137+
118138 it ( 'should return standard price for kling-v1-6 model' , ( ) => {
119139 const { getNodeDisplayPrice } = useNodePricing ( )
120140 const node = createMockNode ( 'KlingTextToVideoNode' , [
@@ -155,6 +175,18 @@ describe('useNodePricing', () => {
155175 expect ( price ) . toBe ( '$1.40/Run' )
156176 } )
157177
178+ it ( 'should return high price for kling-v2-5-turbo model' , ( ) => {
179+ const { getNodeDisplayPrice } = useNodePricing ( )
180+ const node = createMockNode ( 'KlingImage2VideoNode' , [
181+ { name : 'model_name' , value : 'v2-5-turbo' } ,
182+ { name : 'mode' , value : 'pro mode / 10s duration / kling-v2-5-turbo' } ,
183+ { name : 'duration' , value : '10' }
184+ ] )
185+
186+ const price = getNodeDisplayPrice ( node )
187+ expect ( price ) . toBe ( '$0.70/Run' )
188+ } )
189+
158190 it ( 'should return standard price for kling-v1-6 model' , ( ) => {
159191 const { getNodeDisplayPrice } = useNodePricing ( )
160192 const node = createMockNode ( 'KlingImage2VideoNode' , [
0 commit comments