mirror of
https://github.com/sosedoff/pgweb.git
synced 2024-12-15 03:36:33 +03:00
4c40eef99a
* Add func to parse out pg_dump version * Perform client vs server version checking before dump exports * Fix dump tests * Add extra test to validate against empty server version * Fix attachment filenames cleanup function * Add extra test * Fix small typos in comments * Drop third-party package to deal with versions * Tweak the pg dump incompatibility error message * Run CI on pull requests
56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func testDumpExport(t *testing.T) {
|
|
url := fmt.Sprintf("postgres://%s@%s:%s/%s?sslmode=disable", serverUser, serverHost, serverPort, serverDatabase)
|
|
|
|
savePath := "/tmp/dump.sql.gz"
|
|
os.Remove(savePath)
|
|
|
|
saveFile, err := os.Create(savePath)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
|
|
defer func() {
|
|
saveFile.Close()
|
|
os.Remove(savePath)
|
|
}()
|
|
|
|
dump := Dump{}
|
|
|
|
// Test for pg_dump presence
|
|
assert.NoError(t, dump.Validate("10.0"))
|
|
assert.NoError(t, dump.Validate(""))
|
|
assert.Contains(t, dump.Validate("20").Error(), "not compatible with server version 20")
|
|
|
|
// Test full db dump
|
|
err = dump.Export(context.Background(), url, saveFile)
|
|
assert.NoError(t, err)
|
|
|
|
// Test nonexistent database
|
|
invalidURL := fmt.Sprintf("postgres://%s@%s:%s/%s?sslmode=disable", serverUser, serverHost, serverPort, "foobar")
|
|
err = dump.Export(context.Background(), invalidURL, saveFile)
|
|
assert.Contains(t, err.Error(), `database "foobar" does not exist`)
|
|
|
|
// Test dump of non existent db
|
|
dump = Dump{Table: "foobar"}
|
|
err = dump.Export(context.Background(), url, saveFile)
|
|
assert.NotNil(t, err)
|
|
assert.Contains(t, err.Error(), "no matching tables were found")
|
|
|
|
// Should drop "search_path" param from URI
|
|
dump = Dump{}
|
|
searchPathURL := fmt.Sprintf("postgres://%s@%s:%s/%s?sslmode=disable&search_path=private", serverUser, serverHost, serverPort, serverDatabase)
|
|
err = dump.Export(context.Background(), searchPathURL, saveFile)
|
|
assert.NoError(t, err)
|
|
}
|