Skip to content

Commit

Permalink
replace interface calls
Browse files Browse the repository at this point in the history
  • Loading branch information
@patrick-ogrady
patrick-ogrady committed Nov 25, 2020
1 parent 7e944d7 commit c439d62
Show file tree
Hide file tree
Showing 15 changed files with 149 additions and 149 deletions.
50 changes: 25 additions & 25 deletions constructor/coordinator/coordinator_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -244,7 +244,7 @@ func TestProcess(t *testing.T) {
// all responses from the database and "write" transactions require a
// lock. While it would be possible to orchestrate these locks in this
// test, it is simpler to just use a "read" transaction.
dbTxFail := db.NewDatabaseTransaction(ctx, false)
dbTxFail := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxFail).Once()
jobStorage.On("Ready", ctx, dbTxFail).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTxFail, "transfer").Return([]*job.Job{}, nil).Once()
Expand All @@ -261,7 +261,7 @@ func TestProcess(t *testing.T) {
// Determine account must be created
helper.On("HeadBlockExists", ctx).Return(true).Once()

dbTx := db.NewDatabaseTransaction(ctx, false)
dbTx := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx).Once()
jobStorage.On("Ready", ctx, dbTx).Return([]*job.Job{}, nil).Once()
jobStorage.On("Broadcasting", ctx, dbTx).Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -292,7 +292,7 @@ func TestProcess(t *testing.T) {

// Attempt to run transfer again (but determine funds are needed)
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTxFail2 := db.NewDatabaseTransaction(ctx, false)
dbTxFail2 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxFail2).Once()
jobStorage.On("Ready", ctx, dbTxFail2).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTxFail2, "transfer").Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -323,7 +323,7 @@ func TestProcess(t *testing.T) {
// Attempt funds request
helper.On("HeadBlockExists", ctx).Return(true).Once()

dbTx2 := db.NewDatabaseTransaction(ctx, false)
dbTx2 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx2).Once()
jobStorage.On("Ready", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
jobStorage.On("Broadcasting", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -370,7 +370,7 @@ func TestProcess(t *testing.T) {

// Load funds
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTxExtra := db.NewDatabaseTransaction(ctx, false)
dbTxExtra := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxExtra).Once()
jobStorage.On("Ready", ctx, dbTxExtra).Return([]*job.Job{&jobExtra}, nil).Once()
helper.On("AllAccounts", ctx, dbTxExtra).Return([]*types.AccountIdentifier{
Expand Down Expand Up @@ -403,7 +403,7 @@ func TestProcess(t *testing.T) {

// Attempt to transfer again
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTxFail3 := db.NewDatabaseTransaction(ctx, false)
dbTxFail3 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxFail3).Once()
jobStorage.On("Ready", ctx, dbTxFail3).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTxFail3, "transfer").Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -437,7 +437,7 @@ func TestProcess(t *testing.T) {

// Attempt to create recipient
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx3 := db.NewDatabaseTransaction(ctx, false)
dbTx3 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx3).Once()
jobStorage.On("Ready", ctx, dbTx3).Return([]*job.Job{}, nil).Once()
jobStorage.On("Broadcasting", ctx, dbTx3).Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -468,7 +468,7 @@ func TestProcess(t *testing.T) {

// Attempt to create transfer
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx4 := db.NewDatabaseTransaction(ctx, false)
dbTx4 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx4).Once()
jobStorage.On("Ready", ctx, dbTx4).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTx4, "transfer").Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -676,7 +676,7 @@ func TestProcess(t *testing.T) {

// Wait for transfer to complete
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx5 := db.NewDatabaseTransaction(ctx, false)
dbTx5 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx5).Once()
jobStorage.On("Ready", ctx, dbTx5).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTx5, "transfer").Return([]*job.Job{&job4}, nil).Once()
Expand Down Expand Up @@ -720,7 +720,7 @@ func TestProcess(t *testing.T) {
}
go func() {
<-markConfirmed
dbTx6 := db.NewDatabaseTransaction(ctx, false)
dbTx6 := db.ReadTransaction(ctx)
jobStorage.On("Get", ctx, dbTx6, "job4").Return(&job4, nil).Once()
jobStorage.On(
"Update",
Expand All @@ -745,7 +745,7 @@ func TestProcess(t *testing.T) {
}()

helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx7 := db.NewDatabaseTransaction(ctx, false)
dbTx7 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx7).Once()
jobStorage.On("Ready", ctx, dbTx7).Return([]*job.Job{&job4}, nil).Once()
jobStorage.On(
Expand Down Expand Up @@ -880,7 +880,7 @@ func TestProcess_Failed(t *testing.T) {

// Attempt to create transfer
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx := db.NewDatabaseTransaction(ctx, false)
dbTx := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx).Once()
jobStorage.On("Ready", ctx, dbTx).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTx, "transfer").Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -1147,7 +1147,7 @@ func TestProcess_Failed(t *testing.T) {

// Wait for transfer to complete
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx2 := db.NewDatabaseTransaction(ctx, false)
dbTx2 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx2).Once()
jobStorage.On("Ready", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTx2, "transfer").Return([]*job.Job{&j}, nil).Once()
Expand All @@ -1161,7 +1161,7 @@ func TestProcess_Failed(t *testing.T) {

go func() {
<-markConfirmed
dbTx3 := db.NewDatabaseTransaction(ctx, false)
dbTx3 := db.ReadTransaction(ctx)
jobStorage.On("Get", ctx, dbTx3, jobIdentifier).Return(&j, nil).Once()
jobStorage.On(
"Update",
Expand Down Expand Up @@ -1398,7 +1398,7 @@ func TestProcess_DryRun(t *testing.T) {
helper.On("HeadBlockExists", ctx).Return(true).Once()

// Attempt to transfer
dbTx := db.NewDatabaseTransaction(ctx, false)
dbTx := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx).Once()
jobStorage.On("Ready", ctx, dbTx).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTx, "transfer").Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -1509,7 +1509,7 @@ func TestProcess_DryRun(t *testing.T) {

// Process second scenario
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx2 := db.NewDatabaseTransaction(ctx, false)
dbTx2 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx2).Once()
jobStorage.On("Ready", ctx, dbTx2).Return([]*job.Job{&j}, nil).Once()
jobStorage.On("Update", ctx, dbTx2, mock.Anything).Run(func(args mock.Arguments) {
Expand Down Expand Up @@ -1683,7 +1683,7 @@ func TestReturnFunds_NoBalance(t *testing.T) {
helper.On("HeadBlockExists", ctx).Return(true).Once()

// Attempt to transfer
dbTxFail := db.NewDatabaseTransaction(ctx, false)
dbTxFail := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxFail).Once()
jobStorage.On("Ready", ctx, dbTxFail).Return([]*job.Job{}, nil).Once()
jobStorage.On(
Expand Down Expand Up @@ -1748,7 +1748,7 @@ func TestReturnFunds_NoBalance(t *testing.T) {

// Will exit this round because we've tried all workflows.
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx2 := db.NewDatabaseTransaction(ctx, false)
dbTx2 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx2).Once()
jobStorage.On("Ready", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
jobStorage.On("Broadcasting", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -2015,7 +2015,7 @@ func TestReturnFunds(t *testing.T) {
helper.On("HeadBlockExists", ctx).Return(true).Once()

// Attempt to transfer
dbTxFail := db.NewDatabaseTransaction(ctx, false)
dbTxFail := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxFail).Once()
jobStorage.On("Ready", ctx, dbTxFail).Return([]*job.Job{}, nil).Once()
jobStorage.On(
Expand Down Expand Up @@ -2214,7 +2214,7 @@ func TestReturnFunds(t *testing.T) {

// Wait for transfer to complete
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx := db.NewDatabaseTransaction(ctx, false)
dbTx := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx).Once()
jobStorage.On("Ready", ctx, dbTx).Return([]*job.Job{}, nil).Once()
jobStorage.On(
Expand Down Expand Up @@ -2266,7 +2266,7 @@ func TestReturnFunds(t *testing.T) {
}
go func() {
<-markConfirmed
dbTx2 := db.NewDatabaseTransaction(ctx, false)
dbTx2 := db.ReadTransaction(ctx)
jobStorage.On("Get", ctx, dbTx2, jobIdentifier).Return(&j, nil).Once()
jobStorage.On(
"Update",
Expand All @@ -2291,7 +2291,7 @@ func TestReturnFunds(t *testing.T) {

// No balance remaining
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx3 := db.NewDatabaseTransaction(ctx, false)
dbTx3 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx3).Once()
jobStorage.On("Ready", ctx, dbTx3).Return([]*job.Job{}, nil).Once()
jobStorage.On(
Expand Down Expand Up @@ -2329,7 +2329,7 @@ func TestReturnFunds(t *testing.T) {

// Will exit this round because we've tried all workflows.
helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx4 := db.NewDatabaseTransaction(ctx, false)
dbTx4 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx4).Once()
jobStorage.On("Ready", ctx, dbTx4).Return([]*job.Job{}, nil).Once()
jobStorage.On("Broadcasting", ctx, dbTx4).Return([]*job.Job{}, nil).Once()
Expand Down Expand Up @@ -2444,7 +2444,7 @@ func TestNoReservedWorkflows(t *testing.T) {
// all responses from the database and "write" transactions require a
// lock. While it would be possible to orchestrate these locks in this
// test, it is simpler to just use a "read" transaction.
dbTxFail := db.NewDatabaseTransaction(ctx, false)
dbTxFail := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTxFail).Once()
jobStorage.On("Ready", ctx, dbTxFail).Return([]*job.Job{}, nil).Once()
jobStorage.On("Processing", ctx, dbTxFail, "transfer").Return([]*job.Job{}, nil).Once()
Expand All @@ -2458,7 +2458,7 @@ func TestNoReservedWorkflows(t *testing.T) {
}()

helper.On("HeadBlockExists", ctx).Return(true).Once()
dbTx2 := db.NewDatabaseTransaction(ctx, false)
dbTx2 := db.ReadTransaction(ctx)
helper.On("DatabaseTransaction", ctx).Return(dbTx2).Once()
jobStorage.On("Ready", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
jobStorage.On("Broadcasting", ctx, dbTx2).Return([]*job.Job{}, nil).Once()
Expand Down
6 changes: 3 additions & 3 deletions constructor/worker/worker_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -963,7 +963,7 @@ func TestFindBalanceWorker(t *testing.T) {
assert.NotNil(t, db)
defer db.Close(ctx)

dbTx := db.NewDatabaseTransaction(ctx, true)
dbTx := db.Transaction(ctx)
defer dbTx.Discard(ctx)

worker := New(test.mockHelper)
Expand Down Expand Up @@ -1125,7 +1125,7 @@ func TestJob_ComplicatedTransfer(t *testing.T) {
assert.NotNil(t, db)
defer db.Close(ctx)

dbTx := db.NewDatabaseTransaction(ctx, true)
dbTx := db.Transaction(ctx)

network := &types.NetworkIdentifier{
Blockchain: "Bitcoin",
Expand Down Expand Up @@ -1643,7 +1643,7 @@ func TestJob_Failures(t *testing.T) {
assert.NotNil(t, db)
defer db.Close(ctx)

dbTx := db.NewDatabaseTransaction(ctx, true)
dbTx := db.Transaction(ctx)

assert.False(t, j.CheckComplete())

Expand Down
32 changes: 16 additions & 16 deletions storage/badger_storage_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ func TestDatabase(t *testing.T) {
defer database.Close(ctx)

t.Run("No key exists", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, false)
txn := database.ReadTransaction(ctx)
exists, value, err := txn.Get(ctx, []byte("hello"))
assert.False(t, exists)
assert.Nil(t, value)
Expand All @@ -68,14 +68,14 @@ func TestDatabase(t *testing.T) {
})

t.Run("Set key", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
err := txn.Set(ctx, []byte("hello"), []byte("hola"), true)
assert.NoError(t, err)
assert.NoError(t, txn.Commit(ctx))
})

t.Run("Get key", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, false)
txn := database.ReadTransaction(ctx)
exists, value, err := txn.Get(ctx, []byte("hello"))
assert.True(t, exists)
assert.Equal(t, []byte("hola"), value)
Expand All @@ -85,15 +85,15 @@ func TestDatabase(t *testing.T) {

t.Run("Many key set/get", func(t *testing.T) {
for i := 0; i < 1000; i++ {
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
k := []byte(fmt.Sprintf("blah/%d", i))
v := []byte(fmt.Sprintf("%d", i))
err := txn.Set(ctx, k, v, true)
assert.NoError(t, err)
assert.NoError(t, txn.Commit(ctx))

for j := 0; j <= i; j++ {
txn := database.NewDatabaseTransaction(ctx, false)
txn := database.ReadTransaction(ctx)
jk := []byte(fmt.Sprintf("blah/%d", j))
jv := []byte(fmt.Sprintf("%d", j))
exists, value, err := txn.Get(ctx, jk)
Expand All @@ -106,7 +106,7 @@ func TestDatabase(t *testing.T) {
})

t.Run("Scan", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
type scanItem struct {
Key []byte
Value []byte
Expand Down Expand Up @@ -175,11 +175,11 @@ func TestDatabaseTransaction(t *testing.T) {
defer database.Close(ctx)

t.Run("Set and get within a transaction", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
assert.NoError(t, txn.Set(ctx, []byte("hello"), []byte("hola"), true))

// Ensure tx does not affect db
txn2 := database.NewDatabaseTransaction(ctx, false)
txn2 := database.ReadTransaction(ctx)
exists, value, err := txn2.Get(ctx, []byte("hello"))
assert.False(t, exists)
assert.Nil(t, value)
Expand All @@ -188,7 +188,7 @@ func TestDatabaseTransaction(t *testing.T) {

assert.NoError(t, txn.Commit(ctx))

txn3 := database.NewDatabaseTransaction(ctx, false)
txn3 := database.ReadTransaction(ctx)
exists, value, err = txn3.Get(ctx, []byte("hello"))
assert.True(t, exists)
assert.Equal(t, []byte("hola"), value)
Expand All @@ -197,11 +197,11 @@ func TestDatabaseTransaction(t *testing.T) {
})

t.Run("Discard transaction", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
assert.NoError(t, txn.Set(ctx, []byte("hello"), []byte("world"), true))
txn.Discard(ctx)

txn2 := database.NewDatabaseTransaction(ctx, false)
txn2 := database.ReadTransaction(ctx)
exists, value, err := txn2.Get(ctx, []byte("hello"))
txn2.Discard(ctx)
assert.True(t, exists)
Expand All @@ -210,11 +210,11 @@ func TestDatabaseTransaction(t *testing.T) {
})

t.Run("Delete within a transaction", func(t *testing.T) {
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
assert.NoError(t, txn.Delete(ctx, []byte("hello")))
assert.NoError(t, txn.Commit(ctx))

txn2 := database.NewDatabaseTransaction(ctx, false)
txn2 := database.ReadTransaction(ctx)
exists, value, err := txn2.Get(ctx, []byte("hello"))
assert.False(t, exists)
assert.Nil(t, value)
Expand All @@ -239,7 +239,7 @@ func TestBadgerTrain_NoLimit(t *testing.T) {

// Load storage with entries in namespace
namespace := "bogus"
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
for i := 0; i < 10000; i++ {
entry := &BogusEntry{
Index: i,
Expand Down Expand Up @@ -282,7 +282,7 @@ func TestBadgerTrain_Limit(t *testing.T) {

// Load storage with entries in namespace
namespace := "bogus"
txn := database.NewDatabaseTransaction(ctx, true)
txn := database.Transaction(ctx)
for i := 0; i < 10000; i++ {
output, err := reggen.Generate(`[a-z]+`, 50)
assert.NoError(t, err)
Expand Down Expand Up @@ -334,7 +334,7 @@ func TestBadgerTrain_Limit(t *testing.T) {
)
assert.NoError(t, err)

txn2 := database2.NewDatabaseTransaction(ctx, true)
txn2 := database2.Transaction(ctx)
for i := 0; i < 10000; i++ {
output, err := reggen.Generate(`[a-z]+`, 50)
assert.NoError(t, err)
Expand Down

0 comments on commit c439d62

Please sign in to comment.