- // Copyright 2017 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package integrations
-
- import (
- "bytes"
- "database/sql"
- "fmt"
- "io"
- "log"
- "net/http"
- "net/http/cookiejar"
- "net/url"
- "os"
- "path"
- "strings"
- "testing"
-
- "code.gitea.io/gitea/models"
- "code.gitea.io/gitea/modules/setting"
- "code.gitea.io/gitea/routers"
- "code.gitea.io/gitea/routers/routes"
-
- "github.com/Unknwon/com"
- "github.com/stretchr/testify/assert"
- "gopkg.in/macaron.v1"
- "gopkg.in/testfixtures.v2"
- )
-
- var mac *macaron.Macaron
-
- func TestMain(m *testing.M) {
- initIntegrationTest()
- mac = routes.NewMacaron()
- routes.RegisterRoutes(mac)
-
- var helper testfixtures.Helper
- if setting.UseMySQL {
- helper = &testfixtures.MySQL{}
- } else if setting.UsePostgreSQL {
- helper = &testfixtures.PostgreSQL{}
- } else if setting.UseSQLite3 {
- helper = &testfixtures.SQLite{}
- } else {
- fmt.Println("Unsupported RDBMS for integration tests")
- os.Exit(1)
- }
-
- err := models.InitFixtures(
- helper,
- "models/fixtures/",
- )
- if err != nil {
- fmt.Printf("Error initializing test database: %v\n", err)
- os.Exit(1)
- }
- os.Exit(m.Run())
- }
-
- func initIntegrationTest() {
- giteaRoot := os.Getenv("GITEA_ROOT")
- if giteaRoot == "" {
- fmt.Println("Environment variable $GITEA_ROOT not set")
- os.Exit(1)
- }
- setting.AppPath = path.Join(giteaRoot, "gitea")
-
- giteaConf := os.Getenv("GITEA_CONF")
- if giteaConf == "" {
- fmt.Println("Environment variable $GITEA_CONF not set")
- os.Exit(1)
- } else if !path.IsAbs(giteaConf) {
- setting.CustomConf = path.Join(giteaRoot, giteaConf)
- } else {
- setting.CustomConf = giteaConf
- }
-
- setting.NewContext()
- models.LoadConfigs()
-
- switch {
- case setting.UseMySQL:
- db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s)/",
- models.DbCfg.User, models.DbCfg.Passwd, models.DbCfg.Host))
- defer db.Close()
- if err != nil {
- log.Fatalf("sql.Open: %v", err)
- }
- if _, err = db.Exec("CREATE DATABASE IF NOT EXISTS testgitea"); err != nil {
- log.Fatalf("db.Exec: %v", err)
- }
- case setting.UsePostgreSQL:
- db, err := sql.Open("postgres", fmt.Sprintf("postgres://%s:%s@%s/?sslmode=%s",
- models.DbCfg.User, models.DbCfg.Passwd, models.DbCfg.Host, models.DbCfg.SSLMode))
- defer db.Close()
- if err != nil {
- log.Fatalf("sql.Open: %v", err)
- }
- rows, err := db.Query(fmt.Sprintf("SELECT 1 FROM pg_database WHERE datname = '%s'",
- models.DbCfg.Name))
- if err != nil {
- log.Fatalf("db.Query: %v", err)
- }
- defer rows.Close()
-
- if rows.Next() {
- break
- }
- if _, err = db.Exec("CREATE DATABASE testgitea"); err != nil {
- log.Fatalf("db.Exec: %v", err)
- }
- }
- routers.GlobalInit()
- }
-
- func prepareTestEnv(t *testing.T) {
- assert.NoError(t, models.LoadFixtures())
- assert.NoError(t, os.RemoveAll("integrations/gitea-integration"))
- assert.NoError(t, com.CopyDir("integrations/gitea-integration-meta", "integrations/gitea-integration"))
- }
-
- type TestSession struct {
- jar http.CookieJar
- }
-
- func (s *TestSession) GetCookie(name string) *http.Cookie {
- baseURL, err := url.Parse(setting.AppURL)
- if err != nil {
- return nil
- }
-
- for _, c := range s.jar.Cookies(baseURL) {
- if c.Name == name {
- return c
- }
- }
- return nil
- }
-
- func (s *TestSession) MakeRequest(t *testing.T, req *http.Request) *TestResponse {
- baseURL, err := url.Parse(setting.AppURL)
- assert.NoError(t, err)
- for _, c := range s.jar.Cookies(baseURL) {
- req.AddCookie(c)
- }
- resp := MakeRequest(req)
-
- ch := http.Header{}
- ch.Add("Cookie", strings.Join(resp.Headers["Set-Cookie"], ";"))
- cr := http.Request{Header: ch}
- s.jar.SetCookies(baseURL, cr.Cookies())
-
- return resp
- }
-
- func loginUser(t *testing.T, userName, password string) *TestSession {
- req := NewRequest(t, "GET", "/user/login")
- resp := MakeRequest(req)
- assert.EqualValues(t, http.StatusOK, resp.HeaderCode)
-
- doc, err := NewHtmlParser(resp.Body)
- assert.NoError(t, err)
-
- req = NewRequestBody(t, "POST", "/user/login",
- bytes.NewBufferString(url.Values{
- "_csrf": []string{doc.GetInputValueByName("_csrf")},
- "user_name": []string{userName},
- "password": []string{password},
- }.Encode()),
- )
- req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
- resp = MakeRequest(req)
- assert.EqualValues(t, http.StatusFound, resp.HeaderCode)
-
- ch := http.Header{}
- ch.Add("Cookie", strings.Join(resp.Headers["Set-Cookie"], ";"))
- cr := http.Request{Header: ch}
-
- jar, err := cookiejar.New(nil)
- assert.NoError(t, err)
- baseURL, err := url.Parse(setting.AppURL)
- assert.NoError(t, err)
- jar.SetCookies(baseURL, cr.Cookies())
-
- return &TestSession{jar: jar}
- }
-
- type TestResponseWriter struct {
- HeaderCode int
- Writer io.Writer
- Headers http.Header
- }
-
- func (w *TestResponseWriter) Header() http.Header {
- return w.Headers
- }
-
- func (w *TestResponseWriter) Write(b []byte) (int, error) {
- return w.Writer.Write(b)
- }
-
- func (w *TestResponseWriter) WriteHeader(n int) {
- w.HeaderCode = n
- }
-
- type TestResponse struct {
- HeaderCode int
- Body []byte
- Headers http.Header
- }
-
- func NewRequest(t *testing.T, method, url string) *http.Request {
- return NewRequestBody(t, method, url, nil)
- }
-
- func NewRequestBody(t *testing.T, method, url string, body io.Reader) *http.Request {
- request, err := http.NewRequest(method, url, body)
- assert.NoError(t, err)
- request.RequestURI = url
- return request
- }
-
- func MakeRequest(req *http.Request) *TestResponse {
- buffer := bytes.NewBuffer(nil)
- respWriter := &TestResponseWriter{
- Writer: buffer,
- Headers: make(map[string][]string),
- }
- mac.ServeHTTP(respWriter, req)
- return &TestResponse{
- HeaderCode: respWriter.HeaderCode,
- Body: buffer.Bytes(),
- Headers: respWriter.Headers,
- }
- }
|