@@ -357,135 +357,127 @@ func TestRead_CanPickRdonlyForShardedKeyspaces(t *testing.T) {
357357}
358358
359359func TestDiscover_CanPickRightAirbyteType (t * testing.T ) {
360- numberType := "number"
361- booleanType := "boolean"
362- stringType := "string"
363-
364360 var tests = []struct {
365361 MysqlType string
366- JSONSchemaType * string
367- OneOf []OneOfType
362+ JSONSchemaType []string
368363 AirbyteType string
369364 TreatTinyIntAsBoolean bool
370365 IsNullable string
371366 }{
372367 {
373368 MysqlType : "int(11)" ,
374- JSONSchemaType : & numberType ,
369+ JSONSchemaType : [] string { "number" } ,
375370 AirbyteType : "integer" ,
376371 },
377372 {
378373 MysqlType : "smallint(4)" ,
379- JSONSchemaType : & numberType ,
374+ JSONSchemaType : [] string { "number" } ,
380375 AirbyteType : "integer" ,
381376 },
382377 {
383378 MysqlType : "mediumint(8)" ,
384- JSONSchemaType : & numberType ,
379+ JSONSchemaType : [] string { "number" } ,
385380 AirbyteType : "integer" ,
386381 },
387382 {
388383 MysqlType : "tinyint" ,
389- JSONSchemaType : & numberType ,
384+ JSONSchemaType : [] string { "number" } ,
390385 AirbyteType : "integer" ,
391386 TreatTinyIntAsBoolean : true ,
392387 },
393388 {
394389 MysqlType : "tinyint(1)" ,
395- JSONSchemaType : & booleanType ,
390+ JSONSchemaType : [] string { "boolean" } ,
396391 AirbyteType : "" ,
397392 TreatTinyIntAsBoolean : true ,
398393 },
399394 {
400395 MysqlType : "tinyint(1) unsigned" ,
401- JSONSchemaType : & booleanType ,
396+ JSONSchemaType : [] string { "boolean" } ,
402397 AirbyteType : "" ,
403398 TreatTinyIntAsBoolean : true ,
404399 },
405400 {
406401 MysqlType : "tinyint(1)" ,
407- JSONSchemaType : & numberType ,
402+ JSONSchemaType : [] string { "number" } ,
408403 AirbyteType : "integer" ,
409404 TreatTinyIntAsBoolean : false ,
410405 },
411406 {
412407 MysqlType : "tinyint(1) unsigned" ,
413- JSONSchemaType : & numberType ,
408+ JSONSchemaType : [] string { "number" } ,
414409 AirbyteType : "integer" ,
415410 TreatTinyIntAsBoolean : false ,
416411 },
417412 {
418413 MysqlType : "bigint(16)" ,
419- JSONSchemaType : & numberType ,
414+ JSONSchemaType : [] string { "number" } ,
420415 AirbyteType : "integer" ,
421416 },
422417 {
423418 MysqlType : "bigint unsigned" ,
424- JSONSchemaType : & numberType ,
419+ JSONSchemaType : [] string { "number" } ,
425420 AirbyteType : "integer" ,
426421 },
427422 {
428423 MysqlType : "bigint zerofill" ,
429- JSONSchemaType : & numberType ,
424+ JSONSchemaType : [] string { "number" } ,
430425 AirbyteType : "integer" ,
431426 },
432427 {
433428 MysqlType : "datetime" ,
434- JSONSchemaType : & stringType ,
429+ JSONSchemaType : [] string { "string" } ,
435430 AirbyteType : "timestamp_without_timezone" ,
436431 },
437432 {
438433 MysqlType : "datetime(6)" ,
439- JSONSchemaType : & stringType ,
434+ JSONSchemaType : [] string { "string" } ,
440435 AirbyteType : "timestamp_without_timezone" ,
441436 },
442437 {
443438 MysqlType : "time" ,
444- JSONSchemaType : & stringType ,
439+ JSONSchemaType : [] string { "string" } ,
445440 AirbyteType : "time_without_timezone" ,
446441 },
447442 {
448443 MysqlType : "time(6)" ,
449- JSONSchemaType : & stringType ,
444+ JSONSchemaType : [] string { "string" } ,
450445 AirbyteType : "time_without_timezone" ,
451446 },
452447 {
453448 MysqlType : "date" ,
454- JSONSchemaType : & stringType ,
449+ JSONSchemaType : [] string { "string" } ,
455450 AirbyteType : "date" ,
456451 },
457452 {
458453 MysqlType : "text" ,
459- JSONSchemaType : & stringType ,
454+ JSONSchemaType : [] string { "string" } ,
460455 AirbyteType : "" ,
461456 },
462457 {
463458 MysqlType : "varchar(256)" ,
464- JSONSchemaType : & stringType ,
459+ JSONSchemaType : [] string { "string" } ,
465460 AirbyteType : "" ,
466461 },
467462 {
468- MysqlType : "varchar(256)" ,
469- AirbyteType : "" ,
470- IsNullable : "YES" ,
471- OneOf : []OneOfType {
472- {Type : "string" },
473- {Type : "null" },
474- },
463+ MysqlType : "varchar(256)" ,
464+ AirbyteType : "" ,
465+ IsNullable : "YES" ,
466+ JSONSchemaType : []string {"null" , "string" },
475467 },
476468 {
477469 MysqlType : "decimal(12,5)" ,
478- JSONSchemaType : & numberType ,
470+ JSONSchemaType : [] string { "number" } ,
479471 AirbyteType : "" ,
480472 },
481473 {
482474 MysqlType : "double" ,
483- JSONSchemaType : & numberType ,
475+ JSONSchemaType : [] string { "number" } ,
484476 AirbyteType : "" ,
485477 },
486478 {
487479 MysqlType : "float(30)" ,
488- JSONSchemaType : & numberType ,
480+ JSONSchemaType : [] string { "number" } ,
489481 AirbyteType : "" ,
490482 },
491483 }
@@ -496,7 +488,6 @@ func TestDiscover_CanPickRightAirbyteType(t *testing.T) {
496488 p := getJsonSchemaType (typeTest .MysqlType , typeTest .TreatTinyIntAsBoolean , typeTest .IsNullable )
497489 assert .Equal (t , typeTest .AirbyteType , p .AirbyteType )
498490 assert .Equal (t , typeTest .JSONSchemaType , p .Type )
499- assert .Equal (t , typeTest .OneOf , p .OneOf )
500491 })
501492 }
502493}
0 commit comments