cerc-io/go-ethereum
13
0
Fork 0
Code Issues 9 Pull Requests Packages Projects Releases 66 Wiki Activity

Add indexer unit tests for loading watched addresses

Browse Source
This commit is contained in:
Prathamesh Musale 2022-03-21 14:34:12 +05:30 committed by Abdul Rabbani
parent 319b13c44e
commit a47a90e9d3
3 changed files with 132 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
statediff/indexer/database/file/indexer_test.go
View File

@ -1107,6 +1107,18 @@ func TestFileWatchAddressMethods(t *testing.T) {
} }
pgStr := "SELECT * FROM eth_meta.watched_addresses" pgStr := "SELECT * FROM eth_meta.watched_addresses"
t.Run("Load watched addresses (empty table)", func(t *testing.T) {
expectedData := []common.Address{}
rows, err := ind.LoadWatchedAddresses()
require.NoError(t, err)
expectTrue(t, len(rows) == len(expectedData))
for idx, row := range rows {
test_helpers.ExpectEqual(t, row, expectedData[idx])
}
})
t.Run("Insert watched addresses", func(t *testing.T) { t.Run("Insert watched addresses", func(t *testing.T) {
args := []sdtypes.WatchAddressArg{ args := []sdtypes.WatchAddressArg{
{ {
@ -1133,7 +1145,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
}, },
} }
ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt1))) err = ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt1)))
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1180,7 +1193,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
}, },
} }
ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt2))) err = ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt2)))
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1215,7 +1229,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
}, },
} }
ind.RemoveWatchedAddresses(args) err = ind.RemoveWatchedAddresses(args)
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1243,7 +1258,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
} }
expectedData := []res{} expectedData := []res{}
ind.RemoveWatchedAddresses(args) err = ind.RemoveWatchedAddresses(args)
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1294,7 +1310,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
}, },
} }
ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt2))) err = ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt2)))
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1345,7 +1362,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
}, },
} }
ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt3))) err = ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt3)))
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1360,10 +1378,27 @@ func TestFileWatchAddressMethods(t *testing.T) {
} }
}) })
t.Run("Load watched addresses", func(t *testing.T) {
expectedData := []common.Address{
common.HexToAddress(contract4Address),
common.HexToAddress(contract2Address),
common.HexToAddress(contract3Address),
}
rows, err := ind.LoadWatchedAddresses()
require.NoError(t, err)
expectTrue(t, len(rows) == len(expectedData))
for idx, row := range rows {
test_helpers.ExpectEqual(t, row, expectedData[idx])
}
})
t.Run("Clear watched addresses", func(t *testing.T) { t.Run("Clear watched addresses", func(t *testing.T) {
expectedData := []res{} expectedData := []res{}
ind.ClearWatchedAddresses() err = ind.ClearWatchedAddresses()
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}
@ -1381,7 +1416,8 @@ func TestFileWatchAddressMethods(t *testing.T) {
t.Run("Clear watched addresses (empty table)", func(t *testing.T) { t.Run("Clear watched addresses (empty table)", func(t *testing.T) {
expectedData := []res{} expectedData := []res{}
ind.ClearWatchedAddresses() err = ind.ClearWatchedAddresses()
require.NoError(t, err)
dumpWatchedAddressesFileData(t) dumpWatchedAddressesFileData(t)
rows := []res{} rows := []res{}

52
statediff/indexer/database/sql/pgx_indexer_test.go
View File

@ -958,6 +958,18 @@ func TestPGXWatchAddressMethods(t *testing.T) {
} }
pgStr := "SELECT * FROM eth_meta.watched_addresses" pgStr := "SELECT * FROM eth_meta.watched_addresses"
t.Run("Load watched addresses (empty table)", func(t *testing.T) {
expectedData := []common.Address{}
rows, err := ind.LoadWatchedAddresses()
require.NoError(t, err)
expectTrue(t, len(rows) == len(expectedData))
for idx, row := range rows {
test_helpers.ExpectEqual(t, row, expectedData[idx])
}
})
t.Run("Insert watched addresses", func(t *testing.T) { t.Run("Insert watched addresses", func(t *testing.T) {
args := []sdtypes.WatchAddressArg{ args := []sdtypes.WatchAddressArg{
{ {
@ -984,7 +996,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt1))) err = ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt1)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1030,7 +1043,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt2))) err = ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt2)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1064,7 +1078,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.RemoveWatchedAddresses(args) err = ind.RemoveWatchedAddresses(args)
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1091,7 +1106,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
} }
expectedData := []res{} expectedData := []res{}
ind.RemoveWatchedAddresses(args) err = ind.RemoveWatchedAddresses(args)
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1141,7 +1157,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt2))) err = ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt2)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1191,7 +1208,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt3))) err = ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt3)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1205,10 +1223,27 @@ func TestPGXWatchAddressMethods(t *testing.T) {
} }
}) })
t.Run("Load watched addresses", func(t *testing.T) {
expectedData := []common.Address{
common.HexToAddress(contract4Address),
common.HexToAddress(contract2Address),
common.HexToAddress(contract3Address),
}
rows, err := ind.LoadWatchedAddresses()
require.NoError(t, err)
expectTrue(t, len(rows) == len(expectedData))
for idx, row := range rows {
test_helpers.ExpectEqual(t, row, expectedData[idx])
}
})
t.Run("Clear watched addresses", func(t *testing.T) { t.Run("Clear watched addresses", func(t *testing.T) {
expectedData := []res{} expectedData := []res{}
ind.ClearWatchedAddresses() err = ind.ClearWatchedAddresses()
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1225,7 +1260,8 @@ func TestPGXWatchAddressMethods(t *testing.T) {
t.Run("Clear watched addresses (empty table)", func(t *testing.T) { t.Run("Clear watched addresses (empty table)", func(t *testing.T) {
expectedData := []res{} expectedData := []res{}
ind.ClearWatchedAddresses() err = ind.ClearWatchedAddresses()
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)

52
statediff/indexer/database/sql/sqlx_indexer_test.go
View File

@ -941,6 +941,18 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
} }
pgStr := "SELECT * FROM eth_meta.watched_addresses" pgStr := "SELECT * FROM eth_meta.watched_addresses"
t.Run("Load watched addresses (empty table)", func(t *testing.T) {
expectedData := []common.Address{}
rows, err := ind.LoadWatchedAddresses()
require.NoError(t, err)
expectTrue(t, len(rows) == len(expectedData))
for idx, row := range rows {
test_helpers.ExpectEqual(t, row, expectedData[idx])
}
})
t.Run("Insert watched addresses", func(t *testing.T) { t.Run("Insert watched addresses", func(t *testing.T) {
args := []sdtypes.WatchAddressArg{ args := []sdtypes.WatchAddressArg{
{ {
@ -967,7 +979,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt1))) err = ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt1)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1013,7 +1026,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt2))) err = ind.InsertWatchedAddresses(args, big.NewInt(int64(watchedAt2)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1047,7 +1061,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.RemoveWatchedAddresses(args) err = ind.RemoveWatchedAddresses(args)
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1074,7 +1089,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
} }
expectedData := []res{} expectedData := []res{}
ind.RemoveWatchedAddresses(args) err = ind.RemoveWatchedAddresses(args)
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1124,7 +1140,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt2))) err = ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt2)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1174,7 +1191,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
}, },
} }
ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt3))) err = ind.SetWatchedAddresses(args, big.NewInt(int64(watchedAt3)))
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1188,10 +1206,27 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
} }
}) })
t.Run("Load watched addresses", func(t *testing.T) {
expectedData := []common.Address{
common.HexToAddress(contract4Address),
common.HexToAddress(contract2Address),
common.HexToAddress(contract3Address),
}
rows, err := ind.LoadWatchedAddresses()
require.NoError(t, err)
expectTrue(t, len(rows) == len(expectedData))
for idx, row := range rows {
test_helpers.ExpectEqual(t, row, expectedData[idx])
}
})
t.Run("Clear watched addresses", func(t *testing.T) { t.Run("Clear watched addresses", func(t *testing.T) {
expectedData := []res{} expectedData := []res{}
ind.ClearWatchedAddresses() err = ind.ClearWatchedAddresses()
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)
@ -1208,7 +1243,8 @@ func TestSQLXWatchAddressMethods(t *testing.T) {
t.Run("Clear watched addresses (empty table)", func(t *testing.T) { t.Run("Clear watched addresses (empty table)", func(t *testing.T) {
expectedData := []res{} expectedData := []res{}
ind.ClearWatchedAddresses() err = ind.ClearWatchedAddresses()
require.NoError(t, err)
rows := []res{} rows := []res{}
err = db.Select(context.Background(), &rows, pgStr) err = db.Select(context.Background(), &rows, pgStr)