mirror of
https://github.com/fleetdm/fleet
synced 2026-05-22 00:18:27 +00:00
f62d5eda20
for #18427 # Checklist for submitter If some of the following don't apply, delete the relevant line. <!-- Note that API documentation changes are now addressed by the product design team. --> - [x] Changes file added for user-visible changes in `changes/`, `orbit/changes/` or `ee/fleetd-chrome/changes`. See [Changes files](https://fleetdm.com/docs/contributing/committing-changes#changes-files) for more information. - [x] Added/updated tests - [ ] Manual QA for all new/changed functionality
62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
package mysql
|
|
|
|
import (
|
|
"crypto/x509"
|
|
"crypto/x509/pkix"
|
|
"math/big"
|
|
"testing"
|
|
|
|
scep_depot "github.com/fleetdm/fleet/v4/server/mdm/scep/depot"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setup(t *testing.T) scep_depot.Depot {
|
|
ds := CreateNamedMySQLDS(t, t.Name())
|
|
depot, err := ds.NewSCEPDepot()
|
|
require.NoError(t, err)
|
|
return depot
|
|
}
|
|
|
|
func TestAppleMDMSCEPSerial(t *testing.T) {
|
|
depot := setup(t)
|
|
tests := []struct {
|
|
name string
|
|
want *big.Int
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "two is the default value.",
|
|
want: big.NewInt(2),
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
got, err := depot.Serial()
|
|
require.NoError(t, err)
|
|
require.Equal(t, tt.want, got)
|
|
}
|
|
}
|
|
|
|
func TestAppleMDMPutAndHasCN(t *testing.T) {
|
|
depot := setup(t)
|
|
|
|
name := "Fleet Identity"
|
|
serial, err := depot.Serial()
|
|
require.NoError(t, err)
|
|
cert := x509.Certificate{
|
|
SerialNumber: serial,
|
|
Subject: pkix.Name{
|
|
CommonName: name,
|
|
},
|
|
}
|
|
|
|
err = depot.Put(name, &cert)
|
|
require.NoError(t, err)
|
|
|
|
has, err := depot.HasCN(name, 0, &cert, false)
|
|
require.NoError(t, err)
|
|
require.True(t, has)
|
|
|
|
has, err = depot.HasCN("non-existent", 0, &cert, true)
|
|
require.NoError(t, err)
|
|
require.False(t, has)
|
|
}
|