mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 03:30:35 +00:00
4ff91bebf8
* Update gomock * Update mockgen * Gaz * Go mod * Cleanup * Regenerate gomock * Manually fix import
83 lines
3.6 KiB
Bash
Executable File
83 lines
3.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Script to update mock files after proto/prysm/v1alpha1/services.proto changes.
|
|
# Use a space to separate mock destination from its interfaces.
|
|
# Be sure to install mockgen before use: https://github.com/uber-go/mock
|
|
|
|
mock_path="testing/mock"
|
|
iface_mock_path="testing/validator-mock"
|
|
|
|
# github.com/prysmaticlabs/prysm/v5/proto/prysm/v1alpha1
|
|
# ------------------------------------------------------
|
|
proto_mocks_v1alpha1=(
|
|
"$mock_path/beacon_service_mock.go BeaconChainClient"
|
|
"$mock_path/beacon_validator_server_mock.go BeaconNodeValidatorServer,BeaconNodeValidator_WaitForActivationServer,BeaconNodeValidator_WaitForChainStartServer,BeaconNodeValidator_StreamSlotsServer"
|
|
"$mock_path/beacon_validator_client_mock.go BeaconNodeValidatorClient,BeaconNodeValidator_WaitForChainStartClient,BeaconNodeValidator_WaitForActivationClient,BeaconNodeValidator_StreamSlotsClient"
|
|
"$mock_path/node_service_mock.go NodeClient"
|
|
)
|
|
|
|
for ((i = 0; i < ${#proto_mocks_v1alpha1[@]}; i++)); do
|
|
file=${proto_mocks_v1alpha1[i]% *};
|
|
interfaces=${proto_mocks_v1alpha1[i]#* };
|
|
echo "generating $file for interfaces: $interfaces";
|
|
echo
|
|
GO11MODULE=on mockgen -package=mock -destination="$file" github.com/prysmaticlabs/prysm/v5/proto/prysm/v1alpha1 "$interfaces"
|
|
done
|
|
|
|
# github.com/prysmaticlabs/prysm/v5/validator/client/iface
|
|
# --------------------------------------------------------
|
|
iface_mocks=(
|
|
"$iface_mock_path/beacon_chain_client_mock.go BeaconChainClient"
|
|
"$iface_mock_path/prysm_beacon_chain_client_mock.go PrysmBeaconChainClient"
|
|
"$iface_mock_path/node_client_mock.go NodeClient"
|
|
"$iface_mock_path/validator_client_mock.go ValidatorClient"
|
|
)
|
|
|
|
for ((i = 0; i < ${#iface_mocks[@]}; i++)); do
|
|
file=${iface_mocks[i]% *};
|
|
interfaces=${iface_mocks[i]#* };
|
|
echo "generating $file for interfaces: $interfaces";
|
|
GO11MODULE=on mockgen -package=validator_mock -destination="$file" github.com/prysmaticlabs/prysm/v5/validator/client/iface "$interfaces"
|
|
done
|
|
|
|
goimports -w "$mock_path/."
|
|
gofmt -s -w "$mock_path/."
|
|
|
|
# github.com/prysmaticlabs/prysm/v5/validator/client/beacon-api
|
|
# -------------------------------------------------------------
|
|
beacon_api_mock_path="validator/client/beacon-api/mock"
|
|
beacon_api_mocks=(
|
|
"$beacon_api_mock_path/genesis_mock.go genesis.go"
|
|
"$beacon_api_mock_path/duties_mock.go duties.go"
|
|
"$beacon_api_mock_path/json_rest_handler_mock.go json_rest_handler.go"
|
|
"$beacon_api_mock_path/state_validators_mock.go state_validators.go"
|
|
"$beacon_api_mock_path/beacon_block_converter_mock.go beacon_block_converter.go"
|
|
)
|
|
|
|
for ((i = 0; i < ${#beacon_api_mocks[@]}; i++)); do
|
|
file=${beacon_api_mocks[i]% *};
|
|
source=${beacon_api_mocks[i]#* };
|
|
echo "generating $file for file: $source";
|
|
GO11MODULE=on mockgen -package=mock -source="validator/client/beacon-api/$source" -destination="$file"
|
|
done
|
|
|
|
goimports -w "$beacon_api_mock_path/."
|
|
gofmt -s -w "$beacon_api_mock_path/."
|
|
|
|
# github.com/prysmaticlabs/prysm/v5/crypto/bls
|
|
# --------------------------------------------
|
|
crypto_bls_common_mock_path="crypto/bls/common/mock"
|
|
crypto_bls_common_mocks=(
|
|
"$crypto_bls_common_mock_path/interface_mock.go interface.go"
|
|
)
|
|
|
|
for ((i = 0; i < ${#crypto_bls_common_mocks[@]}; i++)); do
|
|
file=${crypto_bls_common_mocks[i]% *};
|
|
source=${crypto_bls_common_mocks[i]#* };
|
|
echo "generating $file for file: $source";
|
|
GO11MODULE=on mockgen -package=mock -source="crypto/bls/common/$source" -destination="$file"
|
|
done
|
|
|
|
goimports -w "$crypto_bls_common_mock_path/."
|
|
gofmt -s -w "$crypto_bls_common_mock_path/."
|