open OUnit2

let printer = function
  | Sqlite3.Data.INT t -> Int64.to_string t
  | Sqlite3.Data.NONE -> "None"
  | Sqlite3.Data.NULL -> "Null"
  | Sqlite3.Data.FLOAT f -> Float.to_string f
  | Sqlite3.Data.TEXT t | Sqlite3.Data.BLOB t -> t

let test_suit =
  [
    ( "Int_of_int" >:: fun _ ->
      assert_equal (Sqlite3.Data.INT 37354L)
        (ImportSQL.Math.int (Sqlite3.Data.INT 37354L)) );
    ( "Int_of_string" >:: fun _ ->
      assert_equal (Sqlite3.Data.INT 37354L)
        (ImportSQL.Math.int (Sqlite3.Data.TEXT "37354")) );
    ( "Int_of_string2" >:: fun _ ->
      assert_equal ~printer (Sqlite3.Data.INT 37354L)
        (ImportSQL.Math.int (Sqlite3.Data.TEXT "37354.0")) );
    ( "Int_of_float" >:: fun _ ->
      assert_equal (Sqlite3.Data.INT 37354L)
        (ImportSQL.Math.int (Sqlite3.Data.FLOAT 37354.0)) );
  ]

let tests = "sql_int" >::: test_suit