package auth
 
 import (

 	"crypto/rand"
 	"encoding/hex"

 	"io/ioutil"

 	"os"
 	"strings"

 	"testing"
 )
 
 func TestEncodeAuth(t *testing.T) {

 	newAuthConfig := &AuthConfig{Username: "ken", Password: "test", Email: "test@example.com"}

 	authStr := encodeAuth(newAuthConfig)

 	decAuthConfig := &AuthConfig{}
 	var err error
 	decAuthConfig.Username, decAuthConfig.Password, err = decodeAuth(authStr)

 	if err != nil {
 		t.Fatal(err)
 	}
 	if newAuthConfig.Username != decAuthConfig.Username {
 		t.Fatal("Encode Username doesn't match decoded Username")
 	}
 	if newAuthConfig.Password != decAuthConfig.Password {
 		t.Fatal("Encode Password doesn't match decoded Password")
 	}
 	if authStr != "a2VuOnRlc3Q=" {
 		t.Fatal("AuthString encoding isn't correct.")
 	}
 }

 func TestLogin(t *testing.T) {

 	os.Setenv("DOCKER_INDEX_URL", "https://indexstaging-docker.dotcloud.com")
 	defer os.Setenv("DOCKER_INDEX_URL", "")

 	authConfig := &AuthConfig{Username: "unittester", Password: "surlautrerivejetattendrai", Email: "noise+unittester@dotcloud.com"}

 	status, err := Login(authConfig, nil)

 	if err != nil {
 		t.Fatal(err)
 	}

 	if status != "Login Succeeded" {

 		t.Fatalf("Expected status \"Login Succeeded\", found \"%s\" instead", status)
 	}
 }
 

 func TestCreateAccount(t *testing.T) {

 	os.Setenv("DOCKER_INDEX_URL", "https://indexstaging-docker.dotcloud.com")
 	defer os.Setenv("DOCKER_INDEX_URL", "")
 	tokenBuffer := make([]byte, 16)
 	_, err := rand.Read(tokenBuffer)
 	if err != nil {
 		t.Fatal(err)
 	}
 	token := hex.EncodeToString(tokenBuffer)[:12]
 	username := "ut" + token

 	authConfig := &AuthConfig{Username: username, Password: "test42", Email: "docker-ut+" + token + "@example.com"}

 	status, err := Login(authConfig, nil)

 	if err != nil {
 		t.Fatal(err)
 	}
 	expectedStatus := "Account created. Please use the confirmation link we sent" +

 		" to your e-mail to activate it."

 	if status != expectedStatus {
 		t.Fatalf("Expected status: \"%s\", found \"%s\" instead.", expectedStatus, status)
 	}
 

 	status, err = Login(authConfig, nil)

 	if err == nil {
 		t.Fatalf("Expected error but found nil instead")
 	}
 
 	expectedError := "Login: Account is not Active"
 
 	if !strings.Contains(err.Error(), expectedError) {

 		t.Fatalf("Expected message \"%s\" but found \"%s\" instead", expectedError, err)

 	}

 }

 
 func TestSameAuthDataPostSave(t *testing.T) {
 	root, err := ioutil.TempDir("", "docker-test")
 	if err != nil {
 		t.Fatal(err)
 	}
 	configFile := &ConfigFile{
 		rootPath: root,
 		Configs:  make(map[string]AuthConfig, 1),
 	}
 
 	configFile.Configs["testIndex"] = AuthConfig{
 		Username: "docker-user",
 		Password: "docker-pass",
 		Email:    "docker@docker.io",
 	}
 
 	err = SaveConfig(configFile)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	authConfig := configFile.Configs["testIndex"]
 	if authConfig.Username != "docker-user" {
 		t.Fail()
 	}
 	if authConfig.Password != "docker-pass" {
 		t.Fail()
 	}
 	if authConfig.Email != "docker@docker.io" {
 		t.Fail()
 	}
 	if authConfig.Auth != "" {
 		t.Fail()
 	}
 }