Resolve #88: Testing for get_code_hash

tests/integration/contracts.hpp.in

@@ -22,5 +22,10 @@ namespace eosio::testing {
 
  static std::vector<uint8_t> crypto_primitives_test_wasm() { return read_wasm("${CMAKE_BINARY_DIR}/../unit/test_contracts/crypto_primitives_tests.wasm"); }
  static std::vector<char> crypto_primitives_test_abi() { return read_abi("${CMAKE_BINARY_DIR}/../unit/test_contracts/crypto_primitives_tests.abi"); }
+
+ static std::vector<uint8_t> get_code_hash_write_test_wasm() { return read_wasm("${CMAKE_BINARY_DIR}/../unit/test_contracts/get_code_hash_write.wasm"); }
+ static std::vector<char> get_code_hash_write_test_abi() { return read_abi("${CMAKE_BINARY_DIR}/../unit/test_contracts/get_code_hash_write.abi"); }
+ static std::vector<uint8_t> get_code_hash_read_test_wasm() { return read_wasm("${CMAKE_BINARY_DIR}/../unit/test_contracts/get_code_hash_read.wasm"); }
+ static std::vector<char> get_code_hash_read_test_abi() { return read_abi("${CMAKE_BINARY_DIR}/../unit/test_contracts/get_code_hash_read.abi"); }
  };
 } //ns eosio::testing

tests/integration/get_code_hash_tests.cpp

@@ -0,0 +1,46 @@
+#include <boost/test/unit_test.hpp>
+#include <eosio/testing/tester.hpp>
+#include <eosio/chain/abi_serializer.hpp>
+
+#include <Runtime/Runtime.h>
+
+#include <fc/variant_object.hpp>
+
+#include <contracts.hpp>
+
+using namespace eosio;
+using namespace eosio::testing;
+using namespace eosio::chain;
+using namespace fc;
+
+using mvo = fc::mutable_variant_object;
+
+struct code_hash {
+ uint64_t id;
+ fc::sha256 hash;
+ uint64_t primary_key() const { return id; }
+};
+FC_REFLECT(code_hash, (id)(hash))
+
+BOOST_AUTO_TEST_SUITE(get_code_hash_tests_suite)
+
+BOOST_FIXTURE_TEST_CASE( get_code_hash_tests, tester ) try {
+ create_accounts( { "test"_n } );
+ produce_block();
+
+ set_code( "test"_n, contracts::get_code_hash_write_test_wasm() );
+ set_abi( "test"_n, contracts::get_code_hash_write_test_abi().data() );
+
+ produce_blocks();
+ push_action("test"_n, "theaction"_n, "test"_n, mvo());
+ code_hash entry;
+ get_table_entry(entry, "test"_n, "test"_n, "code.hash"_n, 0);
+ wdump((entry.hash));
+
+ set_code( "test"_n, contracts::get_code_hash_read_test_wasm() );
+ produce_blocks();
+
+ push_action("test"_n, "theaction"_n, "test"_n, mvo());
+} FC_LOG_AND_RETHROW()
+
+BOOST_AUTO_TEST_SUITE_END()

tests/unit/test_contracts/CMakeLists.txt

@@ -7,6 +7,8 @@ add_contract(explicit_nested_tests explicit_nested_tests explicit_nested_tests.c
 add_contract(transfer_contract transfer_contract transfer.cpp)
 add_contract(minimal_tests minimal_tests minimal_tests.cpp)
 add_contract(crypto_primitives_tests crypto_primitives_tests crypto_primitives_tests.cpp)
+add_contract(get_code_hash_tests get_code_hash_write get_code_hash_write.cpp)
+add_contract(get_code_hash_tests get_code_hash_read get_code_hash_read.cpp)
 add_contract(capi_tests capi_tests capi/capi.c capi/action.c capi/chain.c capi/crypto.c capi/db.c capi/permission.c
  capi/print.c capi/privileged.c capi/system.c capi/transaction.c)
 

tests/unit/test_contracts/get_code_hash_read.cpp

@@ -0,0 +1,27 @@
+#include <eosio/eosio.hpp>
+#include <eosio/action.hpp>
+#include <eosio/name.hpp>
+
+#include "get_code_hash_table.hpp"
+
+class [[eosio::contract]] get_code_hash_tests : public contract {
+public:
+ using contract::contract;
+
+ using hash_table = multi_index<name("code.hash"), code_hash>;
+
+ // Read the old code's hash from database and verify new code's hash differs
+ [[eosio::action]]
+ void theaction() {
+ require_auth(get_self());
+ hash_table hashes(get_self(), get_self().value);
+
+ auto hash = get_code_hash(get_self());
+ check(hash != checksum256(), "Code hash should not be null");
+
+ auto record = hashes.get(0, "Unable to find recorded hash");
+ check(hash != record.hash, "Code hash has not changed");
+ eosio::print("Old hash: ", record.hash, "; new hash: ", hash);
+ }
+};
+

tests/unit/test_contracts/get_code_hash_table.hpp

@@ -0,0 +1,10 @@
+#pragma once
+
+using namespace eosio;
+
+TABLE code_hash {
+ uint64_t id;
+ checksum256 hash;
+ uint64_t primary_key() const { return id; }
+};
+

tests/unit/test_contracts/get_code_hash_write.cpp

@@ -0,0 +1,28 @@
+#include <eosio/eosio.hpp>
+#include <eosio/action.hpp>
+#include <eosio/name.hpp>
+
+#include "get_code_hash_table.hpp"
+
+class [[eosio::contract]] get_code_hash_tests : public contract {
+public:
+ using contract::contract;
+
+ using hash_table = multi_index<name("code.hash"), code_hash>;
+
+ // Write this code's hash to database
+ [[eosio::action]]
+ void theaction() {
+ require_auth(get_self());
+ hash_table hashes(get_self(), get_self().value);
+
+ auto hash = get_code_hash(get_self());
+ check(hash != checksum256(), "Code hash should not be null");
+
+ hashes.emplace(get_self(), [&hash](auto& t) {
+ t.id = 0;
+ t.hash = hash;
+ });
+ }
+};
+