зеркало из https://github.com/mislav/hub.git
102 строки
2.7 KiB
Go
102 строки
2.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/bmizerany/assert"
|
|
"github.com/github/hub/fixtures"
|
|
)
|
|
|
|
func TestUpdater_downloadFile(t *testing.T) {
|
|
mux := http.NewServeMux()
|
|
server := httptest.NewServer(mux)
|
|
defer server.Close()
|
|
|
|
mux.HandleFunc("/gh.zip", func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "GET", r.Method)
|
|
fmt.Fprint(w, "1234")
|
|
})
|
|
|
|
path, err := downloadFile(fmt.Sprintf("%s/gh.zip", server.URL))
|
|
assert.Equal(t, nil, err)
|
|
|
|
content, err := ioutil.ReadFile(path)
|
|
assert.Equal(t, nil, err)
|
|
assert.Equal(t, "1234", string(content))
|
|
assert.Equal(t, "gh.zip", filepath.Base(path))
|
|
}
|
|
|
|
func TestUpdater_unzipExecutable(t *testing.T) {
|
|
target, _ := ioutil.TempFile("", "unzip-test")
|
|
defer target.Close()
|
|
|
|
source, _ := os.Open(filepath.Join("..", "fixtures", "gh.zip"))
|
|
defer source.Close()
|
|
|
|
_, err := io.Copy(target, source)
|
|
assert.Equal(t, nil, err)
|
|
|
|
exec, err := unzipExecutable(target.Name())
|
|
assert.Equal(t, nil, err)
|
|
assert.Equal(t, "gh", filepath.Base(exec))
|
|
}
|
|
|
|
func TestUpdater_timeToUpdate(t *testing.T) {
|
|
// file doesn't exist
|
|
timestampDir, _ := ioutil.TempDir("", "timestampDir-test")
|
|
timestampPath := filepath.Join(timestampDir, "gh-update")
|
|
updater := Updater{timestampPath: timestampPath}
|
|
|
|
assert.T(t, updater.timeToUpdate())
|
|
timestamp, err := ioutil.ReadFile(timestampPath)
|
|
assert.Equal(t, nil, err)
|
|
assert.NotEqual(t, "", string(timestamp))
|
|
|
|
// invalid timestamp format
|
|
timestampFile, _ := ioutil.TempFile("", "timestampFile-test")
|
|
updater = Updater{timestampPath: timestampFile.Name()}
|
|
assert.T(t, updater.timeToUpdate())
|
|
timestamp, err = ioutil.ReadFile(timestampFile.Name())
|
|
assert.Equal(t, nil, err)
|
|
assert.NotEqual(t, "", string(timestamp))
|
|
|
|
// dev version
|
|
updater = Updater{CurrentVersion: "dev"}
|
|
assert.T(t, !updater.timeToUpdate())
|
|
}
|
|
|
|
func TestSaveAlwaysAutoUpdateOption(t *testing.T) {
|
|
checkSavedAutoUpdateOption(t, true, "a", "always")
|
|
checkSavedAutoUpdateOption(t, true, "always", "always")
|
|
}
|
|
|
|
func TestSaveNeverAutoUpdateOption(t *testing.T) {
|
|
checkSavedAutoUpdateOption(t, false, "e", "never")
|
|
checkSavedAutoUpdateOption(t, false, "never", "never")
|
|
}
|
|
|
|
func TestDoesntSaveYesAutoUpdateOption(t *testing.T) {
|
|
checkSavedAutoUpdateOption(t, false, "y", "")
|
|
checkSavedAutoUpdateOption(t, false, "yes", "")
|
|
}
|
|
|
|
func TestDoesntSaveNoAutoUpdateOption(t *testing.T) {
|
|
checkSavedAutoUpdateOption(t, false, "n", "")
|
|
checkSavedAutoUpdateOption(t, false, "no", "")
|
|
}
|
|
|
|
func checkSavedAutoUpdateOption(t *testing.T, always bool, confirm, expected string) {
|
|
repo := fixtures.SetupTestRepo()
|
|
defer repo.TearDown()
|
|
|
|
saveAutoUpdateConfiguration(confirm, always)
|
|
assert.Equal(t, expected, autoUpdateConfig())
|
|
}
|