open Trace
open Test_helpers

let type_file f =
  Ligo.Compile.Utils.type_file f "cameligo" (Contract "main")

let get_program =
  let s = ref None in
  fun () -> match !s with
    | Some s -> ok s
    | None -> (
        let%bind (program , state) = type_file "./contracts/vote.mligo" in
        s := Some (program , state) ;
        ok (program , state)
      )

open Ast_imperative

let init_storage name = e_record_ez [
    ("title" , e_string name) ;
    ("yea", e_nat 0) ;
    ("nay", e_nat 0) ;
    ("voters" , e_typed_set [] (t_address ())) ;
    ("start_time" , e_timestamp 0) ;
    ("finish_time" , e_timestamp 1000000000) ;
  ]

let reset title start_time finish_time =
  let reset_action = e_record_ez [
      ("title" , e_string title) ;
      ("start_time" , e_timestamp start_time) ;
      ("finish_time" , e_timestamp finish_time)]
  in e_constructor "Reset" reset_action

let yea = e_constructor "Vote" (e_constructor "Yea" (e_unit ()))

let init_vote () =
  let%bind (program , state) = get_program () in
  let%bind result =
    Test_helpers.run_typed_program_with_imperative_input
      (program, state) "main" (e_pair yea (init_storage "basic")) in
  let%bind (_, storage) = Ast_core.extract_pair result in
  let%bind storage' = Ast_core.extract_record storage in
(*  let votes = List.assoc (Label "voters") storage' in
  let%bind votes' = extract_map votes in *)
  let yea = List.assoc (Label "yea") storage' in
  let%bind () = Ast_core.Misc.assert_value_eq (yea, Ast_core.e_nat Z.one) in
  ok ()

let main = test_suite "Vote" [
    test "type" init_vote ;
  ]