2021-09-08 18:19:30 +03:00
// Copyright 2021 The Gitea Authors. All rights reserved.
2022-11-27 21:20:29 +03:00
// SPDX-License-Identifier: MIT
2021-09-08 18:19:30 +03:00
package attachment
import (
"os"
"path/filepath"
"testing"
2022-05-20 17:08:52 +03:00
"code.gitea.io/gitea/models/db"
2021-11-19 16:39:57 +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-09-08 18:19:30 +03:00
2023-09-08 07:51:15 +03:00
_ "code.gitea.io/gitea/models/actions"
2021-09-08 18:19:30 +03:00
"github.com/stretchr/testify/assert"
)
func TestMain ( m * testing . M ) {
2023-09-28 04:38:53 +03:00
unittest . MainTest ( m )
2021-09-08 18:19:30 +03:00
}
func TestUploadAttachment ( t * testing . T ) {
2021-11-12 17:36:47 +03:00
assert . NoError ( t , unittest . PrepareTestDatabase ( ) )
2021-09-08 18:19:30 +03:00
2022-08-16 05:22:25 +03:00
user := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 1 } )
2021-09-08 18:19:30 +03:00
fPath := "./attachment_test.go"
f , err := os . Open ( fPath )
assert . NoError ( t , err )
defer f . Close ( )
2023-10-03 13:30:41 +03:00
attach , err := NewAttachment ( db . DefaultContext , & repo_model . Attachment {
2021-09-08 18:19:30 +03:00
RepoID : 1 ,
UploaderID : user . ID ,
Name : filepath . Base ( fPath ) ,
2023-03-12 10:48:07 +03:00
} , f , - 1 )
2021-09-08 18:19:30 +03:00
assert . NoError ( t , err )
2022-05-20 17:08:52 +03:00
attachment , err := repo_model . GetAttachmentByUUID ( db . DefaultContext , attach . UUID )
2021-09-08 18:19:30 +03:00
assert . NoError ( t , err )
assert . EqualValues ( t , user . ID , attachment . UploaderID )
assert . Equal ( t , int64 ( 0 ) , attachment . DownloadCount )
}