2017-07-27 04:20:38 +03:00
// 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.
2022-06-13 12:37:59 +03:00
package issues_test
2017-07-27 04:20:38 +03:00
import (
"testing"
"time"
2022-01-20 02:26:57 +03:00
"code.gitea.io/gitea/models/db"
2022-06-13 12:37:59 +03:00
issues_model "code.gitea.io/gitea/models/issues"
2021-12-10 04:27:50 +03:00
repo_model "code.gitea.io/gitea/models/repo"
2021-11-12 17:36:47 +03:00
"code.gitea.io/gitea/models/unittest"
2021-11-24 12:49:20 +03:00
user_model "code.gitea.io/gitea/models/user"
2021-11-17 15:34:35 +03:00
2017-07-27 04:20:38 +03:00
"github.com/stretchr/testify/assert"
)
func TestCreateComment ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2017-07-27 04:20:38 +03:00
2022-06-13 12:37:59 +03:00
issue := unittest . AssertExistsAndLoadBean ( t , & issues_model . Issue { } ) . ( * issues_model . Issue )
2021-12-10 04:27:50 +03:00
repo := unittest . AssertExistsAndLoadBean ( t , & repo_model . Repository { ID : issue . RepoID } ) . ( * repo_model . Repository )
2021-11-24 12:49:20 +03:00
doer := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : repo . OwnerID } ) . ( * user_model . User )
2017-07-27 04:20:38 +03:00
now := time . Now ( ) . Unix ( )
2022-06-13 12:37:59 +03:00
comment , err := issues_model . CreateComment ( & issues_model . CreateCommentOptions {
Type : issues_model . CommentTypeComment ,
2017-07-27 04:20:38 +03:00
Doer : doer ,
Repo : repo ,
Issue : issue ,
Content : "Hello" ,
} )
assert . NoError ( t , err )
then := time . Now ( ) . Unix ( )
2022-06-13 12:37:59 +03:00
assert . EqualValues ( t , issues_model . CommentTypeComment , comment . Type )
2017-07-27 04:20:38 +03:00
assert . EqualValues ( t , "Hello" , comment . Content )
assert . EqualValues ( t , issue . ID , comment . IssueID )
assert . EqualValues ( t , doer . ID , comment . PosterID )
2021-11-16 11:53:21 +03:00
unittest . AssertInt64InRange ( t , now , then , int64 ( comment . CreatedUnix ) )
unittest . AssertExistsAndLoadBean ( t , comment ) // assert actually added to DB
2017-07-27 04:20:38 +03:00
2022-06-13 12:37:59 +03:00
updatedIssue := unittest . AssertExistsAndLoadBean ( t , & issues_model . Issue { ID : issue . ID } ) . ( * issues_model . Issue )
2021-11-16 11:53:21 +03:00
unittest . AssertInt64InRange ( t , now , then , int64 ( updatedIssue . UpdatedUnix ) )
2017-07-27 04:20:38 +03:00
}
2018-08-06 07:43:22 +03:00
func TestFetchCodeComments ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2018-08-06 07:43:22 +03:00
2022-06-13 12:37:59 +03:00
issue := unittest . AssertExistsAndLoadBean ( t , & issues_model . Issue { ID : 2 } ) . ( * issues_model . Issue )
2021-11-24 12:49:20 +03:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } ) . ( * user_model . User )
2022-06-13 12:37:59 +03:00
res , err := issues_model . FetchCodeComments ( db . DefaultContext , issue , user )
2018-08-06 07:43:22 +03:00
assert . NoError ( t , err )
assert . Contains ( t , res , "README.md" )
assert . Contains ( t , res [ "README.md" ] , int64 ( 4 ) )
assert . Len ( t , res [ "README.md" ] [ 4 ] , 1 )
assert . Equal ( t , int64 ( 4 ) , res [ "README.md" ] [ 4 ] [ 0 ] . ID )
2021-11-24 12:49:20 +03:00
user2 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } ) . ( * user_model . User )
2022-06-13 12:37:59 +03:00
res , err = issues_model . FetchCodeComments ( db . DefaultContext , issue , user2 )
2018-08-06 07:43:22 +03:00
assert . NoError ( t , err )
assert . Len ( t , res , 1 )
}