UmputunandGitHub
ba6bab5b91
* support flags commands, move to cmd * fix target name * test for happy path importer * add export cmd * fix wrong import, lint warns * increase test timeout * add sellp to allow main test server to start * implement all cmds * handle backup/restore errors * fix import status check, hide secret from logs * backup cmd err tests * randimize test port * avoid dup code in Last controller * add target to make all bin archives * remove container in make * add smiple scripts to simplify commands, update readme * add docs on dockerless, enforce app user * add restore info * move last to lastCommentsScope const
72 lines
2.3 KiB
Go
72 lines
2.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
flags "github.com/jessevdk/go-flags"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBackup_Execute(t *testing.T) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, r.URL.Path, "/api/v1/admin/export")
|
|
assert.Equal(t, "GET", r.Method)
|
|
fmt.Fprint(w, "blah\nblah2\n12345678\n")
|
|
}))
|
|
defer ts.Close()
|
|
|
|
cmd := BackupCommand{}
|
|
p := flags.NewParser(&cmd, flags.Default)
|
|
_, err := p.ParseArgs([]string{"--secret=123456", "--site=remark", "--path=/tmp",
|
|
"--file={{.SITE}}-test.export", "--url=" + ts.URL})
|
|
require.Nil(t, err)
|
|
err = cmd.Execute(nil)
|
|
assert.NoError(t, err)
|
|
defer os.Remove("/tmp/remark-test.export")
|
|
|
|
data, err := ioutil.ReadFile("/tmp/remark-test.export")
|
|
require.Nil(t, err)
|
|
assert.Equal(t, "blah\nblah2\n12345678\n", string(data))
|
|
}
|
|
|
|
func TestBackup_ExecuteFailedStatus(t *testing.T) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, r.URL.Path, "/api/v1/admin/export")
|
|
assert.Equal(t, "GET", r.Method)
|
|
w.WriteHeader(400)
|
|
fmt.Fprint(w, "some error")
|
|
}))
|
|
defer ts.Close()
|
|
|
|
cmd := BackupCommand{}
|
|
p := flags.NewParser(&cmd, flags.Default)
|
|
_, err := p.ParseArgs([]string{"--secret=123456", "--site=remark", "--path=/tmp",
|
|
"--file={{.SITE}}-test.export", "--url=" + ts.URL})
|
|
require.Nil(t, err)
|
|
err = cmd.Execute(nil)
|
|
assert.EqualError(t, err, `error response "400 Bad Request", some error`)
|
|
}
|
|
|
|
func TestBackup_ExecuteFailedWrite(t *testing.T) {
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, r.URL.Path, "/api/v1/admin/export")
|
|
assert.Equal(t, "GET", r.Method)
|
|
fmt.Fprint(w, "blah\nblah2\n12345678\n")
|
|
}))
|
|
defer ts.Close()
|
|
|
|
cmd := BackupCommand{}
|
|
p := flags.NewParser(&cmd, flags.Default)
|
|
_, err := p.ParseArgs([]string{"--secret=123456", "--site=remark", "--path=/tmp",
|
|
"--file=/tmp/no-such-dir/{{.SITE}}-test.export", "--url=" + ts.URL})
|
|
require.Nil(t, err)
|
|
err = cmd.Execute(nil)
|
|
assert.EqualError(t, err, `can't create backup file /tmp/no-such-dir/remark-test.export: open /tmp/no-such-dir/remark-test.export: no such file or directory`)
|
|
}
|