generated from PenaSide/GolangTemplate
134 lines
4.3 KiB
Go
134 lines
4.3 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"fmt"
|
|
"gitea.pena/PenaSide/common/encrypt"
|
|
"gitea.pena/PenaSide/common/validate"
|
|
"gitea.pena/PenaSide/customer/internal/models"
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestValidateTG(t *testing.T) {
|
|
err := validateTG("7127966184:AAG1steOCH4wDvHRe9QcsXJPS4dWRyRYsqg", -1002177203276, -1002177203276)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestValidateMail(t *testing.T) {
|
|
err := validateMail(models.MailClientCfg{
|
|
ApiUrl: "https://api.smtp.bz/v1/smtp/send",
|
|
Sender: "noreply@mailing.pena.digital",
|
|
ApiKey: "P0YsjUB137upXrr1NiJefHmXVKW1hmBWlpev",
|
|
MailAddress: "sells@pena.digital",
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestValidateURLs(t *testing.T) {
|
|
urls := []string{
|
|
"http://10.8.0.6:59300",
|
|
"http://10.8.0.6:59303",
|
|
"http://10.8.0.6:3131",
|
|
"10.8.0.6:9001",
|
|
"10.8.0.6:9085",
|
|
"http://10.8.0.6:7036",
|
|
"http://10.6.0.17",
|
|
"https://api.smtp.bz/v1/smtp/send",
|
|
"10.8.0.6:59665",
|
|
"https://sadmin.pena",
|
|
"10.8.0.15:7113",
|
|
}
|
|
err := validateURLs(urls)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestValidateKafka(t *testing.T) {
|
|
err := validateKafka([]string{"localhost:9092"}, "test-topic")
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestValidateEncryptKeys(t *testing.T) {
|
|
priveKey, pubKey, err := generateRSAKeys(2048)
|
|
assert.NoError(t, err)
|
|
err = validate.ValidateEncryptKeys(&encrypt.Encrypt{
|
|
PubKey: pubKey,
|
|
PrivKey: priveKey,
|
|
})
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func generateRSAKeys(bitSize int) (privateKey string, publicKey string, err error) {
|
|
privKey, err := rsa.GenerateKey(rand.Reader, bitSize)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
privKeyPEM := pem.EncodeToMemory(&pem.Block{
|
|
Type: "RSA PRIVATE KEY",
|
|
Bytes: x509.MarshalPKCS1PrivateKey(privKey),
|
|
})
|
|
|
|
pubKeyBytes, err := x509.MarshalPKIXPublicKey(&privKey.PublicKey)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
pubKeyPEM := pem.EncodeToMemory(&pem.Block{
|
|
Type: "RSA PUBLIC KEY",
|
|
Bytes: pubKeyBytes,
|
|
})
|
|
|
|
return string(privKeyPEM), string(pubKeyPEM), nil
|
|
}
|
|
|
|
func TestLoadConfig(t *testing.T) {
|
|
envVars := map[string]string{
|
|
"JWT_ISSUER": "pena-auth-service",
|
|
"JWT_AUDIENCE": "pena",
|
|
"JWT_PUBLIC_KEY": "-----BEGIN PUBLIC KEY-----\nMIGeMA0GCSqGSIb3DQEBAQUAA4GMADCBiAKBgHgnvr7O2tiApjJfid1orFnIGm69\n80fZp+Lpbjo+NC/0whMFga2Biw5b1G2Q/B2u0tpO1Fs/E8z7Lv1nYfr5jx2S8x6B\ndA4TS2kB9Kf0wn0+7wSlyikHoKhbtzwXHZl17GsyEi6wHnsqNBSauyIWhpha8i+Y\n+3GyaOY536H47qyXAgMBAAE=\n-----END PUBLIC KEY-----",
|
|
"CLIENT_HTTP_HOST": "0.0.0.0",
|
|
"CLIENT_HTTP_PORT": "8080",
|
|
"ADMIN_HTTP_HOST": "0.0.0.0",
|
|
"ADMIN_HTTP_PORT": "8081",
|
|
"GRPC_HOST": "0.0.0.0",
|
|
"GRPC_PORT": "9001",
|
|
"GRPC_DOMEN": "customer-service:9000",
|
|
"MONGO_URL": "mongodb://test:test@localhost:27020/",
|
|
"MONGO_DB_NAME": "admin",
|
|
"KAFKA_BROKERS": "localhost:9092",
|
|
"KAFKA_TOPIC_TARIFF": "test-topic",
|
|
"AUTH_MICROSERVICE_URL": "http://localhost:8000/user",
|
|
"HUBADMIN_MICROSERVICE_URL": "http://localhost:8001/tariff",
|
|
"CURRENCY_MICROSERVICE_URL": "http://cbrfworker-service:8000/change",
|
|
"DISCOUNT_MICROSERVICE_GRPC_HOST": "localhost:9040",
|
|
"PAYMENT_MICROSERVICE_GRPC_HOST": "treasurer-service:9085",
|
|
"VERIFICATION_MICROSERVICE_URL": "http://10.8.0.8:7035/verification",
|
|
"TEMPLATEGEN_MICROSERVICE_URL": "10.6.0.17",
|
|
"CODEWORD_MICROSERVICE_GRPC_HOST": "http://localhost:8000/user",
|
|
"TRASH_LOG_HOST": "localhost:7113",
|
|
"API_URL": "https://api.smtp.bz/v1/smtp/send",
|
|
"MAIL_SENDER": "noreply@mailing.pena.digital",
|
|
"MAIL_API_KEY": "P0YsjUB137upXrr1NiJefHmXVKW1hmBWlpev",
|
|
"MAIL_ADDRESS": "mail@mail.com",
|
|
"NOTIFICATION_BOT_TOKEN": "7127966184:AAG1steOCH4wDvHRe9QcsXJPS4dWRyRYsqg",
|
|
"NOTIFICATION_CHANNEL": "-1002177203276",
|
|
"NOTIFICATION_RS_PAY_CHANNEL": "-1002177203276",
|
|
"ADMIN_FRONT_URL": "https://admin-front.ru",
|
|
"ENCRYPT_PUBLIC_KEY": "1",
|
|
"ENCRYPT_PRIVATE_KEY": "2",
|
|
}
|
|
|
|
for key, value := range envVars {
|
|
os.Setenv(key, value)
|
|
}
|
|
|
|
cfg, err := loadConfig()
|
|
assert.Nil(t, err)
|
|
|
|
fmt.Println(cfg)
|
|
}
|