You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

924 lines
26 KiB

4 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
4 years ago
Redo coordinator structure, connect API to node - API: - Modify the constructor so that hardcoded rollup constants don't need to be passed (introduce a `Config` and use `configAPI` internally) - Common: - Update rollup constants with proper *big.Int when required - Add BidCoordinator and Slot structs used by the HistoryDB and Synchronizer. - Add helper methods to AuctionConstants - AuctionVariables: Add column `DefaultSlotSetBidSlotNum` (in the SQL table: `default_slot_set_bid_slot_num`), which indicates at which slotNum does the `DefaultSlotSetBid` specified starts applying. - Config: - Move coordinator exclusive configuration from the node config to the coordinator config - Coordinator: - Reorganize the code towards having the goroutines started and stopped from the coordinator itself instead of the node. - Remove all stop and stopped channels, and use context.Context and sync.WaitGroup instead. - Remove BatchInfo setters and assing variables directly - In ServerProof and ServerProofPool use context instead stop channel. - Use message passing to notify the coordinator about sync updates and reorgs - Introduce the Pipeline, which can be started and stopped by the Coordinator - Introduce the TxManager, which manages ethereum transactions (the TxManager is also in charge of making the forge call to the rollup smart contract). The TxManager keeps ethereum transactions and: 1. Waits for the transaction to be accepted 2. Waits for the transaction to be confirmed for N blocks - In forge logic, first prepare a batch and then wait for an available server proof to have all work ready once the proof server is ready. - Remove the `isForgeSequence` method which was querying the smart contract, and instead use notifications sent by the Synchronizer to figure out if it's forging time. - Update test (which is a minimal test to manually see if the coordinator starts) - HistoryDB: - Add method to get the number of batches in a slot (used to detect when a slot has passed the bid winner forging deadline) - Add method to get the best bid and associated coordinator of a slot (used to detect the forgerAddress that can forge the slot) - General: - Rename some instances of `currentBlock` to `lastBlock` to be more clear. - Node: - Connect the API to the node and call the methods to update cached state when the sync advances blocks. - Call methods to update Coordinator state when the sync advances blocks and finds reorgs. - Synchronizer: - Add Auction field in the Stats, which contain the current slot with info about highest bidder and other related info required to know who can forge in the current block. - Better organization of cached state: - On Sync, update the internal cached state - On Init or Reorg, load the state from HistoryDB into the internal cached state.
4 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Redo coordinator structure, connect API to node - API: - Modify the constructor so that hardcoded rollup constants don't need to be passed (introduce a `Config` and use `configAPI` internally) - Common: - Update rollup constants with proper *big.Int when required - Add BidCoordinator and Slot structs used by the HistoryDB and Synchronizer. - Add helper methods to AuctionConstants - AuctionVariables: Add column `DefaultSlotSetBidSlotNum` (in the SQL table: `default_slot_set_bid_slot_num`), which indicates at which slotNum does the `DefaultSlotSetBid` specified starts applying. - Config: - Move coordinator exclusive configuration from the node config to the coordinator config - Coordinator: - Reorganize the code towards having the goroutines started and stopped from the coordinator itself instead of the node. - Remove all stop and stopped channels, and use context.Context and sync.WaitGroup instead. - Remove BatchInfo setters and assing variables directly - In ServerProof and ServerProofPool use context instead stop channel. - Use message passing to notify the coordinator about sync updates and reorgs - Introduce the Pipeline, which can be started and stopped by the Coordinator - Introduce the TxManager, which manages ethereum transactions (the TxManager is also in charge of making the forge call to the rollup smart contract). The TxManager keeps ethereum transactions and: 1. Waits for the transaction to be accepted 2. Waits for the transaction to be confirmed for N blocks - In forge logic, first prepare a batch and then wait for an available server proof to have all work ready once the proof server is ready. - Remove the `isForgeSequence` method which was querying the smart contract, and instead use notifications sent by the Synchronizer to figure out if it's forging time. - Update test (which is a minimal test to manually see if the coordinator starts) - HistoryDB: - Add method to get the number of batches in a slot (used to detect when a slot has passed the bid winner forging deadline) - Add method to get the best bid and associated coordinator of a slot (used to detect the forgerAddress that can forge the slot) - General: - Rename some instances of `currentBlock` to `lastBlock` to be more clear. - Node: - Connect the API to the node and call the methods to update cached state when the sync advances blocks. - Call methods to update Coordinator state when the sync advances blocks and finds reorgs. - Synchronizer: - Add Auction field in the Stats, which contain the current slot with info about highest bidder and other related info required to know who can forge in the current block. - Better organization of cached state: - On Sync, update the internal cached state - On Init or Reorg, load the state from HistoryDB into the internal cached state.
4 years ago
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Update coordinator, call all api update functions - Common: - Rename Block.EthBlockNum to Block.Num to avoid unneeded repetition - API: - Add UpdateNetworkInfoBlock to update just block information, to be used when the node is not yet synchronized - Node: - Call API.UpdateMetrics and UpdateRecommendedFee in a loop, with configurable time intervals - Synchronizer: - When mapping events by TxHash, use an array to support the possibility of multiple calls of the same function happening in the same transaction (for example, a smart contract in a single transaction could call withdraw with delay twice, which would generate 2 withdraw events, and 2 deposit events). - In Stats, keep entire LastBlock instead of just the blockNum - In Stats, add lastL1BatchBlock - Test Stats and SCVars - Coordinator: - Enable writing the BatchInfo in every step of the pipeline to disk (with JSON text files) for debugging purposes. - Move the Pipeline functionality from the Coordinator to its own struct (Pipeline) - Implement shouldL1lL2Batch - In TxManager, implement logic to perform several attempts when doing ethereum node RPC calls before considering the error. (Both for calls to forgeBatch and transaction receipt) - In TxManager, reorganize the flow and note the specific points in which actions are made when err != nil - HistoryDB: - Implement GetLastL1BatchBlockNum: returns the blockNum of the latest forged l1Batch, to help the coordinator decide when to forge an L1Batch. - EthereumClient and test.Client: - Update EthBlockByNumber to return the last block when the passed number is -1.
4 years ago
Allow serving API only via new cli command - Add new command to the cli/node: `serveapi` that alows serving the API just by connecting to the PostgreSQL database. The mode flag should me passed in order to select whether we are connecting to a synchronizer database or a coordinator database. If `coord` is chosen as mode, the coordinator endpoints can be activated in order to allow inserting l2txs and authorizations into the L2DB. Summary of the implementation details - New SQL table with 3 columns (plus `item_id` pk). The table only contains a single row with `item_id` = 1. Columns: - state: historydb.StateAPI in JSON. This is the struct that is served via the `/state` API endpoint. The node will periodically update this struct and store it int he DB. The api server will query it from the DB to serve it. - config: historydb.NodeConfig in JSON. This struct contains node configuration parameters that the API needs to be aware of. It's updated once every time the node starts. - constants: historydb.Constants in JSON. This struct contains all the hermez network constants gathered via the ethereum client by the node. It's written once every time the node starts. - The HistoryDB contains methods to get and update each one of these columns individually. - The HistoryDB contains all methods that query the DB and prepare objects that will appear in the StateAPI endpoint. - The configuration used in for the `serveapi` cli/node command is defined in `config.APIServer`, and is a subset of `node.Config` in order to allow reusing the same configuration file of the node if desired. - A new object is introduced in the api: `StateAPIUpdater`, which contains all the necessary information to update the StateAPI in the DB periodically by the node. - Moved the types `SCConsts`, `SCVariables` and `SCVariablesPtr` from `syncrhonizer` to `common` for convenience.
3 years ago
Update coordinator, call all api update functions - Common: - Rename Block.EthBlockNum to Block.Num to avoid unneeded repetition - API: - Add UpdateNetworkInfoBlock to update just block information, to be used when the node is not yet synchronized - Node: - Call API.UpdateMetrics and UpdateRecommendedFee in a loop, with configurable time intervals - Synchronizer: - When mapping events by TxHash, use an array to support the possibility of multiple calls of the same function happening in the same transaction (for example, a smart contract in a single transaction could call withdraw with delay twice, which would generate 2 withdraw events, and 2 deposit events). - In Stats, keep entire LastBlock instead of just the blockNum - In Stats, add lastL1BatchBlock - Test Stats and SCVars - Coordinator: - Enable writing the BatchInfo in every step of the pipeline to disk (with JSON text files) for debugging purposes. - Move the Pipeline functionality from the Coordinator to its own struct (Pipeline) - Implement shouldL1lL2Batch - In TxManager, implement logic to perform several attempts when doing ethereum node RPC calls before considering the error. (Both for calls to forgeBatch and transaction receipt) - In TxManager, reorganize the flow and note the specific points in which actions are made when err != nil - HistoryDB: - Implement GetLastL1BatchBlockNum: returns the blockNum of the latest forged l1Batch, to help the coordinator decide when to forge an L1Batch. - EthereumClient and test.Client: - Update EthBlockByNumber to return the last block when the passed number is -1.
4 years ago
Update coordinator, call all api update functions - Common: - Rename Block.EthBlockNum to Block.Num to avoid unneeded repetition - API: - Add UpdateNetworkInfoBlock to update just block information, to be used when the node is not yet synchronized - Node: - Call API.UpdateMetrics and UpdateRecommendedFee in a loop, with configurable time intervals - Synchronizer: - When mapping events by TxHash, use an array to support the possibility of multiple calls of the same function happening in the same transaction (for example, a smart contract in a single transaction could call withdraw with delay twice, which would generate 2 withdraw events, and 2 deposit events). - In Stats, keep entire LastBlock instead of just the blockNum - In Stats, add lastL1BatchBlock - Test Stats and SCVars - Coordinator: - Enable writing the BatchInfo in every step of the pipeline to disk (with JSON text files) for debugging purposes. - Move the Pipeline functionality from the Coordinator to its own struct (Pipeline) - Implement shouldL1lL2Batch - In TxManager, implement logic to perform several attempts when doing ethereum node RPC calls before considering the error. (Both for calls to forgeBatch and transaction receipt) - In TxManager, reorganize the flow and note the specific points in which actions are made when err != nil - HistoryDB: - Implement GetLastL1BatchBlockNum: returns the blockNum of the latest forged l1Batch, to help the coordinator decide when to forge an L1Batch. - EthereumClient and test.Client: - Update EthBlockByNumber to return the last block when the passed number is -1.
4 years ago
  1. package api
  2. import (
  3. "context"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "io"
  8. "io/ioutil"
  9. "math/big"
  10. "net"
  11. "net/http"
  12. "os"
  13. "strconv"
  14. "sync"
  15. "testing"
  16. "time"
  17. ethCommon "github.com/ethereum/go-ethereum/common"
  18. swagger "github.com/getkin/kin-openapi/openapi3filter"
  19. "github.com/gin-gonic/gin"
  20. "github.com/hermeznetwork/hermez-node/api/stateapiupdater"
  21. "github.com/hermeznetwork/hermez-node/common"
  22. "github.com/hermeznetwork/hermez-node/db"
  23. "github.com/hermeznetwork/hermez-node/db/historydb"
  24. "github.com/hermeznetwork/hermez-node/db/l2db"
  25. "github.com/hermeznetwork/hermez-node/log"
  26. "github.com/hermeznetwork/hermez-node/test"
  27. "github.com/hermeznetwork/hermez-node/test/til"
  28. "github.com/hermeznetwork/hermez-node/test/txsets"
  29. "github.com/hermeznetwork/tracerr"
  30. "github.com/stretchr/testify/require"
  31. )
  32. // Pendinger is an interface that allows getting last returned item ID and PendingItems to be used for building fromItem
  33. // when testing paginated endpoints.
  34. type Pendinger interface {
  35. GetPending() (pendingItems, lastItemID uint64)
  36. Len() int
  37. New() Pendinger
  38. }
  39. const apiPort = "4010"
  40. const apiIP = "http://localhost:"
  41. const apiURL = apiIP + apiPort + "/v1/"
  42. var SetBlockchain = `
  43. Type: Blockchain
  44. AddToken(1)
  45. AddToken(2)
  46. AddToken(3)
  47. AddToken(4)
  48. AddToken(5)
  49. AddToken(6)
  50. AddToken(7)
  51. AddToken(8)
  52. > block
  53. // Coordinator accounts, Idxs: 256, 257
  54. CreateAccountCoordinator(0) Coord
  55. CreateAccountCoordinator(1) Coord
  56. // close Block:0, Batch:1
  57. > batch
  58. CreateAccountDeposit(0) A: 11100000000000000
  59. CreateAccountDeposit(1) C: 22222222200000000000
  60. CreateAccountCoordinator(0) C
  61. // close Block:0, Batch:2
  62. > batchL1
  63. // Expected balances:
  64. // Coord(0): 0, Coord(1): 0
  65. // C(0): 0
  66. CreateAccountDeposit(1) A: 33333333300000000000
  67. // close Block:0, Batch:3
  68. > batchL1
  69. // close Block:0, Batch:4
  70. > batchL1
  71. CreateAccountDepositTransfer(0) B-A: 44444444400000000000, 123444444400000000000
  72. // close Block:0, Batch:5
  73. > batchL1
  74. CreateAccountDeposit(0) D: 55555555500000000000
  75. // close Block:0, Batch:6
  76. > batchL1
  77. CreateAccountCoordinator(1) B
  78. Transfer(1) A-B: 11100000000000000 (2)
  79. Transfer(0) B-C: 22200000000000000 (3)
  80. // close Block:0, Batch:7
  81. > batchL1 // forge L1User{1}, forge L1Coord{2}, forge L2{2}
  82. Deposit(0) C: 66666666600000000000
  83. DepositTransfer(0) C-D: 77777777700000000000, 12377777700000000000
  84. Transfer(0) A-B: 33350000000000000 (111)
  85. Transfer(0) C-A: 44450000000000000 (222)
  86. Transfer(1) B-C: 55550000000000000 (123)
  87. Exit(0) A: 66650000000000000 (44)
  88. ForceTransfer(0) D-B: 77777700000000000
  89. ForceExit(0) B: 88888800000000000
  90. // close Block:0, Batch:8
  91. > batchL1
  92. > block
  93. Transfer(0) D-A: 99950000000000000 (77)
  94. Transfer(0) B-D: 12300000000000000 (55)
  95. // close Block:1, Batch:1
  96. > batchL1
  97. CreateAccountCoordinator(0) F
  98. CreateAccountCoordinator(0) G
  99. CreateAccountCoordinator(0) H
  100. CreateAccountCoordinator(0) I
  101. CreateAccountCoordinator(0) J
  102. CreateAccountCoordinator(0) K
  103. CreateAccountCoordinator(0) L
  104. CreateAccountCoordinator(0) M
  105. CreateAccountCoordinator(0) N
  106. CreateAccountCoordinator(0) O
  107. CreateAccountCoordinator(0) P
  108. CreateAccountCoordinator(5) G
  109. CreateAccountCoordinator(5) H
  110. CreateAccountCoordinator(5) I
  111. CreateAccountCoordinator(5) J
  112. CreateAccountCoordinator(5) K
  113. CreateAccountCoordinator(5) L
  114. CreateAccountCoordinator(5) M
  115. CreateAccountCoordinator(5) N
  116. CreateAccountCoordinator(5) O
  117. CreateAccountCoordinator(5) P
  118. CreateAccountCoordinator(2) G
  119. CreateAccountCoordinator(2) H
  120. CreateAccountCoordinator(2) I
  121. CreateAccountCoordinator(2) J
  122. CreateAccountCoordinator(2) K
  123. CreateAccountCoordinator(2) L
  124. CreateAccountCoordinator(2) M
  125. CreateAccountCoordinator(2) N
  126. CreateAccountCoordinator(2) O
  127. CreateAccountCoordinator(2) P
  128. > batch
  129. > block
  130. > batch
  131. > block
  132. > batch
  133. > block
  134. `
  135. type testCommon struct {
  136. blocks []common.Block
  137. tokens []historydb.TokenWithUSD
  138. batches []testBatch
  139. fullBatches []testFullBatch
  140. coordinators []historydb.CoordinatorAPI
  141. accounts []testAccount
  142. txs []testTx
  143. exits []testExit
  144. poolTxsToSend []testPoolTxSend
  145. poolTxsToReceive []testPoolTxReceive
  146. auths []testAuth
  147. router *swagger.Router
  148. bids []testBid
  149. slots []testSlot
  150. auctionVars common.AuctionVariables
  151. rollupVars common.RollupVariables
  152. wdelayerVars common.WDelayerVariables
  153. nextForgers []historydb.NextForgerAPI
  154. }
  155. var tc testCommon
  156. var config configAPI
  157. var api *API
  158. var stateAPIUpdater *stateapiupdater.Updater
  159. // TestMain initializes the API server, and fill HistoryDB and StateDB with fake data,
  160. // emulating the task of the synchronizer in order to have data to be returned
  161. // by the API endpoints that will be tested
  162. func TestMain(m *testing.M) {
  163. // Initializations
  164. // Swagger
  165. router := swagger.NewRouter().WithSwaggerFromFile("./swagger.yml")
  166. // HistoryDB
  167. pass := os.Getenv("POSTGRES_PASS")
  168. database, err := db.InitSQLDB(5432, "localhost", "hermez", pass, "hermez")
  169. if err != nil {
  170. panic(err)
  171. }
  172. apiConnCon := db.NewAPIConnectionController(1, time.Second)
  173. hdb := historydb.NewHistoryDB(database, database, apiConnCon)
  174. if err != nil {
  175. panic(err)
  176. }
  177. // L2DB
  178. l2DB := l2db.NewL2DB(database, database, 10, 1000, 0.0, 1000.0, 24*time.Hour, apiConnCon)
  179. test.WipeDB(l2DB.DB()) // this will clean HistoryDB and L2DB
  180. // Config (smart contract constants)
  181. chainID := uint16(0)
  182. _config := getConfigTest(chainID)
  183. config = configAPI{
  184. RollupConstants: *newRollupConstants(_config.RollupConstants),
  185. AuctionConstants: _config.AuctionConstants,
  186. WDelayerConstants: _config.WDelayerConstants,
  187. }
  188. // API
  189. apiGin := gin.Default()
  190. // Reset DB
  191. test.WipeDB(hdb.DB())
  192. constants := &historydb.Constants{
  193. SCConsts: common.SCConsts{
  194. Rollup: _config.RollupConstants,
  195. Auction: _config.AuctionConstants,
  196. WDelayer: _config.WDelayerConstants,
  197. },
  198. ChainID: chainID,
  199. HermezAddress: _config.HermezAddress,
  200. }
  201. if err := hdb.SetConstants(constants); err != nil {
  202. panic(err)
  203. }
  204. nodeConfig := &historydb.NodeConfig{
  205. MaxPoolTxs: 10,
  206. MinFeeUSD: 0,
  207. MaxFeeUSD: 10000000000,
  208. }
  209. if err := hdb.SetNodeConfig(nodeConfig); err != nil {
  210. panic(err)
  211. }
  212. api, err = NewAPI(
  213. true,
  214. true,
  215. apiGin,
  216. hdb,
  217. l2DB,
  218. )
  219. if err != nil {
  220. log.Error(err)
  221. panic(err)
  222. }
  223. // Start server
  224. listener, err := net.Listen("tcp", ":"+apiPort) //nolint:gosec
  225. if err != nil {
  226. panic(err)
  227. }
  228. server := &http.Server{Handler: apiGin}
  229. go func() {
  230. if err := server.Serve(listener); err != nil &&
  231. tracerr.Unwrap(err) != http.ErrServerClosed {
  232. panic(err)
  233. }
  234. }()
  235. // Generate blockchain data with til
  236. tcc := til.NewContext(chainID, common.RollupConstMaxL1UserTx)
  237. tilCfgExtra := til.ConfigExtra{
  238. BootCoordAddr: ethCommon.HexToAddress("0xE39fEc6224708f0772D2A74fd3f9055A90E0A9f2"),
  239. CoordUser: "Coord",
  240. }
  241. blocksData, err := tcc.GenerateBlocks(SetBlockchain)
  242. if err != nil {
  243. panic(err)
  244. }
  245. err = tcc.FillBlocksExtra(blocksData, &tilCfgExtra)
  246. if err != nil {
  247. panic(err)
  248. }
  249. err = tcc.FillBlocksForgedL1UserTxs(blocksData)
  250. if err != nil {
  251. panic(err)
  252. }
  253. AddAditionalInformation(blocksData)
  254. // Generate L2 Txs with til
  255. commonPoolTxs, err := tcc.GeneratePoolL2Txs(txsets.SetPoolL2MinimumFlow0)
  256. if err != nil {
  257. panic(err)
  258. }
  259. // Extract til generated data, and add it to HistoryDB
  260. var commonBlocks []common.Block
  261. var commonBatches []common.Batch
  262. var commonAccounts []common.Account
  263. var commonExitTree []common.ExitInfo
  264. var commonL1Txs []common.L1Tx
  265. var commonL2Txs []common.L2Tx
  266. // Add ETH token at the beginning of the array
  267. testTokens := []historydb.TokenWithUSD{}
  268. ethUSD := float64(500)
  269. ethNow := time.Now()
  270. testTokens = append(testTokens, historydb.TokenWithUSD{
  271. TokenID: test.EthToken.TokenID,
  272. EthBlockNum: test.EthToken.EthBlockNum,
  273. EthAddr: test.EthToken.EthAddr,
  274. Name: test.EthToken.Name,
  275. Symbol: test.EthToken.Symbol,
  276. Decimals: test.EthToken.Decimals,
  277. USD: &ethUSD,
  278. USDUpdate: &ethNow,
  279. })
  280. err = api.h.UpdateTokenValue(common.EmptyAddr, ethUSD)
  281. if err != nil {
  282. panic(err)
  283. }
  284. for _, block := range blocksData {
  285. // Insert block into HistoryDB
  286. // nolint reason: block is used as read only in the function
  287. if err := api.h.AddBlockSCData(&block); err != nil { //nolint:gosec
  288. log.Error(err)
  289. panic(err)
  290. }
  291. // Extract data
  292. commonBlocks = append(commonBlocks, block.Block)
  293. for i, tkn := range block.Rollup.AddedTokens {
  294. token := historydb.TokenWithUSD{
  295. TokenID: tkn.TokenID,
  296. EthBlockNum: tkn.EthBlockNum,
  297. EthAddr: tkn.EthAddr,
  298. Name: tkn.Name,
  299. Symbol: tkn.Symbol,
  300. Decimals: tkn.Decimals,
  301. }
  302. value := float64(i + 423)
  303. now := time.Now().UTC()
  304. token.USD = &value
  305. token.USDUpdate = &now
  306. // Set value in DB
  307. err = api.h.UpdateTokenValue(token.EthAddr, value)
  308. if err != nil {
  309. panic(err)
  310. }
  311. testTokens = append(testTokens, token)
  312. }
  313. // Set USD value for tokens in DB
  314. for _, batch := range block.Rollup.Batches {
  315. commonL2Txs = append(commonL2Txs, batch.L2Txs...)
  316. for i := range batch.CreatedAccounts {
  317. batch.CreatedAccounts[i].Nonce = common.Nonce(i)
  318. commonAccounts = append(commonAccounts, batch.CreatedAccounts[i])
  319. }
  320. commonBatches = append(commonBatches, batch.Batch)
  321. commonExitTree = append(commonExitTree, batch.ExitTree...)
  322. commonL1Txs = append(commonL1Txs, batch.L1UserTxs...)
  323. commonL1Txs = append(commonL1Txs, batch.L1CoordinatorTxs...)
  324. }
  325. }
  326. // Generate Coordinators and add them to HistoryDB
  327. const nCoords = 10
  328. commonCoords := test.GenCoordinators(nCoords, commonBlocks)
  329. // Update one coordinator to test behaviour when bidder address is repeated
  330. updatedCoordBlock := commonCoords[len(commonCoords)-1].EthBlockNum
  331. commonCoords = append(commonCoords, common.Coordinator{
  332. Bidder: commonCoords[0].Bidder,
  333. Forger: commonCoords[0].Forger,
  334. EthBlockNum: updatedCoordBlock,
  335. URL: commonCoords[0].URL + ".new",
  336. })
  337. if err := api.h.AddCoordinators(commonCoords); err != nil {
  338. panic(err)
  339. }
  340. // Test next forgers
  341. // Set auction vars
  342. // Slots 3 and 6 will have bids that will be invalidated because of minBid update
  343. // Slots 4 and 7 will have valid bids, the rest will be cordinator slots
  344. var slot3MinBid int64 = 3
  345. var slot4MinBid int64 = 4
  346. var slot6MinBid int64 = 6
  347. var slot7MinBid int64 = 7
  348. // First update will indicate how things behave from slot 0
  349. var defaultSlotSetBid [6]*big.Int = [6]*big.Int{
  350. big.NewInt(10), // Slot 0 min bid
  351. big.NewInt(10), // Slot 1 min bid
  352. big.NewInt(10), // Slot 2 min bid
  353. big.NewInt(slot3MinBid), // Slot 3 min bid
  354. big.NewInt(slot4MinBid), // Slot 4 min bid
  355. big.NewInt(10), // Slot 5 min bid
  356. }
  357. auctionVars := common.AuctionVariables{
  358. EthBlockNum: int64(2),
  359. DonationAddress: ethCommon.HexToAddress("0x1111111111111111111111111111111111111111"),
  360. DefaultSlotSetBid: defaultSlotSetBid,
  361. DefaultSlotSetBidSlotNum: 0,
  362. Outbidding: uint16(1),
  363. SlotDeadline: uint8(20),
  364. BootCoordinator: ethCommon.HexToAddress("0x1111111111111111111111111111111111111111"),
  365. BootCoordinatorURL: "https://boot.coordinator.io",
  366. ClosedAuctionSlots: uint16(10),
  367. OpenAuctionSlots: uint16(20),
  368. }
  369. if err := api.h.AddAuctionVars(&auctionVars); err != nil {
  370. panic(err)
  371. }
  372. // Last update in auction vars will indicate how things will behave from slot 5
  373. defaultSlotSetBid = [6]*big.Int{
  374. big.NewInt(10), // Slot 5 min bid
  375. big.NewInt(slot6MinBid), // Slot 6 min bid
  376. big.NewInt(slot7MinBid), // Slot 7 min bid
  377. big.NewInt(10), // Slot 8 min bid
  378. big.NewInt(10), // Slot 9 min bid
  379. big.NewInt(10), // Slot 10 min bid
  380. }
  381. auctionVars = common.AuctionVariables{
  382. EthBlockNum: int64(3),
  383. DonationAddress: ethCommon.HexToAddress("0x1111111111111111111111111111111111111111"),
  384. DefaultSlotSetBid: defaultSlotSetBid,
  385. DefaultSlotSetBidSlotNum: 5,
  386. Outbidding: uint16(1),
  387. SlotDeadline: uint8(20),
  388. BootCoordinator: ethCommon.HexToAddress("0x1111111111111111111111111111111111111111"),
  389. BootCoordinatorURL: "https://boot.coordinator.io",
  390. ClosedAuctionSlots: uint16(10),
  391. OpenAuctionSlots: uint16(20),
  392. }
  393. if err := api.h.AddAuctionVars(&auctionVars); err != nil {
  394. panic(err)
  395. }
  396. // Generate Bids and add them to HistoryDB
  397. bids := []common.Bid{}
  398. // Slot 1 and 2, no bids, wins boot coordinator
  399. // Slot 3, below what's going to be the minimum (wins boot coordinator)
  400. bids = append(bids, common.Bid{
  401. SlotNum: 3,
  402. BidValue: big.NewInt(slot3MinBid - 1),
  403. EthBlockNum: commonBlocks[0].Num,
  404. Bidder: commonCoords[0].Bidder,
  405. })
  406. // Slot 4, valid bid (wins bidder)
  407. bids = append(bids, common.Bid{
  408. SlotNum: 4,
  409. BidValue: big.NewInt(slot4MinBid),
  410. EthBlockNum: commonBlocks[0].Num,
  411. Bidder: commonCoords[0].Bidder,
  412. })
  413. // Slot 5 no bids, wins boot coordinator
  414. // Slot 6, below what's going to be the minimum (wins boot coordinator)
  415. bids = append(bids, common.Bid{
  416. SlotNum: 6,
  417. BidValue: big.NewInt(slot6MinBid - 1),
  418. EthBlockNum: commonBlocks[0].Num,
  419. Bidder: commonCoords[0].Bidder,
  420. })
  421. // Slot 7, valid bid (wins bidder)
  422. bids = append(bids, common.Bid{
  423. SlotNum: 7,
  424. BidValue: big.NewInt(slot7MinBid),
  425. EthBlockNum: commonBlocks[0].Num,
  426. Bidder: commonCoords[0].Bidder,
  427. })
  428. if err = api.h.AddBids(bids); err != nil {
  429. panic(err)
  430. }
  431. bootForger := historydb.NextForgerAPI{
  432. Coordinator: historydb.CoordinatorAPI{
  433. Forger: auctionVars.BootCoordinator,
  434. URL: auctionVars.BootCoordinatorURL,
  435. },
  436. }
  437. // Set next forgers: set all as boot coordinator then replace the non boot coordinators
  438. nextForgers := []historydb.NextForgerAPI{}
  439. var initBlock int64 = 140
  440. var deltaBlocks int64 = 40
  441. for i := 1; i < int(auctionVars.ClosedAuctionSlots)+2; i++ {
  442. fromBlock := initBlock + deltaBlocks*int64(i-1)
  443. bootForger.Period = historydb.Period{
  444. SlotNum: int64(i),
  445. FromBlock: fromBlock,
  446. ToBlock: fromBlock + deltaBlocks - 1,
  447. }
  448. nextForgers = append(nextForgers, bootForger)
  449. }
  450. // Set next forgers that aren't the boot coordinator
  451. nonBootForger := historydb.CoordinatorAPI{
  452. Bidder: commonCoords[0].Bidder,
  453. Forger: commonCoords[0].Forger,
  454. URL: commonCoords[0].URL + ".new",
  455. }
  456. // Slot 4
  457. nextForgers[3].Coordinator = nonBootForger
  458. // Slot 7
  459. nextForgers[6].Coordinator = nonBootForger
  460. var buckets [common.RollupConstNumBuckets]common.BucketParams
  461. for i := range buckets {
  462. buckets[i].CeilUSD = big.NewInt(int64(i) * 10)
  463. buckets[i].Withdrawals = big.NewInt(int64(i) * 100)
  464. buckets[i].BlockWithdrawalRate = big.NewInt(int64(i) * 1000)
  465. buckets[i].MaxWithdrawals = big.NewInt(int64(i) * 10000)
  466. }
  467. // Generate SC vars and add them to HistoryDB (if needed)
  468. rollupVars := common.RollupVariables{
  469. EthBlockNum: int64(3),
  470. FeeAddToken: big.NewInt(100),
  471. ForgeL1L2BatchTimeout: int64(44),
  472. WithdrawalDelay: uint64(3000),
  473. Buckets: buckets,
  474. SafeMode: false,
  475. }
  476. wdelayerVars := common.WDelayerVariables{
  477. WithdrawalDelay: uint64(3000),
  478. }
  479. stateAPIUpdater = stateapiupdater.NewUpdater(hdb, nodeConfig, &common.SCVariables{
  480. Rollup: rollupVars,
  481. Auction: auctionVars,
  482. WDelayer: wdelayerVars,
  483. }, constants)
  484. // Generate test data, as expected to be received/sended from/to the API
  485. testCoords := genTestCoordinators(commonCoords)
  486. testBids := genTestBids(commonBlocks, testCoords, bids)
  487. testExits := genTestExits(commonExitTree, testTokens, commonAccounts)
  488. testTxs := genTestTxs(commonL1Txs, commonL2Txs, commonAccounts, testTokens, commonBlocks)
  489. testBatches, testFullBatches := genTestBatches(commonBlocks, commonBatches, testTxs)
  490. poolTxsToSend, poolTxsToReceive := genTestPoolTxs(commonPoolTxs, testTokens, commonAccounts)
  491. // Add balance and nonce to historyDB
  492. accounts := genTestAccounts(commonAccounts, testTokens)
  493. accUpdates := []common.AccountUpdate{}
  494. for i := 0; i < len(accounts); i++ {
  495. balance := new(big.Int)
  496. balance.SetString(string(*accounts[i].Balance), 10)
  497. idx, err := stringToIdx(string(accounts[i].Idx), "foo")
  498. if err != nil {
  499. panic(err)
  500. }
  501. accUpdates = append(accUpdates, common.AccountUpdate{
  502. EthBlockNum: 0,
  503. BatchNum: 1,
  504. Idx: *idx,
  505. Nonce: 0,
  506. Balance: balance,
  507. })
  508. accUpdates = append(accUpdates, common.AccountUpdate{
  509. EthBlockNum: 0,
  510. BatchNum: 1,
  511. Idx: *idx,
  512. Nonce: accounts[i].Nonce,
  513. Balance: balance,
  514. })
  515. }
  516. if err := api.h.AddAccountUpdates(accUpdates); err != nil {
  517. panic(err)
  518. }
  519. tc = testCommon{
  520. blocks: commonBlocks,
  521. tokens: testTokens,
  522. batches: testBatches,
  523. fullBatches: testFullBatches,
  524. coordinators: testCoords,
  525. accounts: accounts,
  526. txs: testTxs,
  527. exits: testExits,
  528. poolTxsToSend: poolTxsToSend,
  529. poolTxsToReceive: poolTxsToReceive,
  530. auths: genTestAuths(test.GenAuths(5, _config.ChainID, _config.HermezAddress)),
  531. router: router,
  532. bids: testBids,
  533. slots: api.genTestSlots(
  534. 20,
  535. commonBlocks[len(commonBlocks)-1].Num,
  536. testBids,
  537. auctionVars,
  538. ),
  539. auctionVars: auctionVars,
  540. rollupVars: rollupVars,
  541. wdelayerVars: wdelayerVars,
  542. nextForgers: nextForgers,
  543. }
  544. // Run tests
  545. result := m.Run()
  546. // Fake server
  547. if os.Getenv("FAKE_SERVER") == "yes" {
  548. for {
  549. log.Info("Running fake server at " + apiURL + " until ^C is received")
  550. time.Sleep(30 * time.Second)
  551. }
  552. }
  553. // Stop server
  554. if err := server.Shutdown(context.Background()); err != nil {
  555. panic(err)
  556. }
  557. if err := database.Close(); err != nil {
  558. panic(err)
  559. }
  560. os.Exit(result)
  561. }
  562. func TestTimeout(t *testing.T) {
  563. pass := os.Getenv("POSTGRES_PASS")
  564. databaseTO, err := db.ConnectSQLDB(5432, "localhost", "hermez", pass, "hermez")
  565. require.NoError(t, err)
  566. apiConnConTO := db.NewAPIConnectionController(1, 100*time.Millisecond)
  567. hdbTO := historydb.NewHistoryDB(databaseTO, databaseTO, apiConnConTO)
  568. require.NoError(t, err)
  569. // L2DB
  570. l2DBTO := l2db.NewL2DB(databaseTO, databaseTO, 10, 1000, 1.0, 1000.0, 24*time.Hour, apiConnConTO)
  571. // API
  572. apiGinTO := gin.Default()
  573. finishWait := make(chan interface{})
  574. startWait := make(chan interface{})
  575. apiGinTO.GET("/v1/wait", func(c *gin.Context) {
  576. cancel, err := apiConnConTO.Acquire()
  577. defer cancel()
  578. require.NoError(t, err)
  579. defer apiConnConTO.Release()
  580. startWait <- nil
  581. <-finishWait
  582. })
  583. // Start server
  584. serverTO := &http.Server{Handler: apiGinTO}
  585. listener, err := net.Listen("tcp", ":4444") //nolint:gosec
  586. require.NoError(t, err)
  587. go func() {
  588. if err := serverTO.Serve(listener); err != nil &&
  589. tracerr.Unwrap(err) != http.ErrServerClosed {
  590. require.NoError(t, err)
  591. }
  592. }()
  593. _, err = NewAPI(
  594. true,
  595. true,
  596. apiGinTO,
  597. hdbTO,
  598. l2DBTO,
  599. )
  600. require.NoError(t, err)
  601. client := &http.Client{}
  602. httpReq, err := http.NewRequest("GET", "http://localhost:4444/v1/tokens", nil)
  603. require.NoError(t, err)
  604. httpReqWait, err := http.NewRequest("GET", "http://localhost:4444/v1/wait", nil)
  605. require.NoError(t, err)
  606. // Request that will get timed out
  607. var wg sync.WaitGroup
  608. wg.Add(1)
  609. go func() {
  610. // Request that will make the API busy
  611. _, err = client.Do(httpReqWait)
  612. require.NoError(t, err)
  613. wg.Done()
  614. }()
  615. <-startWait
  616. resp, err := client.Do(httpReq)
  617. require.NoError(t, err)
  618. require.Equal(t, http.StatusServiceUnavailable, resp.StatusCode)
  619. defer resp.Body.Close() //nolint
  620. body, err := ioutil.ReadAll(resp.Body)
  621. require.NoError(t, err)
  622. // Unmarshal body into return struct
  623. msg := &errorMsg{}
  624. err = json.Unmarshal(body, msg)
  625. require.NoError(t, err)
  626. // Check that the error was the expected down
  627. require.Equal(t, errSQLTimeout, msg.Message)
  628. finishWait <- nil
  629. // Stop server
  630. wg.Wait()
  631. require.NoError(t, serverTO.Shutdown(context.Background()))
  632. require.NoError(t, databaseTO.Close())
  633. }
  634. func doGoodReqPaginated(
  635. path, order string,
  636. iterStruct Pendinger,
  637. appendIter func(res interface{}),
  638. ) error {
  639. var next uint64
  640. firstIte := true
  641. expectedTotal := 0
  642. totalReceived := 0
  643. for {
  644. // Calculate fromItem
  645. iterPath := path
  646. if !firstIte {
  647. iterPath += "&fromItem=" + strconv.Itoa(int(next))
  648. }
  649. // Call API to get this iteration items
  650. iterStruct = iterStruct.New()
  651. if err := doGoodReq(
  652. "GET", iterPath+"&order="+order, nil,
  653. iterStruct,
  654. ); err != nil {
  655. return tracerr.Wrap(err)
  656. }
  657. appendIter(iterStruct)
  658. // Keep iterating?
  659. remaining, lastID := iterStruct.GetPending()
  660. if remaining == 0 {
  661. break
  662. }
  663. if order == historydb.OrderDesc {
  664. next = lastID - 1
  665. } else {
  666. next = lastID + 1
  667. }
  668. // Check that the expected amount of items is consistent across iterations
  669. totalReceived += iterStruct.Len()
  670. if firstIte {
  671. firstIte = false
  672. expectedTotal = totalReceived + int(remaining)
  673. }
  674. if expectedTotal != totalReceived+int(remaining) {
  675. panic(fmt.Sprintf(
  676. "pagination error, totalReceived + remaining should be %d, but is %d",
  677. expectedTotal, totalReceived+int(remaining),
  678. ))
  679. }
  680. }
  681. return nil
  682. }
  683. func doGoodReq(method, path string, reqBody io.Reader, returnStruct interface{}) error {
  684. ctx := context.Background()
  685. client := &http.Client{}
  686. httpReq, err := http.NewRequest(method, path, reqBody)
  687. if err != nil {
  688. return tracerr.Wrap(err)
  689. }
  690. if reqBody != nil {
  691. httpReq.Header.Add("Content-Type", "application/json")
  692. }
  693. route, pathParams, err := tc.router.FindRoute(httpReq.Method, httpReq.URL)
  694. if err != nil {
  695. return tracerr.Wrap(err)
  696. }
  697. // Validate request against swagger spec
  698. requestValidationInput := &swagger.RequestValidationInput{
  699. Request: httpReq,
  700. PathParams: pathParams,
  701. Route: route,
  702. }
  703. if err := swagger.ValidateRequest(ctx, requestValidationInput); err != nil {
  704. return tracerr.Wrap(err)
  705. }
  706. // Do API call
  707. resp, err := client.Do(httpReq)
  708. if err != nil {
  709. return tracerr.Wrap(err)
  710. }
  711. if resp.Body == nil && returnStruct != nil {
  712. return tracerr.Wrap(errors.New("Nil body"))
  713. }
  714. //nolint
  715. defer resp.Body.Close()
  716. body, err := ioutil.ReadAll(resp.Body)
  717. if err != nil {
  718. return tracerr.Wrap(err)
  719. }
  720. if resp.StatusCode != 200 {
  721. return tracerr.Wrap(fmt.Errorf("%d response. Body: %s", resp.StatusCode, string(body)))
  722. }
  723. if returnStruct == nil {
  724. return nil
  725. }
  726. // Unmarshal body into return struct
  727. if err := json.Unmarshal(body, returnStruct); err != nil {
  728. log.Error("invalid json: " + string(body))
  729. log.Error(err)
  730. return tracerr.Wrap(err)
  731. }
  732. // log.Info(string(body))
  733. // Validate response against swagger spec
  734. responseValidationInput := &swagger.ResponseValidationInput{
  735. RequestValidationInput: requestValidationInput,
  736. Status: resp.StatusCode,
  737. Header: resp.Header,
  738. }
  739. responseValidationInput = responseValidationInput.SetBodyBytes(body)
  740. return swagger.ValidateResponse(ctx, responseValidationInput)
  741. }
  742. func doBadReq(method, path string, reqBody io.Reader, expectedResponseCode int) error {
  743. ctx := context.Background()
  744. client := &http.Client{}
  745. httpReq, _ := http.NewRequest(method, path, reqBody)
  746. route, pathParams, err := tc.router.FindRoute(httpReq.Method, httpReq.URL)
  747. if err != nil {
  748. return tracerr.Wrap(err)
  749. }
  750. // Validate request against swagger spec
  751. requestValidationInput := &swagger.RequestValidationInput{
  752. Request: httpReq,
  753. PathParams: pathParams,
  754. Route: route,
  755. }
  756. if err := swagger.ValidateRequest(ctx, requestValidationInput); err != nil {
  757. if expectedResponseCode != 400 {
  758. return tracerr.Wrap(err)
  759. }
  760. log.Warn("The request does not match the API spec")
  761. }
  762. // Do API call
  763. resp, err := client.Do(httpReq)
  764. if err != nil {
  765. return tracerr.Wrap(err)
  766. }
  767. if resp.Body == nil {
  768. return tracerr.Wrap(errors.New("Nil body"))
  769. }
  770. //nolint
  771. defer resp.Body.Close()
  772. body, err := ioutil.ReadAll(resp.Body)
  773. if err != nil {
  774. return tracerr.Wrap(err)
  775. }
  776. if resp.StatusCode != expectedResponseCode {
  777. return tracerr.Wrap(fmt.Errorf("Unexpected response code: %d. Body: %s", resp.StatusCode, string(body)))
  778. }
  779. // Validate response against swagger spec
  780. responseValidationInput := &swagger.ResponseValidationInput{
  781. RequestValidationInput: requestValidationInput,
  782. Status: resp.StatusCode,
  783. Header: resp.Header,
  784. }
  785. responseValidationInput = responseValidationInput.SetBodyBytes(body)
  786. return swagger.ValidateResponse(ctx, responseValidationInput)
  787. }
  788. func doSimpleReq(method, endpoint string) (string, error) {
  789. client := &http.Client{}
  790. httpReq, err := http.NewRequest(method, endpoint, nil)
  791. if err != nil {
  792. return "", tracerr.Wrap(err)
  793. }
  794. resp, err := client.Do(httpReq)
  795. if err != nil {
  796. return "", tracerr.Wrap(err)
  797. }
  798. //nolint
  799. defer resp.Body.Close()
  800. body, err := ioutil.ReadAll(resp.Body)
  801. if err != nil {
  802. return "", tracerr.Wrap(err)
  803. }
  804. return string(body), nil
  805. }
  806. // test helpers
  807. func getTimestamp(blockNum int64, blocks []common.Block) time.Time {
  808. for i := 0; i < len(blocks); i++ {
  809. if blocks[i].Num == blockNum {
  810. return blocks[i].Timestamp
  811. }
  812. }
  813. panic("timesamp not found")
  814. }
  815. func getTokenByID(id common.TokenID, tokens []historydb.TokenWithUSD) historydb.TokenWithUSD {
  816. for i := 0; i < len(tokens); i++ {
  817. if tokens[i].TokenID == id {
  818. return tokens[i]
  819. }
  820. }
  821. panic("token not found")
  822. }
  823. func getTokenByIdx(idx common.Idx, tokens []historydb.TokenWithUSD, accs []common.Account) historydb.TokenWithUSD {
  824. for _, acc := range accs {
  825. if idx == acc.Idx {
  826. return getTokenByID(acc.TokenID, tokens)
  827. }
  828. }
  829. panic("token not found")
  830. }
  831. func getAccountByIdx(idx common.Idx, accs []common.Account) *common.Account {
  832. for _, acc := range accs {
  833. if acc.Idx == idx {
  834. return &acc
  835. }
  836. }
  837. panic("account not found")
  838. }
  839. func getBlockByNum(ethBlockNum int64, blocks []common.Block) common.Block {
  840. for _, b := range blocks {
  841. if b.Num == ethBlockNum {
  842. return b
  843. }
  844. }
  845. panic("block not found")
  846. }
  847. func getCoordinatorByBidder(bidder ethCommon.Address, coordinators []historydb.CoordinatorAPI) historydb.CoordinatorAPI {
  848. var coordLastUpdate historydb.CoordinatorAPI
  849. found := false
  850. for _, c := range coordinators {
  851. if c.Bidder == bidder {
  852. coordLastUpdate = c
  853. found = true
  854. }
  855. }
  856. if !found {
  857. panic("coordinator not found")
  858. }
  859. return coordLastUpdate
  860. }