package channeldb import ( "bytes" "crypto/sha256" "fmt" "math/rand" "reflect" "testing" "time" "github.com/davecgh/go-spew/spew" "github.com/roasbeef/btcutil" ) func makeFakePayment() *OutgoingPayment { fakeInvoice := &Invoice{ CreationDate: time.Now(), Memo: []byte("fake memo"), Receipt: []byte("fake receipt"), } copy(fakeInvoice.Terms.PaymentPreimage[:], rev[:]) fakeInvoice.Terms.Value = btcutil.Amount(10000) fakePath := make([][33]byte, 3) for i := 0; i < 3; i++ { copy(fakePath[i][:], bytes.Repeat([]byte{byte(i)}, 33)) } return &OutgoingPayment{ Invoice: *fakeInvoice, Fee: 101, Path: fakePath, TimeLockLength: 1000, PaymentHash: sha256.Sum256(rev[:]), } } // randomBytes creates random []byte with length in range [minLen, maxLen) func randomBytes(minLen, maxLen int) ([]byte, error) { randBuf := make([]byte, minLen+rand.Intn(maxLen-minLen)) if _, err := rand.Read(randBuf); err != nil { return nil, fmt.Errorf("Internal error. "+ "Cannot generate random string: %v", err) } return randBuf, nil } func makeRandomFakePayment() (*OutgoingPayment, error) { var err error fakeInvoice := &Invoice{ CreationDate: time.Now(), } fakeInvoice.Memo, err = randomBytes(1, 50) if err != nil { return nil, err } fakeInvoice.Receipt, err = randomBytes(1, 50) if err != nil { return nil, err } preImg, err := randomBytes(32, 33) if err != nil { return nil, err } copy(fakeInvoice.Terms.PaymentPreimage[:], preImg) fakeInvoice.Terms.Value = btcutil.Amount(rand.Intn(10000)) fakePathLen := 1 + rand.Intn(5) fakePath := make([][33]byte, fakePathLen) for i := 0; i < fakePathLen; i++ { b, err := randomBytes(33, 34) if err != nil { return nil, err } copy(fakePath[i][:], b) } rHash := sha256.Sum256(fakeInvoice.Terms.PaymentPreimage[:]) fakePayment := &OutgoingPayment{ Invoice: *fakeInvoice, Fee: btcutil.Amount(rand.Intn(1001)), Path: fakePath, TimeLockLength: uint32(rand.Intn(10000)), PaymentHash: rHash, } return fakePayment, nil } func TestOutgoingPaymentSerialization(t *testing.T) { fakePayment := makeFakePayment() var b bytes.Buffer if err := serializeOutgoingPayment(&b, fakePayment); err != nil { t.Fatalf("unable to serialize outgoing payment: %v", err) } newPayment, err := deserializeOutgoingPayment(&b) if err != nil { t.Fatalf("unable to deserialize outgoing payment: %v", err) } if !reflect.DeepEqual(fakePayment, newPayment) { t.Fatalf("Payments do not match after "+ "serialization/deserialization %v vs %v", spew.Sdump(fakePayment), spew.Sdump(newPayment), ) } } func TestOutgoingPaymentWorkflow(t *testing.T) { db, cleanUp, err := makeTestDB() defer cleanUp() if err != nil { t.Fatalf("unable to make test db: %v", err) } fakePayment := makeFakePayment() if err = db.AddPayment(fakePayment); err != nil { t.Fatalf("unable to put payment in DB: %v", err) } payments, err := db.FetchAllPayments() if err != nil { t.Fatalf("unable to fetch payments from DB: %v", err) } expectedPayments := []*OutgoingPayment{fakePayment} if !reflect.DeepEqual(payments, expectedPayments) { t.Fatalf("Wrong payments after reading from DB."+ "Got %v, want %v", spew.Sdump(payments), spew.Sdump(expectedPayments), ) } // Make some random payments for i := 0; i < 5; i++ { randomPayment, err := makeRandomFakePayment() if err != nil { t.Fatalf("Internal error in tests: %v", err) } if err = db.AddPayment(randomPayment); err != nil { t.Fatalf("unable to put payment in DB: %v", err) } expectedPayments = append(expectedPayments, randomPayment) } payments, err = db.FetchAllPayments() if err != nil { t.Fatalf("Can't get payments from DB: %v", err) } if !reflect.DeepEqual(payments, expectedPayments) { t.Fatalf("Wrong payments after reading from DB."+ "Got %v, want %v", spew.Sdump(payments), spew.Sdump(expectedPayments), ) } // Delete all payments. if err = db.DeleteAllPayments(); err != nil { t.Fatalf("unable to delete payments from DB: %v", err) } // Check that there is no payments after deletion paymentsAfterDeletion, err := db.FetchAllPayments() if err != nil { t.Fatalf("Can't get payments after deletion: %v", err) } if len(paymentsAfterDeletion) != 0 { t.Fatalf("After deletion DB has %v payments, want %v", len(paymentsAfterDeletion), 0) } }