29 lines
771 B
Go
29 lines
771 B
Go
package customtemplate_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/dataservices"
|
|
"github.com/portainer/portainer/api/datastore"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCustomTemplateCreateTx(t *testing.T) {
|
|
_, ds := datastore.MustNewTestStore(t, true, false)
|
|
require.NotNil(t, ds)
|
|
|
|
require.NoError(t, ds.UpdateTx(func(tx dataservices.DataStoreTx) error {
|
|
return tx.CustomTemplate().Create(&portainer.CustomTemplate{ID: 1})
|
|
}))
|
|
|
|
var template *portainer.CustomTemplate
|
|
require.NoError(t, ds.ViewTx(func(tx dataservices.DataStoreTx) error {
|
|
var err error
|
|
template, err = tx.CustomTemplate().Read(1)
|
|
return err
|
|
}))
|
|
|
|
require.Equal(t, portainer.CustomTemplateID(1), template.ID)
|
|
}
|