2019-02-04 07:56:53 +08:00
// Copyright 2019 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 (
"context"
"fmt"
"net"
"net/http"
"net/url"
"os"
2019-10-12 08:13:27 +08:00
"path"
2019-02-04 07:56:53 +08:00
"path/filepath"
2021-11-06 14:23:32 +08:00
"strconv"
2019-02-04 07:56:53 +08:00
"testing"
"time"
2019-03-27 17:33:00 +08:00
"code.gitea.io/gitea/modules/git"
2019-02-04 07:56:53 +08:00
"code.gitea.io/gitea/modules/setting"
2019-02-07 15:13:12 +08:00
"code.gitea.io/gitea/modules/ssh"
2020-08-12 04:05:34 +08:00
"code.gitea.io/gitea/modules/util"
2019-08-24 00:40:30 +08:00
2019-02-04 07:56:53 +08:00
"github.com/stretchr/testify/assert"
)
func withKeyFile ( t * testing . T , keyname string , callback func ( string ) ) {
2021-09-22 13:38:34 +08:00
tmpDir , err := os . MkdirTemp ( "" , "key-file" )
2019-08-14 19:19:13 +08:00
assert . NoError ( t , err )
2020-08-12 04:05:34 +08:00
defer util . RemoveAll ( tmpDir )
2019-08-14 19:19:13 +08:00
2022-01-21 01:46:10 +08:00
err = os . Chmod ( tmpDir , 0 o700 )
2019-08-14 19:19:13 +08:00
assert . NoError ( t , err )
keyFile := filepath . Join ( tmpDir , keyname )
err = ssh . GenKeyPair ( keyFile )
2019-02-04 07:56:53 +08:00
assert . NoError ( t , err )
2021-09-22 13:38:34 +08:00
err = os . WriteFile ( path . Join ( tmpDir , "ssh" ) , [ ] byte ( "#!/bin/bash\n" +
2022-01-21 01:46:10 +08:00
"ssh -o \"UserKnownHostsFile=/dev/null\" -o \"StrictHostKeyChecking=no\" -o \"IdentitiesOnly=yes\" -i \"" + keyFile + "\" \"$@\"" ) , 0 o700 )
2019-10-12 08:13:27 +08:00
assert . NoError ( t , err )
2022-01-21 01:46:10 +08:00
// Setup ssh wrapper
2019-10-12 08:13:27 +08:00
os . Setenv ( "GIT_SSH" , path . Join ( tmpDir , "ssh" ) )
2019-02-04 07:56:53 +08:00
os . Setenv ( "GIT_SSH_COMMAND" ,
2019-08-17 06:28:55 +08:00
"ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o IdentitiesOnly=yes -i \"" + keyFile + "\"" )
2019-02-04 07:56:53 +08:00
os . Setenv ( "GIT_SSH_VARIANT" , "ssh" )
callback ( keyFile )
}
func createSSHUrl ( gitPath string , u * url . URL ) * url . URL {
u2 := * u
u2 . Scheme = "ssh"
u2 . User = url . User ( "git" )
2021-11-06 14:23:32 +08:00
u2 . Host = net . JoinHostPort ( setting . SSH . ListenHost , strconv . Itoa ( setting . SSH . ListenPort ) )
2019-02-04 07:56:53 +08:00
u2 . Path = gitPath
return & u2
}
2021-04-17 02:30:16 +08:00
func onGiteaRunTB ( t testing . TB , callback func ( testing . TB , * url . URL ) , prepare ... bool ) {
2019-11-14 05:06:35 +08:00
if len ( prepare ) == 0 || prepare [ 0 ] {
2019-12-08 09:14:50 +08:00
defer prepareTestEnv ( t , 1 ) ( )
2019-11-14 05:06:35 +08:00
}
2019-02-04 07:56:53 +08:00
s := http . Server {
2020-11-13 20:51:07 +08:00
Handler : c ,
2019-02-04 07:56:53 +08:00
}
u , err := url . Parse ( setting . AppURL )
assert . NoError ( t , err )
listener , err := net . Listen ( "tcp" , u . Host )
2019-12-16 00:21:16 +08:00
i := 0
for err != nil && i <= 10 {
time . Sleep ( 100 * time . Millisecond )
listener , err = net . Listen ( "tcp" , u . Host )
i ++
}
2019-02-04 07:56:53 +08:00
assert . NoError ( t , err )
2019-11-26 23:35:41 +08:00
u . Host = listener . Addr ( ) . String ( )
2019-02-04 07:56:53 +08:00
defer func ( ) {
ctx , cancel := context . WithTimeout ( context . Background ( ) , 2 * time . Minute )
s . Shutdown ( ctx )
cancel ( )
} ( )
go s . Serve ( listener )
2022-01-21 01:46:10 +08:00
// Started by config go ssh.Listen(setting.SSH.ListenHost, setting.SSH.ListenPort, setting.SSH.ServerCiphers, setting.SSH.ServerKeyExchanges, setting.SSH.ServerMACs)
2019-02-04 07:56:53 +08:00
callback ( t , u )
}
2021-04-17 02:30:16 +08:00
func onGiteaRun ( t * testing . T , callback func ( * testing . T , * url . URL ) , prepare ... bool ) {
onGiteaRunTB ( t , func ( t testing . TB , u * url . URL ) {
callback ( t . ( * testing . T ) , u )
} , prepare ... )
}
2019-02-04 07:56:53 +08:00
func doGitClone ( dstLocalPath string , u * url . URL ) func ( * testing . T ) {
return func ( t * testing . T ) {
2022-01-26 02:15:58 +08:00
assert . NoError ( t , git . CloneWithArgs ( context . Background ( ) , u . String ( ) , dstLocalPath , git . AllowLFSFiltersArgs ( ) , git . CloneRepoOptions { } ) )
2020-12-25 17:59:32 +08:00
exist , err := util . IsExist ( filepath . Join ( dstLocalPath , "README.md" ) )
assert . NoError ( t , err )
assert . True ( t , exist )
2019-02-04 07:56:53 +08:00
}
}
2022-01-24 05:19:32 +08:00
func doPartialGitClone ( dstLocalPath string , u * url . URL ) func ( * testing . T ) {
return func ( t * testing . T ) {
2022-01-26 02:15:58 +08:00
assert . NoError ( t , git . CloneWithArgs ( context . Background ( ) , u . String ( ) , dstLocalPath , git . AllowLFSFiltersArgs ( ) , git . CloneRepoOptions {
2022-01-24 05:19:32 +08:00
Filter : "blob:none" ,
} ) )
exist , err := util . IsExist ( filepath . Join ( dstLocalPath , "README.md" ) )
assert . NoError ( t , err )
assert . True ( t , exist )
}
}
2020-04-28 16:32:23 +08:00
func doGitCloneFail ( u * url . URL ) func ( * testing . T ) {
2019-02-04 07:56:53 +08:00
return func ( t * testing . T ) {
2021-09-22 13:38:34 +08:00
tmpDir , err := os . MkdirTemp ( "" , "doGitCloneFail" )
2020-04-28 16:32:23 +08:00
assert . NoError ( t , err )
2020-08-12 04:05:34 +08:00
defer util . RemoveAll ( tmpDir )
2022-01-20 07:26:57 +08:00
assert . Error ( t , git . Clone ( git . DefaultContext , u . String ( ) , tmpDir , git . CloneRepoOptions { } ) )
2020-12-25 17:59:32 +08:00
exist , err := util . IsExist ( filepath . Join ( tmpDir , "README.md" ) )
assert . NoError ( t , err )
assert . False ( t , exist )
2019-02-04 07:56:53 +08:00
}
}
func doGitInitTestRepository ( dstPath string ) func ( * testing . T ) {
return func ( t * testing . T ) {
// Init repository in dstPath
2022-01-20 07:26:57 +08:00
assert . NoError ( t , git . InitRepository ( git . DefaultContext , dstPath , false ) )
2021-12-21 04:55:05 +08:00
// forcibly set default branch to master
_ , err := git . NewCommand ( "symbolic-ref" , "HEAD" , git . BranchPrefix + "master" ) . RunInDir ( dstPath )
assert . NoError ( t , err )
2022-01-21 01:46:10 +08:00
assert . NoError ( t , os . WriteFile ( filepath . Join ( dstPath , "README.md" ) , [ ] byte ( fmt . Sprintf ( "# Testing Repository\n\nOriginally created in: %s" , dstPath ) ) , 0 o644 ) )
2019-02-04 07:56:53 +08:00
assert . NoError ( t , git . AddChanges ( dstPath , true ) )
signature := git . Signature {
Email : "test@example.com" ,
Name : "test" ,
When : time . Now ( ) ,
}
assert . NoError ( t , git . CommitChanges ( dstPath , git . CommitChangesOptions {
Committer : & signature ,
Author : & signature ,
Message : "Initial Commit" ,
} ) )
}
}
func doGitAddRemote ( dstPath , remoteName string , u * url . URL ) func ( * testing . T ) {
return func ( t * testing . T ) {
_ , err := git . NewCommand ( "remote" , "add" , remoteName , u . String ( ) ) . RunInDir ( dstPath )
assert . NoError ( t , err )
}
}
2019-05-31 18:12:15 +08:00
func doGitPushTestRepository ( dstPath string , args ... string ) func ( * testing . T ) {
2019-02-04 07:56:53 +08:00
return func ( t * testing . T ) {
2019-05-31 18:12:15 +08:00
_ , err := git . NewCommand ( append ( [ ] string { "push" , "-u" } , args ... ) ... ) . RunInDir ( dstPath )
2019-02-04 07:56:53 +08:00
assert . NoError ( t , err )
}
}
2019-05-31 18:12:15 +08:00
func doGitPushTestRepositoryFail ( dstPath string , args ... string ) func ( * testing . T ) {
2019-02-04 07:56:53 +08:00
return func ( t * testing . T ) {
2019-05-31 18:12:15 +08:00
_ , err := git . NewCommand ( append ( [ ] string { "push" } , args ... ) ... ) . RunInDir ( dstPath )
2019-02-04 07:56:53 +08:00
assert . Error ( t , err )
}
}
2019-05-31 18:12:15 +08:00
func doGitCreateBranch ( dstPath , branch string ) func ( * testing . T ) {
return func ( t * testing . T ) {
_ , err := git . NewCommand ( "checkout" , "-b" , branch ) . RunInDir ( dstPath )
assert . NoError ( t , err )
}
}
func doGitCheckoutBranch ( dstPath string , args ... string ) func ( * testing . T ) {
return func ( t * testing . T ) {
2022-01-26 02:15:58 +08:00
_ , err := git . NewCommandNoGlobals ( append ( append ( git . AllowLFSFiltersArgs ( ) , "checkout" ) , args ... ) ... ) . RunInDir ( dstPath )
2019-05-31 18:12:15 +08:00
assert . NoError ( t , err )
}
}
func doGitMerge ( dstPath string , args ... string ) func ( * testing . T ) {
return func ( t * testing . T ) {
_ , err := git . NewCommand ( append ( [ ] string { "merge" } , args ... ) ... ) . RunInDir ( dstPath )
assert . NoError ( t , err )
}
}
func doGitPull ( dstPath string , args ... string ) func ( * testing . T ) {
return func ( t * testing . T ) {
2022-01-26 02:15:58 +08:00
_ , err := git . NewCommandNoGlobals ( append ( append ( git . AllowLFSFiltersArgs ( ) , "pull" ) , args ... ) ... ) . RunInDir ( dstPath )
2019-05-31 18:12:15 +08:00
assert . NoError ( t , err )
}
}