124 lines
2.9 KiB
Go
124 lines
2.9 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"certmanager/internal/config"
|
|
"certmanager/internal/database"
|
|
"certmanager/internal/descriptor"
|
|
"certmanager/pkg/auxid"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func XXTestDatabaseIssuer(t *testing.T) {
|
|
var err error
|
|
conf := config.NewConfig()
|
|
err = conf.ReadFile()
|
|
require.NoError(t, err)
|
|
|
|
db, err := database.NewDatabase(conf.DataDir)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, db)
|
|
|
|
err = db.InitDatabase()
|
|
require.NoError(t, err)
|
|
|
|
ctx, _ := context.WithTimeout(context.Background(), 1*time.Second)
|
|
|
|
issuerID := auxid.GenID()
|
|
issuer := &descriptor.Issuer{
|
|
ID: issuerID,
|
|
Name: "foo.bar",
|
|
Cert: "ASDFF",
|
|
Key: "QWERT",
|
|
Revoked: true,
|
|
}
|
|
err = db.InsertIssuer(ctx, issuer)
|
|
require.NoError(t, err)
|
|
fmt.Printf("issuerId: %d\n", issuerID)
|
|
|
|
issuer.Revoked = true
|
|
err = db.UpdateIssuerByID(ctx, issuerID, issuer)
|
|
require.NoError(t, err)
|
|
|
|
issuerExists, issuer, err := db.GetIssuerByID(ctx, issuerID)
|
|
require.NoError(t, err)
|
|
require.True(t, issuerExists)
|
|
require.NotNil(t, issuer)
|
|
require.True(t, issuer.Revoked)
|
|
|
|
issuerYAML, err := yaml.Marshal(issuer)
|
|
require.NoError(t, err)
|
|
fmt.Printf("Issuer: \n%s\n---\n", string(issuerYAML))
|
|
|
|
listIssuers, err := db.ListIssuers(ctx)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, listIssuers)
|
|
|
|
listIssuersYAML, err := yaml.Marshal(listIssuers)
|
|
require.NoError(t, err)
|
|
fmt.Printf("listIssuers: \n%s\n", string(listIssuersYAML))
|
|
|
|
err = db.DeleteIssuerByID(ctx, issuerID)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func XXXTestDatabaseService(t *testing.T) {
|
|
var err error
|
|
conf := config.NewConfig()
|
|
err = conf.ReadFile()
|
|
require.NoError(t, err)
|
|
|
|
db, err := database.NewDatabase(conf.DataDir)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, db)
|
|
|
|
err = db.InitDatabase()
|
|
require.NoError(t, err)
|
|
|
|
ctx, _ := context.WithTimeout(context.Background(), 1*time.Second)
|
|
|
|
serviceID := auxid.GenID()
|
|
service := &descriptor.Service{
|
|
ID: serviceID,
|
|
Name: "foo.bar",
|
|
IssuerID: 123456,
|
|
Cert: "ASDFF",
|
|
Key: "QWERT",
|
|
Revoked: true,
|
|
}
|
|
err = db.InsertService(ctx, service)
|
|
require.NoError(t, err)
|
|
fmt.Printf("serviceId: %d\n", serviceID)
|
|
|
|
service.Revoked = true
|
|
err = db.UpdateServiceByID(ctx, serviceID, service)
|
|
require.NoError(t, err)
|
|
|
|
serviceExists, service, err := db.GetServiceByID(ctx, serviceID)
|
|
require.NoError(t, err)
|
|
require.True(t, serviceExists)
|
|
require.NotNil(t, service)
|
|
require.True(t, service.Revoked)
|
|
|
|
serviceYAML, err := yaml.Marshal(service)
|
|
require.NoError(t, err)
|
|
fmt.Printf("Service: \n%s\n---\n", string(serviceYAML))
|
|
|
|
listServices, err := db.ListServices(ctx)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, listServices)
|
|
|
|
listServicesYAML, err := yaml.Marshal(listServices)
|
|
require.NoError(t, err)
|
|
fmt.Printf("listServices: \n%s\n", string(listServicesYAML))
|
|
|
|
err = db.DeleteServiceByID(ctx, serviceID)
|
|
require.NoError(t, err)
|
|
}
|