56 lines
1.6 KiB
OCaml
56 lines
1.6 KiB
OCaml
open Trace
|
|
open Ligo.Run
|
|
open Test_helpers
|
|
|
|
let get_program =
|
|
let s = ref None in
|
|
fun () -> match !s with
|
|
| Some s -> ok s
|
|
| None -> (
|
|
let%bind program = type_file Cameligo "./contracts/vote.mligo" in
|
|
s := Some program ;
|
|
ok program
|
|
)
|
|
|
|
open Ast_simplified
|
|
|
|
let init_storage name = ez_e_record [
|
|
("title" , e_string name) ;
|
|
("candidates" , e_map [
|
|
(e_string "Yes" , e_int 0) ;
|
|
(e_string "No" , e_int 0) ;
|
|
]) ;
|
|
("voters" , e_typed_set [] t_address) ;
|
|
("beginning_time" , e_timestamp 0) ;
|
|
("finish_time" , e_timestamp 1000000000) ;
|
|
]
|
|
|
|
let init title beginning_time finish_time =
|
|
let init_action = ez_e_record [
|
|
("title" , e_string title) ;
|
|
("beginning_time" , e_timestamp beginning_time) ;
|
|
("finish_time" , e_timestamp finish_time) ;
|
|
] in
|
|
e_constructor "Init" init_action
|
|
|
|
let vote str =
|
|
let vote = e_string str in
|
|
e_constructor "Vote" vote
|
|
|
|
let init_vote () =
|
|
let%bind program = get_program () in
|
|
let%bind result = Ligo.Run.run_simplityped program "main" (e_pair (vote "Yes") (init_storage "basic")) in
|
|
let%bind (_ , storage) = extract_pair result in
|
|
let%bind storage' = extract_record storage in
|
|
let votes = List.assoc "candidates" storage' in
|
|
let%bind votes' = extract_map votes in
|
|
let%bind (_ , yess) =
|
|
trace_option (simple_error "") @@
|
|
List.find_opt (fun (k , _) -> Ast_simplified.Misc.is_value_eq (k , e_string "Yes")) votes' in
|
|
let%bind () = Ast_simplified.Misc.assert_value_eq (yess , e_int 1) in
|
|
ok ()
|
|
|
|
let main = test_suite "Vote" [
|
|
test "type" init_vote ;
|
|
]
|