From 31359b1770e6f69cbc1ab97c235acad8988239d0 Mon Sep 17 00:00:00 2001
From: simon987 <fortier.simon@protonmail.com>
Date: Thu, 28 Feb 2019 21:45:33 -0500
Subject: [PATCH] Update project version

---
 api/models.go            | 1 +
 api/project.go           | 1 +
 test/api_project_test.go | 4 ++++
 3 files changed, 6 insertions(+)

diff --git a/api/models.go b/api/models.go
index b972eb4..27d0255 100644
--- a/api/models.go
+++ b/api/models.go
@@ -155,6 +155,7 @@ type UpdateProjectRequest struct {
 	Paused     bool       `json:"paused"`
 	AssignRate rate.Limit `json:"assign_rate"`
 	SubmitRate rate.Limit `json:"submit_rate"`
+	Version    string     `json:"version"`
 }
 
 func (req *UpdateProjectRequest) isValid(pid int64) bool {
diff --git a/api/project.go b/api/project.go
index c68a4e3..986445e 100644
--- a/api/project.go
+++ b/api/project.go
@@ -178,6 +178,7 @@ func (api *WebAPI) UpdateProject(r *Request) {
 		Paused:     updateReq.Paused,
 		AssignRate: updateReq.AssignRate,
 		SubmitRate: updateReq.SubmitRate,
+		Version:    updateReq.Version,
 	}
 	sess := api.Session.StartFasthttp(r.Ctx)
 	manager := sess.Get("manager")
diff --git a/test/api_project_test.go b/test/api_project_test.go
index dd0c346..2457c26 100644
--- a/test/api_project_test.go
+++ b/test/api_project_test.go
@@ -137,6 +137,7 @@ func TestUpdateProjectValid(t *testing.T) {
 		Paused:     true,
 		AssignRate: 1,
 		SubmitRate: 2,
+		Version:    "VersionB",
 	}, pid, testAdminCtx)
 
 	if updateResp.Ok != true {
@@ -157,6 +158,9 @@ func TestUpdateProjectValid(t *testing.T) {
 	if proj.Project.GitRepo != "GitRepoB" {
 		t.Error()
 	}
+	if proj.Project.Version != "VersionB" {
+		t.Error()
+	}
 	if proj.Project.Priority != 2 {
 		t.Error()
 	}