Implement CSV file for watched addresses

cerc-io · Jun 27, 2022 · 09af53b · 09af53b
1 parent 0da954b
commit 09af53b
Show file tree

Hide file tree

Showing 12 changed files with 737 additions and 193 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -1,4 +1,4 @@
-version: '3.2'
+version: "3.2"
 
 services:
   migrations:

diff --git a/statediff/indexer/database/file/config.go b/statediff/indexer/database/file/config.go
@@ -64,7 +64,7 @@ var TestConfig = Config{
 	Mode:                     CSV,
 	OutputDir:                "./statediffing_test",
 	FilePath:                 "./statediffing_test_file.sql",
-	WatchedAddressesFilePath: "./statediffing_watched_addresses_test_file.sql",
+	WatchedAddressesFilePath: "./statediffing_watched_addresses_test_file.csv",
 	NodeInfo: node.Info{
 		GenesisBlock: "0xd4e56740f876aef8c010b86a40d5f56745a118d0906a34e69aec8c0db1cb8fa3",
 		NetworkID:    "1",

diff --git a/statediff/indexer/database/file/csv_indexer_legacy_test.go b/statediff/indexer/database/file/csv_indexer_legacy_test.go
@@ -33,9 +33,11 @@ import (
 	"github.com/ethereum/go-ethereum/statediff/indexer/database/sql/postgres"
 	"github.com/ethereum/go-ethereum/statediff/indexer/interfaces"
 	"github.com/ethereum/go-ethereum/statediff/indexer/ipld"
+	"github.com/ethereum/go-ethereum/statediff/types"
 )
 
 const dbDirectory = "/file"
+const pgCopyStatement = `COPY %s FROM '%s' CSV`
 
 func setupCSVLegacy(t *testing.T) {
 	mockLegacyBlock = legacyData.MockBlock
@@ -81,7 +83,6 @@ func setupCSVLegacy(t *testing.T) {
 }
 
 func dumpCSVFileData(t *testing.T) {
-	pgCopyStatement := `COPY %s FROM '%s' CSV`
 	outputDir := filepath.Join(dbDirectory, file.TestConfig.OutputDir)
 
 	for _, tbl := range file.Tables {
@@ -102,6 +103,14 @@ func dumpCSVFileData(t *testing.T) {
 	}
 }
 
+func dumpWatchedAddressesCSVFileData(t *testing.T) {
+	outputFilePath := filepath.Join(dbDirectory, file.TestConfig.WatchedAddressesFilePath)
+	stm := fmt.Sprintf(pgCopyStatement, types.TableWatchedAddresses.Name, outputFilePath)
+
+	_, err = sqlxdb.Exec(stm)
+	require.NoError(t, err)
+}
+
 func tearDownCSV(t *testing.T) {
 	file.TearDownDB(t, sqlxdb)