gandalftest: implement createRepository in the testing server

gandalftest/server.go

@@ -13,6 +13,7 @@ import (
 	"sync"
 
 	"github.com/gorilla/pat"
+	"github.com/tsuru/gandalf/repository"
 )
 
 type user struct {
@@ -40,7 +41,9 @@ type GandalfServer struct {
 	muxer     *pat.Router
 	users     []string
 	keys      map[string][]key
-	usersLock sync.Mutex
+	repos     []repository.Repository
+	usersLock sync.RWMutex
+	repoLock  sync.Mutex
 	failures  chan Failure
 }
 
@@ -98,6 +101,7 @@ func (s *GandalfServer) buildMuxer() {
 	s.muxer = pat.New()
 	s.muxer.Post("/user", http.HandlerFunc(s.createUser))
 	s.muxer.Delete("/user/{name}", http.HandlerFunc(s.removeUser))
+	s.muxer.Post("/repository", http.HandlerFunc(s.createRepository))
 }
 
 func (s *GandalfServer) createUser(w http.ResponseWriter, r *http.Request) {
@@ -126,13 +130,7 @@ func (s *GandalfServer) removeUser(w http.ResponseWriter, r *http.Request) {
 	username := r.URL.Query().Get(":name")
 	s.usersLock.Lock()
 	defer s.usersLock.Unlock()
-	index := -1
-	for i, user := range s.users {
-		if user == username {
-			index = i
-			break
-		}
-	}
+	_, index := s.findUser(username)
 	if index < 0 {
 		http.Error(w, "user not found", http.StatusNotFound)
 		return
@@ -143,6 +141,42 @@ func (s *GandalfServer) removeUser(w http.ResponseWriter, r *http.Request) {
 	delete(s.keys, username)
 }
 
+func (s *GandalfServer) createRepository(w http.ResponseWriter, r *http.Request) {
+	var repo repository.Repository
+	defer r.Body.Close()
+	err := json.NewDecoder(r.Body).Decode(&repo)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+	users := append(repo.Users, repo.ReadOnlyUsers...)
+	for _, username := range users {
+		_, index := s.findUser(username)
+		if index < 0 {
+			http.Error(w, fmt.Sprintf("user %q not found", username), http.StatusBadRequest)
+			return
+		}
+	}
+	s.repoLock.Lock()
+	defer s.repoLock.Unlock()
+	for _, r := range s.repos {
+		if r.Name == repo.Name {
+			http.Error(w, "repository already exists", http.StatusConflict)
+			return
+		}
+	}
+	s.repos = append(s.repos, repo)
+}
+
+func (s *GandalfServer) findUser(name string) (username string, index int) {
+	for i, user := range s.users {
+		if user == name {
+			return user, i
+		}
+	}
+	return "", -1
+}
+
 func (s *GandalfServer) getFailure(method, path string) (Failure, bool) {
 	var f Failure
 	select {

gandalftest/server_test.go

@@ -11,6 +11,7 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/tsuru/gandalf/repository"
 	"gopkg.in/check.v1"
 )
 
@@ -122,6 +123,48 @@ func (s *S) TestRemoveUserNotFound(c *check.C) {
 	c.Assert(recorder.Body.String(), check.Equals, "user not found\n")
 }
 
+func (s *S) TestCreateRepository(c *check.C) {
+	server, err := NewServer("127.0.0.1:0")
+	c.Assert(err, check.IsNil)
+	defer server.Stop()
+	server.users = []string{"user1", "user2", "user3"}
+	recorder := httptest.NewRecorder()
+	body := strings.NewReader(`{"Name":"myrepo","Users":["user1","user2"],"ReadOnlyUsers":["user3"],"IsPublic":true}`)
+	request, _ := http.NewRequest("POST", "/repository", body)
+	server.ServeHTTP(recorder, request)
+	c.Assert(recorder.Code, check.Equals, http.StatusOK)
+	c.Assert(server.repos, check.HasLen, 1)
+	c.Assert(server.repos[0].Name, check.Equals, "myrepo")
+	c.Assert(server.repos[0].Users, check.DeepEquals, []string{"user1", "user2"})
+	c.Assert(server.repos[0].ReadOnlyUsers, check.DeepEquals, []string{"user3"})
+	c.Assert(server.repos[0].IsPublic, check.Equals, true)
+}
+
+func (s *S) TestCreateRepositoryDuplicateName(c *check.C) {
+	server, err := NewServer("127.0.0.1:0")
+	c.Assert(err, check.IsNil)
+	defer server.Stop()
+	server.repos = []repository.Repository{{Name: "somerepo"}}
+	recorder := httptest.NewRecorder()
+	body := strings.NewReader(`{"Name":"somerepo","IsPublic":false}`)
+	request, _ := http.NewRequest("POST", "/repository", body)
+	server.ServeHTTP(recorder, request)
+	c.Assert(recorder.Code, check.Equals, http.StatusConflict)
+	c.Assert(recorder.Body.String(), check.Equals, "repository already exists\n")
+}
+
+func (s *S) TestCreateRepositoryUserNotFound(c *check.C) {
+	server, err := NewServer("127.0.0.1:0")
+	c.Assert(err, check.IsNil)
+	defer server.Stop()
+	recorder := httptest.NewRecorder()
+	body := strings.NewReader(`{"Name":"myrepo","Users":["user1","user2"],"ReadOnlyUsers":["user3"],"IsPublic":true}`)
+	request, _ := http.NewRequest("POST", "/repository", body)
+	server.ServeHTTP(recorder, request)
+	c.Assert(recorder.Code, check.Equals, http.StatusBadRequest)
+	c.Assert(recorder.Body.String(), check.Equals, `user "user1" not found`+"\n")
+}
+
 func (s *S) TestPrepareFailure(c *check.C) {
 	server, err := NewServer("127.0.0.1:0")
 	c.Assert(err, check.IsNil)