|
7 | 7 | "fmt"
|
8 | 8 | "testing"
|
9 | 9 |
|
| 10 | + "github.com/stretchr/testify/require" |
| 11 | + |
10 | 12 | "github.com/filecoin-project/curio/harmony/harmonydb"
|
11 | 13 |
|
12 | 14 | "github.com/filecoin-project/lotus/itests/kit"
|
@@ -67,11 +69,12 @@ func TestTransaction(t *testing.T) {
|
67 | 69 |
|
68 | 70 | withSetup(t, func(miner *kit.TestMiner) {
|
69 | 71 | testID := harmonydb.ITestNewID()
|
70 |
| - cdb := setupTestDB(t, testID) |
| 72 | + cdb, err := harmonydb.NewFromConfigWithITestID(t, testID) |
| 73 | + require.NoError(t, err) |
71 | 74 | if _, err := cdb.Exec(ctx, "INSERT INTO itest_scratch (some_int) VALUES (4), (5), (6)"); err != nil {
|
72 | 75 | t.Fatal("E0", err)
|
73 | 76 | }
|
74 |
| - _, err := cdb.BeginTransaction(ctx, func(tx *harmonydb.Tx) (commit bool, err error) { |
| 77 | + _, err = cdb.BeginTransaction(ctx, func(tx *harmonydb.Tx) (commit bool, err error) { |
75 | 78 | if _, err := tx.Exec("INSERT INTO itest_scratch (some_int) VALUES (7), (8), (9)"); err != nil {
|
76 | 79 | t.Fatal("E1", err)
|
77 | 80 | }
|
@@ -129,7 +132,8 @@ func TestPartialWalk(t *testing.T) {
|
129 | 132 |
|
130 | 133 | withSetup(t, func(miner *kit.TestMiner) {
|
131 | 134 | testID := harmonydb.ITestNewID()
|
132 |
| - cdb := setupTestDB(t, testID) |
| 135 | + cdb, err := harmonydb.NewFromConfigWithITestID(t, testID) |
| 136 | + require.NoError(t, err) |
133 | 137 | if _, err := cdb.Exec(ctx, `
|
134 | 138 | INSERT INTO
|
135 | 139 | itest_scratch (content, some_int)
|
|
0 commit comments