1
0
mirror of https://github.com/docker/docker-credential-helpers.git synced 2026-06-13 16:01:28 +05:30

Implement credential programs reading from Stdin.

Signed-off-by: David Calavera <david.calavera@gmail.com>
This commit is contained in:
David Calavera
2016-02-09 10:53:34 -08:00
parent de9748d6ed
commit f4a0e81b0b
7 changed files with 266 additions and 71 deletions
+142
View File
@@ -0,0 +1,142 @@
package plugin
import (
"bytes"
"encoding/json"
"fmt"
"strings"
"testing"
"github.com/calavera/docker-credential-helpers/credentials"
)
type memoryStore struct {
creds map[string]*credentials.Credentials
}
func newMemoryStore() *memoryStore {
return &memoryStore{
creds: make(map[string]*credentials.Credentials),
}
}
func (m *memoryStore) Add(creds *credentials.Credentials) error {
m.creds[creds.ServerURL] = creds
return nil
}
func (m *memoryStore) Delete(serverURL string) error {
delete(m.creds, serverURL)
return nil
}
func (m *memoryStore) Get(serverURL string) (string, string, error) {
c, ok := m.creds[serverURL]
if !ok {
return "", "", fmt.Errorf("creds not found for %s", serverURL)
}
return c.Username, c.Password, nil
}
func TestStore(t *testing.T) {
serverURL := "https://index.docker.io/v1/"
creds := &credentials.Credentials{
ServerURL: serverURL,
Username: "foo",
Password: "bar",
}
b, err := json.Marshal(creds)
if err != nil {
t.Fatal(err)
}
in := bytes.NewReader(b)
h := newMemoryStore()
if err := store(h, in); err != nil {
t.Fatal(err)
}
c, ok := h.creds[serverURL]
if !ok {
t.Fatalf("creds not found for %s\n", serverURL)
}
if c.Username != "foo" {
t.Fatalf("expected username foo, got %s\n", c.Username)
}
if c.Password != "bar" {
t.Fatalf("expected username bar, got %s\n", c.Password)
}
}
func TestGet(t *testing.T) {
serverURL := "https://index.docker.io/v1/"
creds := &credentials.Credentials{
ServerURL: serverURL,
Username: "foo",
Password: "bar",
}
b, err := json.Marshal(creds)
if err != nil {
t.Fatal(err)
}
in := bytes.NewReader(b)
h := newMemoryStore()
if err := store(h, in); err != nil {
t.Fatal(err)
}
buf := strings.NewReader(serverURL)
w := new(bytes.Buffer)
if err := get(h, buf, w); err != nil {
t.Fatal(err)
}
if w.Len() == 0 {
t.Fatalf("expected output in the writer, got %d", w.Len())
}
var c credentialsGetResponse
if err := json.NewDecoder(w).Decode(&c); err != nil {
t.Fatal(err)
}
if c.Username != "foo" {
t.Fatalf("expected username foo, got %s\n", c.Username)
}
if c.Password != "bar" {
t.Fatalf("expected username bar, got %s\n", c.Password)
}
}
func TestErase(t *testing.T) {
serverURL := "https://index.docker.io/v1/"
creds := &credentials.Credentials{
ServerURL: serverURL,
Username: "foo",
Password: "bar",
}
b, err := json.Marshal(creds)
if err != nil {
t.Fatal(err)
}
in := bytes.NewReader(b)
h := newMemoryStore()
if err := store(h, in); err != nil {
t.Fatal(err)
}
buf := strings.NewReader(serverURL)
if err := erase(h, buf); err != nil {
t.Fatal(err)
}
w := new(bytes.Buffer)
if err := get(h, buf, w); err == nil {
t.Fatal("expected error getting missing creds, got empty")
}
}