@@ -115,6 +115,26 @@ describe('useNodePricing', () => {
115
115
expect ( price ) . toBe ( '$1.40/Run' )
116
116
} )
117
117
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.7/Run' )
136
+ } )
137
+
118
138
it ( 'should return standard price for kling-v1-6 model' , ( ) => {
119
139
const { getNodeDisplayPrice } = useNodePricing ( )
120
140
const node = createMockNode ( 'KlingTextToVideoNode' , [
@@ -155,6 +175,18 @@ describe('useNodePricing', () => {
155
175
expect ( price ) . toBe ( '$1.40/Run' )
156
176
} )
157
177
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.7/Run' )
188
+ } )
189
+
158
190
it ( 'should return standard price for kling-v1-6 model' , ( ) => {
159
191
const { getNodeDisplayPrice } = useNodePricing ( )
160
192
const node = createMockNode ( 'KlingImage2VideoNode' , [
0 commit comments