mirror/Gokapi
1
0
Fork 0
mirror of https://github.com/Forceu/Gokapi.git synced 2026-05-25 01:09:38 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
a8c79ce6eaf32b73fbc4a2eaeaa4290707ab80ce
Gokapi/internal/test/testconfiguration/TestConfiguration.go
T
Paul Draeger a1efcdf2cf Added env variable to set a minium password length (#312) 
Co-authored-by: Marc Ole Bulling <Marc-Ole@gmx.de>
2025-10-20 17:48:46 +02:00

527 lines
16 KiB
Go
Raw Blame History

//go:build test
package testconfiguration
import (
"crypto/sha1"
"encoding/hex"
"errors"
"fmt"
"github.com/forceu/gokapi/internal/configuration/database"
"github.com/forceu/gokapi/internal/helper"
"github.com/forceu/gokapi/internal/models"
"github.com/forceu/gokapi/internal/storage/filesystem"
"github.com/forceu/gokapi/internal/storage/filesystem/s3filesystem/aws"
"github.com/forceu/gokapi/internal/storage/processingstatus/pstatusdb"
"github.com/johannesboyne/gofakes3"
"github.com/johannesboyne/gofakes3/backend/s3mem"
"net/http/httptest"
"os"
"strings"
"time"
)
const (
baseDir = "test"
dataDir = baseDir + "/data"
configFile = baseDir + "/config.json"
SqliteUrl = "sqlite://" + dataDir + "/gokapi.sqlite"
SaltAdmin = "LW6fW4Pjv8GtdWVLSZD66gYEev6NAaXxOVBw7C"
)
func SetDirEnv() {
os.Setenv("GOKAPI_CONFIG_DIR", baseDir)
os.Setenv("GOKAPI_DATA_DIR", dataDir)
err := os.MkdirAll(baseDir, 0777)
if err != nil {
panic(err)
}
err = os.MkdirAll(dataDir, 0777)
if err != nil {
panic(err)
}
}
// Create creates a configuration for unit testing. If initFiles is set, test metaData and content is created
func Create(initFiles bool) {
SetDirEnv()
err := os.WriteFile(configFile, configTestFile, 0777)
if err != nil {
panic(err)
}
config, err := database.ParseUrl(SqliteUrl, false)
if err != nil {
panic(err)
}
database.Connect(config)
writeUsers()
writeTestSessions()
writeTestFiles()
database.SaveHotlink(models.File{Id: "n1tSTAGj8zan9KaT4u6p", HotlinkId: "PhSs6mFtf8O5YGlLMfNw9rYXx9XRNkzCnJZpQBi7inunv3Z4A.jpg", ExpireAt: time.Now().Add(time.Hour).Unix()})
database.SaveHotlink(models.File{Id: "t4efyghvrill1w1zary2", HotlinkId: "wjqlzpq2.jpg", ExpireAt: time.Now().Add(time.Hour).Unix()})
writeApiKeys()
writeTestUploadStatus()
database.Close()
if initFiles {
_ = os.MkdirAll(dataDir, 0777)
_ = os.WriteFile(dataDir+"/a8fdc205a9f19cc1c7507a60c4f01b13d11d7fd0", []byte("123"), 0777)
_ = os.WriteFile(dataDir+"/a8fdc205a9f19cc1c7507a60c4f01b13d11d7fd1", []byte("123"), 0777)
_ = os.WriteFile(dataDir+"/c4f9375f9834b4e7f0a528cc65c055702bf5f24a", []byte("456"), 0777)
_ = os.WriteFile(dataDir+"/e017693e4a04a59d0b0f400fe98177fe7ee13cf7", []byte("789"), 0777)
_ = os.WriteFile(dataDir+"/2341354656543213246465465465432456898794", []byte("abc"), 0777)
_ = os.WriteFile(dataDir+"/03cfd743661f07975fa2f1220c5194cbaff48451", []byte("def"), 0777)
_ = os.WriteFile(dataDir+"/replacetest1", []byte("abc"), 0777)
_ = os.WriteFile(dataDir+"/replacetest2", []byte("abc"), 0777)
_ = os.WriteFile(dataDir+"/replacetest3", []byte("abc"), 0777)
_ = os.WriteFile(dataDir+"/pendingdeletion", []byte("ghi"), 0777)
_ = os.WriteFile(dataDir+"/unlimitedtest", []byte("def"), 0777)
_ = os.WriteFile(baseDir+"/fileupload.jpg", []byte("abc"), 0777)
}
}
func writeUsers() {
admin := models.User{
Id: 5,
Name: "Test",
Permissions: models.UserPermissionAll,
UserLevel: models.UserLevelSuperAdmin,
LastOnline: 0,
Password: hashSalt("adminadmin", "LW6fW4Pjv8GtdWVLSZD66gYEev6NAaXxOVBw7C"),
ResetPassword: false,
}
user := models.User{
Id: 7,
Name: "User",
Permissions: models.UserPermissionNone,
UserLevel: models.UserLevelUser,
LastOnline: 0,
Password: hashSalt("useruser", "LW6fW4Pjv8GtdWVLSZD66gYEev6NAaXxOVBw7C"),
ResetPassword: false,
}
database.SaveUser(admin, false)
database.SaveUser(user, false)
}
// Copied from configuration
func hashSalt(password, salt string) string {
if password == "" {
return ""
}
if salt == "" {
panic(errors.New("no salt provided"))
}
pwBytes := []byte(password + salt)
hash := sha1.New()
hash.Write(pwBytes)
return hex.EncodeToString(hash.Sum(nil))
}
// WriteEncryptedFile writes metadata for an encrypted file and returns the id
func WriteEncryptedFile() string {
name := helper.GenerateRandomString(10)
database.SaveMetaData(models.File{
Id: name,
Name: name,
SHA1: name,
Encryption: models.EncryptionInfo{
IsEncrypted: true,
},
UnlimitedDownloads: true,
UnlimitedTime: true,
})
return name
}
// WriteSslCertificates writes a valid or invalid SSL certificate
func WriteSslCertificates(valid bool) {
os.Mkdir(baseDir, 0777)
if valid {
os.WriteFile(baseDir+"/ssl.crt", sslCertValid, 0700)
os.WriteFile(baseDir+"/ssl.key", sslKeyValid, 0700)
} else {
os.WriteFile(baseDir+"/ssl.crt", sslCertExpired, 0700)
os.WriteFile(baseDir+"/ssl.key", sslKeyExpired, 0700)
}
}
// WriteCloudConfigFile writes a valid or invalid AWS config file
func WriteCloudConfigFile(valid bool) {
os.Mkdir(baseDir, 0777)
if valid {
os.WriteFile(baseDir+"/cloudconfig.yml", cloudConfigTestFile, 0700)
} else {
os.WriteFile(baseDir+"/cloudconfig.yml", []byte("invalid"), 0700)
}
}
// Delete deletes the configuration for unit testing
func Delete() {
os.RemoveAll(baseDir)
}
var testServer *httptest.Server
// EnableS3 sets env variables for mock S3
func EnableS3() {
if !aws.IsMockApi {
return
}
os.Setenv("GOKAPI_AWS_BUCKET", "gokapi-test")
os.Setenv("GOKAPI_AWS_REGION", "mock-region-1")
os.Setenv("GOKAPI_AWS_KEY", "accId")
os.Setenv("GOKAPI_AWS_KEY_SECRET", "accKey")
filesystem.SetAws()
}
func UseMockS3Server() bool {
if os.Getenv("REAL_AWS_CREDENTIALS") != "true" {
fmt.Println("Using MOCK S3 SERVER! To test real credentials, pass REAL_AWS_CREDENTIALS=true")
fmt.Println("To mock the API, run test with --tags test,awsmock")
return true
}
fmt.Println("Warning, using REAL AWS S3 API! This test will fail if no valid credentials have been provided.")
fmt.Println("To mock the API, run test with --tags test,awsmock or pass REAL_AWS_CREDENTIALS=false")
return false
}
func StartS3TestServer() *httptest.Server {
backend := s3mem.New()
_ = backend.CreateBucket("gokapi")
_ = backend.CreateBucket("gokapi-test")
_, _ = backend.PutObject("gokapi-test", "x341354656543213246465465465432456898794", nil, strings.NewReader("content"), 7)
faker := gofakes3.New(backend)
server := httptest.NewServer(faker.Server())
os.Setenv("GOKAPI_AWS_ENDPOINT", server.URL)
return server
}
// DisableS3 unsets env variables for mock S3
func DisableS3() {
aws.LogOut()
if !aws.IsMockApi {
return
}
os.Unsetenv("GOKAPI_AWS_BUCKET")
os.Unsetenv("AWS_REGION")
os.Unsetenv("AWS_ACCESS_KEY_ID")
os.Unsetenv("AWS_SECRET_ACCESS_KEY")
filesystem.SetLocal()
}
func writeTestSessions() {
database.SaveSession("validsession", models.Session{
RenewAt: 2147483645,
ValidUntil: 2147483646,
UserId: 7,
})
database.SaveSession("logoutsession", models.Session{
RenewAt: 2147483645,
ValidUntil: 2147483646,
UserId: 7,
})
database.SaveSession("logoutsession2", models.Session{
RenewAt: 2147483645,
ValidUntil: 2147483646,
UserId: 7,
})
database.SaveSession("needsRenewal", models.Session{
RenewAt: 0,
ValidUntil: 2147483646,
UserId: 7,
})
database.SaveSession("expiredsession", models.Session{
RenewAt: 0,
ValidUntil: 0,
UserId: 7,
})
database.SaveSession("validSessionInvalidUser", models.Session{
RenewAt: 2147483645,
ValidUntil: 2147483645,
UserId: 5000,
})
database.SaveSession("validSessionInvalidUser", models.Session{
RenewAt: 2147483645,
ValidUntil: 2147483645,
UserId: 5000,
})
}
func writeTestUploadStatus() {
pstatusdb.Set(models.UploadStatus{
ChunkId: "validstatus_0",
CurrentStatus: 0,
})
pstatusdb.Set(models.UploadStatus{
ChunkId: "validstatus_1",
CurrentStatus: 1,
})
}
func writeApiKeys() {
database.SaveApiKey(models.ApiKey{
Id: "validkey",
FriendlyName: "First Key",
Permissions: models.ApiPermAll, // TODO
UserId: 5,
PublicId: "taiyeo6uLie6nu6eip0ieweiM5mahv",
})
database.SaveApiKey(models.ApiKey{
Id: "validkeyid7",
FriendlyName: "Key for uid 7",
Permissions: models.ApiPermAll, // TODO
UserId: 7,
PublicId: "vu0eemi8eehaisuth3pahDai2eo6ze",
})
database.SaveApiKey(models.ApiKey{
Id: "GAh1IhXDvYnqfYLazWBqMB9HSFmNPO",
FriendlyName: "Second Key",
LastUsed: 1620671580,
Permissions: models.ApiPermAll, // TODO
UserId: 5,
PublicId: "yaeVohng1ohNohsh1vailizeil5ka5",
})
database.SaveApiKey(models.ApiKey{
Id: "jiREglQJW0bOqJakfjdVfe8T1EM8n8",
FriendlyName: "Unnamed Key",
Permissions: models.ApiPermAll, // TODO
UserId: 5,
PublicId: "ahYie4ophoo5OoGhahCe1neic6thah",
})
database.SaveApiKey(models.ApiKey{
Id: "okeCMWqhVMZSpt5c1qpCWhKvJJPifb",
FriendlyName: "Unnamed Key",
Permissions: models.ApiPermAll, // TODO
UserId: 5,
PublicId: "ugoo0roowoanahthei7ohSail5OChu",
})
}
func writeTestFiles() {
database.SaveMetaData(models.File{
Id: "Wzol7LyY2QVczXynJtVo",
Name: "smallfile2",
Size: "8 B",
SHA1: "e017693e4a04a59d0b0f400fe98177fe7ee13cf7",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 1,
ContentType: "text/html",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "e4TjE7CokWK0giiLNxDL",
Name: "smallfile2",
Size: "8 B",
SHA1: "e017693e4a04a59d0b0f400fe98177fe7ee13cf7",
ExpireAt: 2147483645,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 2,
ContentType: "text/html",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "wefffewhtrhhtrhtrhtr",
Name: "smallfile3",
Size: "8 B",
SHA1: "e017693e4a04a59d0b0f400fe98177fe7ee13cf7",
ExpireAt: 2147483645,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 1,
ContentType: "text/html",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "deletedfile123456789",
Name: "DeletedFile",
Size: "8 B",
SHA1: "invalid",
ExpireAt: 2147483645,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 2,
ContentType: "text/html",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "jpLXGJKigM4hjtA6T6sN",
Name: "smallfile",
Size: "7 B",
SHA1: "c4f9375f9834b4e7f0a528cc65c055702bf5f24a",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:18",
DownloadsRemaining: 1,
ContentType: "text/html",
PasswordHash: "7b30508aa9b233ab4b8a11b2af5816bdb58ca3e7",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "jpLXGJKigM4hjtA6T6sN2",
Name: "smallfile",
Size: "7 B",
SHA1: "c4f9375f9834b4e7f0a528cc65c055702bf5f24a",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:18",
DownloadsRemaining: 1,
ContentType: "text/html",
PasswordHash: "7b30508aa9b233ab4b8a11b2af5816bdb58ca3e7",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "n1tSTAGj8zan9KaT4u6p",
Name: "picture.jpg",
Size: "4 B",
SHA1: "a8fdc205a9f19cc1c7507a60c4f01b13d11d7fd0",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 1,
ContentType: "text/html",
HotlinkId: "PhSs6mFtf8O5YGlLMfNw9rYXx9XRNkzCnJZpQBi7inunv3Z4A.jpg",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "t4efyghvrill1w1zary2",
Name: "picture.jpg",
Size: "4 B",
SHA1: "a8fdc205a9f19cc1c7507a60c4f01b13d11d7fd1",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 1,
ContentType: "text/html",
HotlinkId: "wjqlzpq2.jpg",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "cleanuptest123456789",
Name: "cleanup",
Size: "4 B",
SHA1: "2341354656543213246465465465432456898794",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 0,
ContentType: "text/html",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "awsTest1234567890123",
Name: "Aws Test File",
Size: "20 MB",
SHA1: "x341354656543213246465465465432456898794",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 4,
ContentType: "application/octet-stream",
AwsBucket: "gokapi-test",
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "unlimitedDownload",
Name: "unlimitedDownload",
Size: "8 B",
SHA1: "unlimitedtest",
ExpireAt: 2147483646,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 0,
ContentType: "text/html",
UnlimitedDownloads: true,
UserId: 5,
})
database.SaveMetaData(models.File{
Id: "unlimitedTime",
Name: "unlimitedTime",
Size: "8 B",
SHA1: "unlimitedtest",
ExpireAt: 0,
ExpireAtString: "2021-05-04 15:19",
DownloadsRemaining: 1,
ContentType: "text/html",
UnlimitedTime: true,
UserId: 5,
})
}
var configTestFile = []byte(`{
"Authentication": {
"Method": 0,
"SaltAdmin": "` + SaltAdmin + `",
"SaltFiles": "lL5wMTtnVCn5TPbpRaSe4vAQodWW0hgk00WCZE",
"Username": "test",
"HeaderKey": "",
"OauthProvider": "",
"OAuthClientId": "",
"OAuthClientSecret": "",
"OauthGroupScope": "",
"OAuthRecheckInterval": 12,
"HeaderUsers": null,
"OAuthGroups": [],
"OauthUsers": []
},
"Port":"127.0.0.1:53843",
"ServerUrl": "http://127.0.0.1:53843/",
"RedirectUrl": "https://test.com/",
"PublicName": "Gokapi Test Version",
"DataDir": "` + dataDir + `",
"DatabaseUrl": "` + SqliteUrl + `",
"ConfigVersion": 22,
"MaxFileSizeMB": 25,
"MaxMemory": 10,
"ChunkSize": 45,
"Encryption": {
"Level": 0,
"Cipher": null,
"Salt": "",
"Checksum": "",
"ChecksumSalt": ""
},
"MaxParallelUploads": 4,
"UseSsl": false,
"PicturesAlwaysLocal": false,
"SaveIp": false,
"IncludeFilename": false,
"MinLengthPassword": 8
}`)
var sslCertValid = []byte(`-----BEGIN CERTIFICATE-----
MIIBVzCB/aADAgECAgEBMAoGCCqGSM49BAMCMBExDzANBgNVBAoTBkdva2FwaTAe
Fw0yMTA1MTExNzMwMzVaFw0zODAxMTkwMzE0MDVaMBExDzANBgNVBAoTBkdva2Fw
aTBZMBMGByqGSM49AgEGCCqGSM49AwEHA0IABPVFhEGE9MomZ8jLt011yvDnWx8k
i2jPNG/FzDjXpfgY/PhDWzR+HS3uoMSsAPnxlg/Xqz681ifvO2Ke8tsjZUujRjBE
MA4GA1UdDwEB/wQEAwIFoDATBgNVHSUEDDAKBggrBgEFBQcDATAMBgNVHRMBAf8E
AjAAMA8GA1UdEQQIMAaHBH8AAAEwCgYIKoZIzj0EAwIDSQAwRgIhAPOAn+51jcMH
tKO1wjI6vA0avJIuZNUh7wxq0y6K22mzAiEAisbOg45sRuD2V3ffsGfY6d3XyQvC
2A69IsVJJwxqr+g=
-----END CERTIFICATE-----`)
var sslKeyValid = []byte(`-----BEGIN EC PRIVATE KEY-----
MHcCAQEEINFOm9o9K15wzt2fHcnujDPPNERk02zYiMGfYChhaS8zoAoGCCqGSM49
AwEHoUQDQgAE9UWEQYT0yiZnyMu3TXXK8OdbHySLaM80b8XMONel+Bj8+ENbNH4d
Le6gxKwA+fGWD9erPrzWJ+87Yp7y2yNlSw==
-----END EC PRIVATE KEY-----`)
var sslCertExpired = []byte(`-----BEGIN CERTIFICATE-----
MIIBVjCB/aADAgECAgEBMAoGCCqGSM49BAMCMBExDzANBgNVBAoTBkdva2FwaTAe
Fw0yMTA1MTExNzU1MDVaFw0yMTA1MTExNzU1MDZaMBExDzANBgNVBAoTBkdva2Fw
aTBZMBMGByqGSM49AgEGCCqGSM49AwEHA0IABF+tcmF6JjtKhltXTWo9mlLCLJ+4
C2cAi8ahZS9tIaz/QHC1/Gl3i4Nx8QwubYVw9TScAPMUZTZr7TYJ6Gc3vuWjRjBE
MA4GA1UdDwEB/wQEAwIFoDATBgNVHSUEDDAKBggrBgEFBQcDATAMBgNVHRMBAf8E
AjAAMA8GA1UdEQQIMAaHBH8AAAEwCgYIKoZIzj0EAwIDSAAwRQIhAI0ZfsFfr/K/
lcHL0rWkwwlCKIe16v74VFob0pzREW1JAiA0hTFSlv12254Lqf5hEUWPXDeQsq+o
tTe2z6xKh0dwkQ==
-----END CERTIFICATE-----`)
var sslKeyExpired = []byte(`-----BEGIN EC PRIVATE KEY-----
MHcCAQEEIG4kCb5tqz0HyRMBY+HItWtZuT2RmH9w1vsyO2XJcHlLoAoGCCqGSM49
AwEHoUQDQgAEX61yYXomO0qGW1dNaj2aUsIsn7gLZwCLxqFlL20hrP9AcLX8aXeL
g3HxDC5thXD1NJwA8xRlNmvtNgnoZze+5Q==
-----END EC PRIVATE KEY-----`)
var cloudConfigTestFile = []byte(`
##
## Example AWS S3 config. Modify this file and save it to config/cloudconfig.yml
##
aws:
Bucket: "gokapi"
Region: "test-region"
Endpoint: "test-endpoint"
KeyId: "test-keyid"
KeySecret: "test-secret"
`)
Reference in New Issue View Git Blame Copy Permalink