@@ -12,19 +12,20 @@ import (
1212
1313func TestParameter_Inference (t * testing.T ) {
1414 t .Run ("Should infer types correctly" , func (t * testing.T ) {
15- values := [5 ]driver.NamedValue {{Name : "" , Value : float32 (5.1 )}, {Name : "" , Value : time .Now ()}, {Name : "" , Value : int64 (5 )}, {Name : "" , Value : true }, {Name : "" , Value : DbSqlParam {Value : "6.2" , Type : Decimal }}}
15+ values := [5 ]driver.NamedValue {{Name : "" , Value : float32 (5.1 )}, {Name : "" , Value : time .Now ()}, {Name : "" , Value : int64 (5 )}, {Name : "" , Value : true }, {Name : "" , Value : DBSqlParam {Value : "6.2" , Type : Decimal }}}
1616 parameters := convertNamedValuesToSparkParams (values [:])
1717 assert .Equal (t , strconv .FormatFloat (float64 (5.1 ), 'f' , - 1 , 64 ), * parameters [0 ].Value .StringValue )
1818 assert .NotNil (t , parameters [1 ].Value .StringValue )
1919 assert .Equal (t , string ("TIMESTAMP" ), * parameters [1 ].Type )
2020 assert .Equal (t , & cli_service.TSparkParameterValue {StringValue : strPtr ("5" )}, parameters [2 ].Value )
2121 assert .Equal (t , string ("true" ), * parameters [3 ].Value .StringValue )
22+ assert .Equal (t , string ("DECIMAL" ), * parameters [4 ].Type )
2223 assert .Equal (t , string ("6.2" ), * parameters [4 ].Value .StringValue )
2324 })
2425}
2526func TestParameters_Names (t * testing.T ) {
2627 t .Run ("Should infer types correctly" , func (t * testing.T ) {
27- values := [2 ]driver.NamedValue {{Name : "1" , Value : int (26 )}, {Name : "2 " , Value : DbSqlParam { Type : Decimal , Value : "6.2" }}}
28+ values := [2 ]driver.NamedValue {{Name : "1" , Value : int (26 )}, {Name : "" , Value : DBSqlParam { Name : "2" , Type : Decimal , Value : "6.2" }}}
2829 parameters := convertNamedValuesToSparkParams (values [:])
2930 assert .Equal (t , string ("1" ), * parameters [0 ].Name )
3031 assert .Equal (t , cli_service.TSparkParameterValue {StringValue : strPtr ("26" )}, * parameters [0 ].Value )
0 commit comments