4bbc97aeb6
It now takes a `proto_header` in parameter, and it returns a full `shell_header`. This prepares the inclusion of the context's hash in the `shell_header`.
247 lines
10 KiB
OCaml
247 lines
10 KiB
OCaml
(**************************************************************************)
|
|
(* *)
|
|
(* Copyright (c) 2014 - 2016. *)
|
|
(* Dynamic Ledger Solutions, Inc. <contact@tezos.com> *)
|
|
(* *)
|
|
(* All rights reserved. No warranty, explicit or implicit, provided. *)
|
|
(* *)
|
|
(**************************************************************************)
|
|
|
|
open Client_embedded_proto_alpha
|
|
open Tezos_context
|
|
open Client_alpha
|
|
|
|
module Helpers = Proto_alpha_helpers
|
|
module Assert = Helpers.Assert
|
|
|
|
let test_double_endorsement contract block =
|
|
|
|
(* Double endorsement for the same level *)
|
|
Helpers.Mining.mine block contract [] >>=? fun b1 ->
|
|
|
|
(* branch root *)
|
|
Helpers.Mining.mine (`Hash b1) contract [] >>=? fun b2 ->
|
|
(* changing branch *)
|
|
Helpers.Mining.mine (`Hash b1) contract [] >>=? fun b2' ->
|
|
|
|
(* branch root *)
|
|
Helpers.Endorse.endorse contract (`Hash b2) >>=? fun op ->
|
|
Helpers.Mining.mine (`Hash b2) contract [ op ] >>=? fun _b3 ->
|
|
|
|
Helpers.Endorse.endorse contract (`Hash b2') >>=? fun op ->
|
|
Helpers.Mining.mine (`Hash b2') contract [ op ] >>=? fun b3' ->
|
|
|
|
Helpers.Endorse.endorse contract (`Hash b3') >>=? fun op ->
|
|
Helpers.Mining.mine (`Hash b3') contract [ op ] >>=? fun b4' ->
|
|
|
|
(* TODO: Inject double endorsement op ! *)
|
|
Helpers.Mining.mine (`Hash b4') contract []
|
|
|
|
(* FIXME: Mining.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.Mining.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 head =
|
|
let block = `Hash head in
|
|
begin
|
|
Helpers.Endorse.endorse ~slot:1 contract block >>=? fun op ->
|
|
Helpers.Mining.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:2 contract block >>=? fun op ->
|
|
Helpers.Mining.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:3 contract block >>=? fun op ->
|
|
Helpers.Mining.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:4 contract block >>=? fun op ->
|
|
Helpers.Mining.mine block miner [ op ] >>=? fun _ ->
|
|
Helpers.Endorse.endorse ~slot:5 contract block >>=? fun op ->
|
|
Helpers.Mining.mine block miner [ op ] >>=? fun _ ->
|
|
return ()
|
|
end >>= fun res ->
|
|
Assert.failed_to_preapply ~msg:__LOC__ begin function
|
|
| Mining.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.Mining.mine block contract [ op ] >>= fun res ->
|
|
Assert.failed_to_preapply ~msg:__LOC__ ~op:(Blob op) begin function
|
|
| Mining.Invalid_endorsement_slot _ -> true
|
|
| _ -> false
|
|
end res ;
|
|
Helpers.Endorse.endorse ~slot:16 contract block >>=? fun op ->
|
|
Helpers.Mining.mine block contract [ op ] >>= fun res ->
|
|
Assert.failed_to_preapply ~msg:__LOC__ ~op:(Blob op) begin function
|
|
| Mining.Invalid_endorsement_slot _ -> true
|
|
| _ -> false
|
|
end res ;
|
|
return ()
|
|
|
|
let test_endorsement_rewards
|
|
block ({ Helpers.Account.b5 = b1 ; _ } as baccounts) =
|
|
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 block baccounts >>=? fun accounts ->
|
|
get_endorser_except [ b1 ] accounts >>=? fun (account0, slot0) ->
|
|
Helpers.Account.balance ~block account0 >>=? fun balance0 ->
|
|
Helpers.Endorse.endorse ~slot:slot0 account0 block >>=? fun ops ->
|
|
Helpers.Mining.mine block b1 [ ops ] >>=? fun head0 ->
|
|
Helpers.display_level (`Hash head0) >>=? fun () ->
|
|
Assert.balance_equal ~block:(`Hash head0) ~msg:__LOC__ account0
|
|
(Int64.sub (Tez.to_cents balance0) bond) >>=? fun () ->
|
|
|
|
|
|
(* #2 endorse & inject in a block *)
|
|
let block0 = `Hash head0 in
|
|
Helpers.Endorse.endorsers_list block0 baccounts >>=? fun accounts ->
|
|
get_endorser_except [ b1 ; account0 ] accounts >>=? fun (account1, slot1) ->
|
|
Helpers.Account.balance ~block:block0 account1 >>=? fun balance1 ->
|
|
Helpers.Endorse.endorse ~slot:slot1 account1 block0 >>=? fun ops ->
|
|
Helpers.Mining.mine block0 b1 [ ops ] >>=? fun head1 ->
|
|
Helpers.display_level (`Hash head1) >>=? fun () ->
|
|
Assert.balance_equal ~block:(`Hash head1) ~msg:__LOC__ account1
|
|
(Int64.sub (Tez.to_cents balance1) bond) >>=? fun () ->
|
|
|
|
(* Check rewards after one cycle for account0 *)
|
|
Helpers.Mining.mine (`Hash head1) b1 [] >>=? fun head2 ->
|
|
Helpers.display_level (`Hash head2) >>=? fun () ->
|
|
Helpers.Mining.mine (`Hash head2) b1 [] >>=? fun head3 ->
|
|
Helpers.display_level (`Hash head3) >>=? fun () ->
|
|
Helpers.Mining.mine (`Hash head3) b1 [] >>=? fun head4 ->
|
|
Helpers.display_level (`Hash head4) >>=? fun () ->
|
|
Helpers.Mining.endorsement_reward block0 >>=? fun rw0 ->
|
|
Assert.balance_equal ~block:(`Hash head4) ~msg:__LOC__ account0
|
|
(Int64.add (Tez.to_cents balance0) rw0) >>=? fun () ->
|
|
|
|
(* Check rewards after one cycle for account1 *)
|
|
Helpers.Mining.endorsement_reward (`Hash head1) >>=? fun rw1 ->
|
|
Assert.balance_equal ~block:(`Hash head4) ~msg:__LOC__ account1
|
|
(Int64.add (Tez.to_cents balance1) rw1) >>=? fun () ->
|
|
|
|
(* #2 endorse and check reward only on the good chain *)
|
|
Helpers.Mining.mine (`Hash head4) b1 []>>=? fun head ->
|
|
Helpers.display_level (`Hash head) >>=? fun () ->
|
|
Helpers.Mining.mine (`Hash head4) b1 [] >>=? fun fork ->
|
|
Helpers.display_level (`Hash fork) >>=? fun () ->
|
|
|
|
(* working on head *)
|
|
Helpers.Endorse.endorsers_list (`Hash head) baccounts >>=? fun accounts ->
|
|
get_endorser_except [ b1 ] accounts >>=? fun (account3, slot3) ->
|
|
Helpers.Account.balance ~block:(`Hash head) account3 >>=? fun balance3 ->
|
|
Helpers.Endorse.endorse
|
|
~slot:slot3 account3 (`Hash head) >>=? fun ops ->
|
|
Helpers.Mining.mine (`Hash head) b1 [ ops ] >>=? fun new_head ->
|
|
Helpers.display_level (`Hash new_head) >>=? fun () ->
|
|
|
|
(* working on fork *)
|
|
Helpers.Endorse.endorsers_list (`Hash fork) baccounts >>=? fun accounts ->
|
|
get_endorser_except [ b1 ] accounts >>=? fun (account4, slot4) ->
|
|
Helpers.Account.balance ~block:(`Hash new_head) account4 >>=? fun _balance4 ->
|
|
Helpers.Endorse.endorse ~slot:slot4 account4 (`Hash fork) >>=? fun ops ->
|
|
Helpers.Mining.mine (`Hash fork) b1 [ ops ] >>=? fun _new_fork ->
|
|
Helpers.display_level (`Hash _new_fork) >>=? fun () ->
|
|
Helpers.Account.balance ~block:(`Hash new_head) account4 >>=? fun balance4 ->
|
|
|
|
Helpers.Mining.mine (`Hash new_head) b1 [] >>=? fun head ->
|
|
Helpers.display_level (`Hash head) >>=? fun () ->
|
|
Helpers.Mining.mine (`Hash head) b1 [] >>=? fun head ->
|
|
Helpers.display_level (`Hash head) >>=? fun () ->
|
|
|
|
(* Check rewards after one cycle *)
|
|
Helpers.Mining.endorsement_reward (`Hash new_head) >>=? fun reward ->
|
|
Assert.balance_equal ~block:(`Hash head) ~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 head) ~msg:__LOC__ account4 (Tez.to_cents balance4)
|
|
end >>=? fun () ->
|
|
return head
|
|
|
|
let test_endorsement_rights contract block =
|
|
Helpers.Endorse.endorsement_rights contract block >>|? fun possibilities ->
|
|
possibilities <> []
|
|
|
|
let run head (({ b1 ; b2 ; b3 ; b4 ; b5 } : Helpers.Account.bootstrap_accounts) as baccounts) =
|
|
|
|
let default_account = Helpers.Account.create "default_account" in
|
|
|
|
test_endorsement_rights default_account head >>=? fun has_right_to_endorse ->
|
|
Assert.equal_bool ~msg:__LOC__ has_right_to_endorse false ;
|
|
test_endorsement_rights b1 head >>=? fun has_right_to_endorse ->
|
|
Assert.equal_bool ~msg:__LOC__ has_right_to_endorse true ;
|
|
test_endorsement_rights b1 head >>=? fun has_right_to_endorse ->
|
|
Assert.equal_bool ~msg:__LOC__ has_right_to_endorse true ;
|
|
|
|
Assert.balance_equal ~block:head ~msg:__LOC__ b1 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal ~block:head ~msg:__LOC__ b2 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal ~block:head ~msg:__LOC__ b3 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal ~block:head ~msg:__LOC__ b4 4_000_000_00L >>=? fun () ->
|
|
Assert.balance_equal ~block:head ~msg:__LOC__ b5 4_000_000_00L >>=? fun () ->
|
|
|
|
(* Check Rewards *)
|
|
test_endorsement_rewards head baccounts >>=? fun head ->
|
|
|
|
(* 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 head >>= fun () ->
|
|
test_wrong_delegate ~miner:b1 b5 head >>= fun () ->
|
|
|
|
(* Endorse with a wrong slot : -1 and max (16) *)
|
|
test_invalid_endorsement_slot b3 (`Hash head) >>=? fun () ->
|
|
|
|
(* FIXME: Mining.Invalid_signature is still unclassified *)
|
|
test_invalid_signature (`Hash head) >>=? fun _ ->
|
|
|
|
(* FIXME: cannot inject double endorsement operation yet, but the
|
|
code is still here
|
|
Double endorsement *)
|
|
test_double_endorsement b4 (`Hash head) >>=? fun new_head ->
|
|
|
|
return new_head
|
|
|
|
let main () =
|
|
Helpers.init () >>=? fun (_node_pid, hash) ->
|
|
run (`Hash hash) Helpers.Account.bootstrap_accounts >>=? fun _blkh ->
|
|
return ()
|
|
|
|
|
|
let tests = [
|
|
"main", (fun _ -> main ()) ;
|
|
]
|
|
|
|
let () =
|
|
Test.run "endorsement." tests
|