mirror/Gokapi
1
0
Fork 0
mirror of https://github.com/Forceu/Gokapi.git synced 2026-05-20 22:58:37 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
e6227a95beea94024f41de8abb434b06d0c7ac20
Gokapi/internal/test/testconfiguration/TestConfiguration.go
T
Marc Ole Bulling e6227a95be Fixed AWS tests, refactoring
2021-05-17 11:04:39 +02:00

394 lines
12 KiB
Go
Raw Blame History

// +build test
package testconfiguration
import (
"Gokapi/internal/storage/cloudstorage/aws"
"bytes"
"fmt"
"github.com/johannesboyne/gofakes3"
"github.com/johannesboyne/gofakes3/backend/s3mem"
"net/http/httptest"
"os"
)
const (
dataDir = "test"
configFile = dataDir + "/config.json"
)
// Create creates a configuration for unit testing
func Create(initFiles bool) {
os.Setenv("GOKAPI_CONFIG_DIR", "test")
os.Setenv("GOKAPI_DATA_DIR", "test")
os.Mkdir(dataDir, 0777)
os.WriteFile(configFile, configTestFile, 0777)
if initFiles {
os.Mkdir("test/data", 0777)
os.WriteFile("test/data/a8fdc205a9f19cc1c7507a60c4f01b13d11d7fd0", []byte("123"), 0777)
os.WriteFile("test/data/c4f9375f9834b4e7f0a528cc65c055702bf5f24a", []byte("456"), 0777)
os.WriteFile("test/data/e017693e4a04a59d0b0f400fe98177fe7ee13cf7", []byte("789"), 0777)
os.WriteFile("test/data/2341354656543213246465465465432456898794", []byte("abc"), 0777)
os.WriteFile("test/fileupload.jpg", []byte("abc"), 0777)
}
}
// WriteUpgradeConfigFile writes a Gokapi v1.1.0 config file
func WriteUpgradeConfigFile() {
os.Mkdir(dataDir, 0777)
os.WriteFile(configFile, configUpgradeTestFile, 0777)
}
// WriteSslCertificates writes a valid or invalid SSL certificate
func WriteSslCertificates(valid bool) {
os.Mkdir(dataDir, 0777)
if valid {
os.WriteFile("test/ssl.crt", sslCertValid, 0700)
os.WriteFile("test/ssl.key", sslKeyValid, 0700)
} else {
os.WriteFile("test/ssl.crt", sslCertExpired, 0700)
os.WriteFile("test/ssl.key", sslKeyExpired, 0700)
}
}
// WriteCloudConfigFile writes a valid or invalid AWS config file
func WriteCloudConfigFile(valid bool) {
os.Mkdir(dataDir, 0777)
if valid {
os.WriteFile("test/cloudconfig.yml", cloudConfigTestFile, 0700)
} else {
os.WriteFile("test/cloudconfig.yml", []byte("invalid"), 0700)
}
}
// Delete deletes the configuration for unit testing
func Delete() {
os.RemoveAll(dataDir)
}
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")
}
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, bytes.NewReader([]byte{}), 0)
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() {
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")
}
var configTestFile = []byte(`{
"Port":"127.0.0.1:53843",
"AdminName":"test",
"AdminPassword":"10340aece68aa4fb14507ae45b05506026f276cf",
"ServerUrl":"http://127.0.0.1:53843/",
"DefaultDownloads":3,
"DefaultExpiry":20,
"DefaultPassword":"123",
"RedirectUrl":"https://test.com/",
"Sessions":{
"validsession":{
"RenewAt":2147483645,
"ValidUntil":2147483646
},
"logoutsession":{
"RenewAt":2147483645,
"ValidUntil":2147483646
},
"needsRenewal":{
"RenewAt":0,
"ValidUntil":2147483646
},
"expiredsession":{
"RenewAt":0,
"ValidUntil":0
}
},
"Files":{
"Wzol7LyY2QVczXynJtVo":{
"Id":"Wzol7LyY2QVczXynJtVo",
"Name":"smallfile2",
"Size":"8 B",
"SHA256":"e017693e4a04a59d0b0f400fe98177fe7ee13cf7",
"ExpireAt":2147483646,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":1,
"PasswordHash":"",
"ContentType":"text/html",
"HotlinkId":""
},
"e4TjE7CokWK0giiLNxDL":{
"Id":"e4TjE7CokWK0giiLNxDL",
"Name":"smallfile2",
"Size":"8 B",
"SHA256":"e017693e4a04a59d0b0f400fe98177fe7ee13cf7",
"ExpireAt":2147483645,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":2,
"PasswordHash":"",
"ContentType":"text/html",
"HotlinkId":""
},
"wefffewhtrhhtrhtrhtr":{
"Id":"wefffewhtrhhtrhtrhtr",
"Name":"smallfile3",
"Size":"8 B",
"SHA256":"e017693e4a04a59d0b0f400fe98177fe7ee13cf7",
"ExpireAt":2147483645,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":1,
"PasswordHash":"",
"ContentType":"text/html",
"HotlinkId":""
},
"deletedfile123456789":{
"Id":"deletedfile123456789",
"Name":"DeletedFile",
"Size":"8 B",
"SHA256":"invalid",
"ExpireAt":2147483645,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":2,
"PasswordHash":"",
"ContentType":"text/html",
"HotlinkId":""
},
"jpLXGJKigM4hjtA6T6sN":{
"Id":"jpLXGJKigM4hjtA6T6sN",
"Name":"smallfile",
"Size":"7 B",
"SHA256":"c4f9375f9834b4e7f0a528cc65c055702bf5f24a",
"ExpireAt":2147483646,
"ExpireAtString":"2021-05-04 15:18",
"DownloadsRemaining":1,
"ContentType":"text/html",
"PasswordHash":"7b30508aa9b233ab4b8a11b2af5816bdb58ca3e7",
"HotlinkId":""
},
"jpLXGJKigM4hjtA6T6sN2":{
"Id":"jpLXGJKigM4hjtA6T6sN2",
"Name":"smallfile",
"Size":"7 B",
"SHA256":"c4f9375f9834b4e7f0a528cc65c055702bf5f24a",
"ExpireAt":2147483646,
"ExpireAtString":"2021-05-04 15:18",
"DownloadsRemaining":1,
"ContentType":"text/html",
"PasswordHash":"7b30508aa9b233ab4b8a11b2af5816bdb58ca3e7",
"HotlinkId":""
},
"n1tSTAGj8zan9KaT4u6p":{
"Id":"n1tSTAGj8zan9KaT4u6p",
"Name":"picture.jpg",
"Size":"4 B",
"SHA256":"a8fdc205a9f19cc1c7507a60c4f01b13d11d7fd0",
"ExpireAt":2147483646,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":1,
"PasswordHash":"",
"ContentType":"text/html",
"HotlinkId":"PhSs6mFtf8O5YGlLMfNw9rYXx9XRNkzCnJZpQBi7inunv3Z4A.jpg"
},
"cleanuptest123456789":{
"Id":"cleanuptest123456789",
"Name":"cleanup",
"Size":"4 B",
"SHA256":"2341354656543213246465465465432456898794",
"ExpireAt":2147483646,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":0,
"PasswordHash":"",
"ContentType":"text/html",
"HotlinkId":""
},
"awsTest1234567890123":{
"Id":"awsTest1234567890123",
"Name":"Aws Test File",
"Size":"20 MB",
"SHA256":"x341354656543213246465465465432456898794",
"ExpireAt":2147483646,
"ExpireAtString":"2021-05-04 15:19",
"DownloadsRemaining":4,
"PasswordHash":"",
"ContentType":"application/octet-stream",
"AwsBucket":"gokapi-test",
"HotlinkId":""
}
},
"Hotlinks":{
"PhSs6mFtf8O5YGlLMfNw9rYXx9XRNkzCnJZpQBi7inunv3Z4A.jpg":{
"Id":"PhSs6mFtf8O5YGlLMfNw9rYXx9XRNkzCnJZpQBi7inunv3Z4A.jpg",
"FileId":"n1tSTAGj8zan9KaT4u6p"
}
},
"DownloadStatus":{
"69JCbLVxx2KxfvB6FYkrDn3oCU7BWT":{
"Id":"69JCbLVxx2KxfvB6FYkrDn3oCU7BWT",
"FileId":"cleanuptest123456789",
"ExpireAt":2147483646
}
},
"ApiKeys":{
"validkey":{
"Id":"validkey",
"FriendlyName":"First Key",
"LastUsed":0,
"LastUsedString":""
},
"GAh1IhXDvYnqfYLazWBqMB9HSFmNPO":{
"Id":"GAh1IhXDvYnqfYLazWBqMB9HSFmNPO",
"FriendlyName":"Second Key",
"LastUsed":1620671580,
"LastUsedString":"used"
},
"jiREglQJW0bOqJakfjdVfe8T1EM8n8":{
"Id":"jiREglQJW0bOqJakfjdVfe8T1EM8n8",
"FriendlyName":"Unnamed Key",
"LastUsed":0,
"LastUsedString":""
},
"okeCMWqhVMZSpt5c1qpCWhKvJJPifb":{
"Id":"okeCMWqhVMZSpt5c1qpCWhKvJJPifb",
"FriendlyName":"Unnamed Key",
"LastUsed":0,
"LastUsedString":""
}
},
"ConfigVersion":7,
"SaltAdmin":"LW6fW4Pjv8GtdWVLSZD66gYEev6NAaXxOVBw7C",
"SaltFiles":"lL5wMTtnVCn5TPbpRaSe4vAQodWW0hgk00WCZE",
"LengthId":20,
"DataDir":"test/data",
"UseSsl":false
}`)
var configUpgradeTestFile = []byte(`{
"Port":"127.0.0.1:53844",
"AdminName":"admin",
"AdminPassword":"7450c2403ab85f0e8d5436818b66b99fdd287ac6",
"ServerUrl":"https://gokapi.url/",
"DefaultDownloads":1,
"DefaultExpiry":14,
"DefaultPassword":"123",
"RedirectUrl":"https://github.com/Forceu/Gokapi/",
"Sessions":{
"y0t-OQGF5UPFHyFOLab38SNjrc_a4xdIHTsZclkLpxuSwwTzS_qEETsinkgVIdWNMnQjhcaZtgCoJdpu":{
"RenewAt":1619774155,
"ValidUntil":1622362555
}
},
"Files":{
"MgXJLe4XLfpXcL12ec4i":{
"Id":"MgXJLe4XLfpXcL12ec4i",
"Name":"gokapi-linux_amd64",
"Size":"10.2 MB",
"SHA256":"b08f5989e1c6d57b45fffe39a8edc5da715799b7",
"ExpireAt":1620980170,
"ExpireAtString":"2021-05-14 10:16",
"DownloadsRemaining":1,
"PasswordHash":"e143a1801faba4c5c6fdc2e823127c988940f72e"
},
"doLN1pgbb945DfhGottx":{
"Id":"doLN1pgbb945DfhGottx",
"Name":"config.json",
"Size":"945 B",
"SHA256":"d2d6fd5fbf4a4bb1b1ae2f19130dd75b5adc0a0b",
"ExpireAt":1620980181,
"ExpireAtString":"2021-05-14 10:16",
"DownloadsRemaining":1,
"PasswordHash":"e143a1801faba4c5c6fdc2e823127c988940f72e"
},
"q06tcBco9gdJTf_pZ8xf":{
"Id":"q06tcBco9gdJTf_pZ8xf",
"Name":"gokapi-linux_amd64",
"Size":"10.2 MB",
"SHA256":"b08f5989e1c6d57b45fffe39a8edc5da715799b7",
"ExpireAt":1620980160,
"ExpireAtString":"2021-05-14 10:16",
"DownloadsRemaining":1,
"PasswordHash":""
}
}
}`)
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