100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
|
// Copyright (C) 2018 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/julienschmidt/httprouter"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
testBridgeURL = "http://example.com"
|
||
|
|
testBridgeUser = "testuser@storj.io"
|
||
|
|
testBridgePass = "secret"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewTestEnv(ts *httptest.Server) Env {
|
||
|
|
return Env{
|
||
|
|
URL: ts.URL,
|
||
|
|
User: testBridgeUser,
|
||
|
|
Password: testBridgePass,
|
||
|
|
Mnemonic: testMnemonic,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNoAuthTestEnv(ts *httptest.Server) Env {
|
||
|
|
return Env{
|
||
|
|
URL: ts.URL,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewBadPassTestEnv(ts *httptest.Server) Env {
|
||
|
|
return Env{
|
||
|
|
URL: ts.URL,
|
||
|
|
User: testBridgeUser,
|
||
|
|
Password: "bad password",
|
||
|
|
Mnemonic: testMnemonic,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNoMnemonicTestEnv(ts *httptest.Server) Env {
|
||
|
|
return Env{
|
||
|
|
URL: ts.URL,
|
||
|
|
User: testBridgeUser,
|
||
|
|
Password: testBridgePass,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func basicAuth(h httprouter.Handle, requiredUser, requiredPassword string) httprouter.Handle {
|
||
|
|
return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
||
|
|
// Get the Basic Authentication credentials
|
||
|
|
user, password, hasAuth := r.BasicAuth()
|
||
|
|
|
||
|
|
if hasAuth && user == requiredUser && password == requiredPassword {
|
||
|
|
// Delegate request to the given handle
|
||
|
|
h(w, r, ps)
|
||
|
|
} else {
|
||
|
|
// Request Basic Authentication otherwise
|
||
|
|
w.Header().Set("WWW-Authenticate", "Basic realm=Restricted")
|
||
|
|
http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewEnv(t *testing.T) {
|
||
|
|
for _, tt := range []struct {
|
||
|
|
env Env
|
||
|
|
url string
|
||
|
|
}{
|
||
|
|
{Env{}, ""},
|
||
|
|
{NewEnv(), DefaultURL},
|
||
|
|
{Env{URL: testBridgeURL}, testBridgeURL},
|
||
|
|
} {
|
||
|
|
assert.Equal(t, tt.url, tt.env.URL)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewEnvVars(t *testing.T) {
|
||
|
|
os.Setenv("STORJ_BRIDGE", testBridgeURL)
|
||
|
|
defer os.Unsetenv("STORJ_BRIDGE")
|
||
|
|
os.Setenv("STORJ_BRIDGE_USER", testBridgeUser)
|
||
|
|
defer os.Unsetenv("STORJ_BRIDGE_USER")
|
||
|
|
os.Setenv("STORJ_BRIDGE_PASS", testBridgePass)
|
||
|
|
defer os.Unsetenv("STORJ_BRIDGE_PASS")
|
||
|
|
os.Setenv("STORJ_ENCRYPTION_KEY", testMnemonic)
|
||
|
|
defer os.Unsetenv("STORJ_ENCRYPTION_KEY")
|
||
|
|
|
||
|
|
env := NewEnv()
|
||
|
|
|
||
|
|
assert.Equal(t, testBridgeURL, env.URL)
|
||
|
|
assert.Equal(t, testBridgeUser, env.User)
|
||
|
|
assert.Equal(t, sha256Sum(testBridgePass), env.Password)
|
||
|
|
assert.Equal(t, testMnemonic, env.Mnemonic)
|
||
|
|
}
|