mirror of
https://github.com/lightningnetwork/lnd.git
synced 2024-11-19 09:53:54 +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)
|
|
|
|
}
|