mirror of
https://github.com/lightningnetwork/lnd.git
synced 2024-11-19 09:53:54 +01:00
86 lines
1.5 KiB
Go
86 lines
1.5 KiB
Go
package contractcourt
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime/pprof"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
|
)
|
|
|
|
// timeout implements a test level timeout.
|
|
func timeout() func() {
|
|
done := make(chan struct{})
|
|
go func() {
|
|
select {
|
|
case <-time.After(5 * time.Second):
|
|
pprof.Lookup("goroutine").WriteTo(os.Stdout, 1)
|
|
|
|
panic("test timeout")
|
|
case <-done:
|
|
}
|
|
}()
|
|
|
|
return func() {
|
|
close(done)
|
|
}
|
|
}
|
|
|
|
func copyFile(dest, src string) error {
|
|
s, err := os.Open(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer s.Close()
|
|
|
|
d, err := os.Create(dest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := io.Copy(d, s); err != nil {
|
|
d.Close()
|
|
return err
|
|
}
|
|
|
|
return d.Close()
|
|
}
|
|
|
|
// copyChannelState copies the OpenChannel state by copying the database and
|
|
// creating a new struct from it. The copied state is returned.
|
|
func copyChannelState(t *testing.T, state *channeldb.OpenChannel) (
|
|
*channeldb.OpenChannel, error) {
|
|
|
|
// Make a copy of the DB.
|
|
dbFile := filepath.Join(state.Db.GetParentDB().Path(), "channel.db")
|
|
tempDbPath := t.TempDir()
|
|
|
|
tempDbFile := filepath.Join(tempDbPath, "channel.db")
|
|
err := copyFile(tempDbFile, dbFile)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
newDb, err := channeldb.Open(tempDbPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
chans, err := newDb.ChannelStateDB().FetchAllChannels()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// We only support DBs with a single channel, for now.
|
|
if len(chans) != 1 {
|
|
return nil, fmt.Errorf("found %d chans in the db",
|
|
len(chans))
|
|
}
|
|
|
|
return chans[0], nil
|
|
}
|