mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-01-19 05:45:21 +01:00
35 lines
559 B
Go
35 lines
559 B
Go
|
package kvdb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RunTests is a helper function to run the tests in a package with
|
||
|
|
// initialization and tear-down of a test kvdb backend.
|
||
|
|
func RunTests(m *testing.M) {
|
||
|
|
var close func() error
|
||
|
|
if PostgresBackend {
|
||
|
|
var err error
|
||
|
|
close, err = StartEmbeddedPostgres()
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error: %v\n", err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// os.Exit() does not respect defer statements
|
||
|
|
code := m.Run()
|
||
|
|
|
||
|
|
if close != nil {
|
||
|
|
err := close()
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error: %v\n", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Exit(code)
|
||
|
|
|
||
|
|
}
|