250 lines
10 KiB
OCaml
250 lines
10 KiB
OCaml
(**************************************************************************)
|
|
(* *)
|
|
(* Copyright (c) 2014 - 2017. *)
|
|
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
|
|
(* *)
|
|
(* All rights reserved. No warranty, explicit or implicit, provided. *)
|
|
(* *)
|
|
(**************************************************************************)
|
|
|
|
module Helpers = Proto_alpha_helpers
|
|
module Assert = Helpers.Assert
|
|
|
|
let { Helpers.Account.b1 ; b2 ; b3 ; b4 ; b5 } =
|
|
Helpers.Account.bootstrap_accounts
|
|
|
|
let default_account =
|
|
Helpers.Account.create "default_account"
|
|
|
|
let test_double_endorsement contract block =
|
|
|
|
(* Double endorsement for the same level *)
|
|
Helpers.Baking.mine block contract [] >>=? fun b1 ->
|
|
|
|
(* branch root *)
|
|
Helpers.Baking.mine (`Hash b1) contract [] >>=? fun b2 ->
|
|
(* changing branch *)
|
|
Helpers.Baking.mine (`Hash b1) contract [] >>=? fun b2' ->
|
|
|
|
(* branch root *)
|
|
Helpers.Endorse.endorse contract (`Hash b2) >>=? fun op ->
|
|
Helpers.Baking.mine (`Hash b2) contract [ op ] >>=? fun _b3 ->
|
|
|
|
Helpers.Endorse.endorse contract (`Hash b2') >>=? fun op ->
|
|
Helpers.Baking.mine (`Hash b2') contract [ op ] >>=? fun b3' ->
|
|
|
|
Helpers.Endorse.endorse contract (`Hash b3') >>=? fun op ->
|
|
Helpers.Baking.mine (`Hash b3') contract [ op ] >>=? fun b4' ->
|
|
|
|
(* TODO: Inject double endorsement op ! *)
|
|
Helpers.Baking.mine (`Hash b4') contract []
|
|
|
|
(* FIXME: Baking.Invalid_signature is unclassified *)
|
|
let test_invalid_signature block =
|
|
let public_key =
|
|
Environment.Ed25519.Public_key.of_b58check_exn
|
|
"edpkv8EUUH68jmo3f7Um5PezmfGrRF24gnfLpH3sVNwJnV5bVCxL2n" in
|
|
let secret_key =
|
|
Environment.Ed25519.Secret_key.of_b58check_exn
|
|
"edskRuR1azSfboG86YPTyxrQgosh5zChf5bVDmptqLTb5EuXAm9rsn\
|
|
DYfTKhq7rDQujdn5WWzwUMeV3agaZ6J2vPQT58jJAJPi" in
|
|
let account =
|
|
Helpers.Account.create ~keys:(secret_key, public_key) "WRONG SIGNATURE" in
|
|
Helpers.Baking.mine block account [] >>= fun res ->
|
|
Assert.generic_economic_error ~msg:__LOC__ res ;
|
|
return ()
|
|
|
|
let contain_tzerror ?(msg="") ~f t =
|
|
t >>= function
|
|
| Ok _ -> failwith "%s: Expected error found success" msg
|
|
| Error error when not (List.exists f error) ->
|
|
failwith "@[<v 2>Unexpected error@ %a@]" pp_print_error error
|
|
| _ -> return ()
|
|
|
|
let test_wrong_delegate ~miner contract block =
|
|
begin
|
|
Helpers.Endorse.endorse ~slot:1 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:2 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:3 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:4 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:5 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block miner [ op ] >>=? fun _ ->
|
|
return ()
|
|
end >>= fun res ->
|
|
Assert.failed_to_preapply ~msg:__LOC__ begin function
|
|
| Baking.Wrong_delegate _ -> true
|
|
| _ -> false
|
|
end res ;
|
|
Lwt.return_unit
|
|
|
|
let test_invalid_endorsement_slot contract block =
|
|
Helpers.Endorse.endorse ~slot:~-1 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block contract [ op ] >>= fun res ->
|
|
Assert.failed_to_preapply ~msg:__LOC__ ~op:(Blob op) begin function
|
|
| Baking.Invalid_endorsement_slot _ -> true
|
|
| _ -> false
|
|
end res ;
|
|
Helpers.Endorse.endorse ~slot:16 contract block >>=? fun op ->
|
|
Helpers.Baking.mine block contract [ op ] >>= fun res ->
|
|
Assert.failed_to_preapply ~msg:__LOC__ ~op:(Blob op) begin function
|
|
| Baking.Invalid_endorsement_slot _ -> true
|
|
| _ -> false
|
|
end res ;
|
|
return ()
|
|
|
|
let test_endorsement_rewards block0 =
|
|
|
|
let get_endorser_except bs accounts =
|
|
let account, cpt = ref accounts.(0), ref 0 in
|
|
while List.mem !account bs do
|
|
incr cpt ;
|
|
account := accounts.(!cpt)
|
|
done ;
|
|
return (!account, !cpt) in
|
|
|
|
let bond = Tez.to_cents Constants.endorsement_bond_cost in
|
|
|
|
(* Endorsement Rights *)
|
|
(* #1 endorse & inject in a block *)
|
|
Helpers.Endorse.endorsers_list block0 >>=? fun accounts ->
|
|
get_endorser_except [ b1 ] accounts >>=? fun (account0, slot0) ->
|
|
Helpers.Account.balance ~block:block0 account0 >>=? fun balance0 ->
|
|
Helpers.Endorse.endorse ~slot:slot0 account0 block0 >>=? fun op ->
|
|
Helpers.Baking.mine block0 b1 [ op ] >>=? fun hash1 ->
|
|
Helpers.display_level (`Hash hash1) >>=? fun () ->
|
|
Assert.balance_equal ~block:(`Hash hash1) ~msg:__LOC__ account0
|
|
(Int64.sub (Tez.to_cents balance0) bond) >>=? fun () ->
|
|
|
|
(* #2 endorse & inject in a block *)
|
|
let block1 = `Hash hash1 in
|
|
Helpers.Endorse.endorsers_list block1 >>=? fun accounts ->
|
|
get_endorser_except [ b1 ; account0 ] accounts >>=? fun (account1, slot1) ->
|
|
Helpers.Account.balance ~block:block1 account1 >>=? fun balance1 ->
|
|
Helpers.Endorse.endorse ~slot:slot1 account1 block1 >>=? fun op ->
|
|
Helpers.Baking.mine block1 b1 [ op ] >>=? fun hash2 ->
|
|
Helpers.display_level (`Hash hash2) >>=? fun () ->
|
|
Assert.balance_equal ~block:(`Hash hash2) ~msg:__LOC__ account1
|
|
(Int64.sub (Tez.to_cents balance1) bond) >>=? fun () ->
|
|
|
|
(* Check rewards after one cycle for account0 *)
|
|
Helpers.Baking.mine (`Hash hash2) b1 [] >>=? fun hash3 ->
|
|
Helpers.display_level (`Hash hash3) >>=? fun () ->
|
|
Helpers.Baking.mine (`Hash hash3) b1 [] >>=? fun hash4 ->
|
|
Helpers.display_level (`Hash hash4) >>=? fun () ->
|
|
Helpers.Baking.mine (`Hash hash4) b1 [] >>=? fun hash5 ->
|
|
Helpers.display_level (`Hash hash5) >>=? fun () ->
|
|
Helpers.Baking.endorsement_reward block1 >>=? fun rw0 ->
|
|
Assert.balance_equal ~block:(`Hash hash5) ~msg:__LOC__ account0
|
|
(Int64.add (Tez.to_cents balance0) rw0) >>=? fun () ->
|
|
|
|
(* Check rewards after one cycle for account1 *)
|
|
Helpers.Baking.endorsement_reward (`Hash hash2) >>=? fun rw1 ->
|
|
Assert.balance_equal ~block:(`Hash hash5) ~msg:__LOC__ account1
|
|
(Int64.add (Tez.to_cents balance1) rw1) >>=? fun () ->
|
|
|
|
(* #2 endorse and check reward only on the good chain *)
|
|
Helpers.Baking.mine (`Hash hash5) b1 []>>=? fun hash6a ->
|
|
Helpers.display_level (`Hash hash6a) >>=? fun () ->
|
|
Helpers.Baking.mine (`Hash hash5) b1 [] >>=? fun hash6b ->
|
|
Helpers.display_level (`Hash hash6b) >>=? fun () ->
|
|
|
|
(* working on head *)
|
|
Helpers.Endorse.endorsers_list (`Hash hash6a) >>=? fun accounts ->
|
|
get_endorser_except [ b1 ] accounts >>=? fun (account3, slot3) ->
|
|
Helpers.Account.balance ~block:(`Hash hash6a) account3 >>=? fun balance3 ->
|
|
Helpers.Endorse.endorse
|
|
~slot:slot3 account3 (`Hash hash6a) >>=? fun ops ->
|
|
Helpers.Baking.mine (`Hash hash6a) b1 [ ops ] >>=? fun hash7a ->
|
|
Helpers.display_level (`Hash hash7a) >>=? fun () ->
|
|
|
|
(* working on fork *)
|
|
Helpers.Endorse.endorsers_list (`Hash hash6b) >>=? fun accounts ->
|
|
get_endorser_except [ b1 ] accounts >>=? fun (account4, slot4) ->
|
|
Helpers.Account.balance ~block:(`Hash hash7a) account4 >>=? fun _balance4 ->
|
|
Helpers.Endorse.endorse ~slot:slot4 account4 (`Hash hash6b) >>=? fun ops ->
|
|
Helpers.Baking.mine (`Hash hash6b) b1 [ ops ] >>=? fun _new_fork ->
|
|
Helpers.display_level (`Hash _new_fork) >>=? fun () ->
|
|
Helpers.Account.balance ~block:(`Hash hash7a) account4 >>=? fun balance4 ->
|
|
|
|
Helpers.Baking.mine (`Hash hash7a) b1 [] >>=? fun hash8a ->
|
|
Helpers.display_level (`Hash hash8a) >>=? fun () ->
|
|
Helpers.Baking.mine (`Hash hash8a) b1 [] >>=? fun hash9a ->
|
|
Helpers.display_level (`Hash hash9a) >>=? fun () ->
|
|
|
|
(* Check rewards after one cycle *)
|
|
Helpers.Baking.endorsement_reward (`Hash hash7a) >>=? fun reward ->
|
|
Assert.balance_equal ~block:(`Hash hash9a) ~msg:__LOC__ account3
|
|
(Int64.add (Tez.to_cents balance3) reward) >>=? fun () ->
|
|
|
|
(* Check no reward for the fork *)
|
|
begin
|
|
if account3 = account4 then return ()
|
|
(* if account4 is different from account3, we need to check that there
|
|
is no reward for him since the endorsement was in the fork branch *)
|
|
else Assert.balance_equal ~block:(`Hash hash9a) ~msg:__LOC__ account4 (Tez.to_cents balance4)
|
|
end >>=? fun () ->
|
|
return ()
|
|
|
|
let test_endorsement_rights contract block =
|
|
Helpers.Endorse.endorsement_rights contract block >>|? fun possibilities ->
|
|
possibilities <> []
|
|
|
|
let run genesis =
|
|
|
|
test_endorsement_rights
|
|
default_account genesis >>=? fun has_right_to_endorse ->
|
|
Assert.equal_bool ~msg:__LOC__ has_right_to_endorse false ;
|
|
test_endorsement_rights b1 genesis >>=? fun has_right_to_endorse ->
|
|
Assert.equal_bool ~msg:__LOC__ has_right_to_endorse true ;
|
|
test_endorsement_rights b1 genesis >>=? fun has_right_to_endorse ->
|
|
Assert.equal_bool ~msg:__LOC__ has_right_to_endorse true ;
|
|
|
|
Assert.balance_equal
|
|
~block:genesis ~msg:__LOC__ b1 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal
|
|
~block:genesis ~msg:__LOC__ b2 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal
|
|
~block:genesis ~msg:__LOC__ b3 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal
|
|
~block:genesis ~msg:__LOC__ b4 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal
|
|
~block:genesis ~msg:__LOC__ b5 4_000_000_00L >>=? fun () ->
|
|
|
|
(* Check Rewards *)
|
|
test_endorsement_rewards genesis >>=? fun () ->
|
|
|
|
(* Endorse with a contract with wrong delegate:
|
|
- contract with no endorsement rights
|
|
- contract which signs at every available slots *)
|
|
test_wrong_delegate ~miner:b1 default_account genesis >>= fun () ->
|
|
test_wrong_delegate ~miner:b1 b5 genesis >>= fun () ->
|
|
|
|
(* Endorse with a wrong slot : -1 and max (16) *)
|
|
test_invalid_endorsement_slot b3 genesis >>=? fun () ->
|
|
|
|
(* FIXME: Baking.Invalid_signature is still unclassified *)
|
|
test_invalid_signature genesis >>=? fun _ ->
|
|
|
|
(* FIXME: cannot inject double endorsement operation yet, but the
|
|
code is still here
|
|
Double endorsement *)
|
|
test_double_endorsement b4 genesis >>=? fun _ ->
|
|
|
|
return ()
|
|
|
|
let main () =
|
|
Helpers.init ~rpc_port:18100 () >>=? fun (_node_pid, genesis) ->
|
|
run (`Hash genesis)
|
|
|
|
|
|
let tests = [
|
|
"main", (fun _ -> main ()) ;
|
|
]
|
|
|
|
let () =
|
|
Test.run "endorsement." tests
|