Prepare session policy helper methods

pkg/acceptance/helpers/session_policy_client.go

@@ -1,7 +1,12 @@
 package helpers
 
 import (
+	"context"
+	"testing"
+
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/acceptance/helpers/random"
 	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk"
+	"github.com/stretchr/testify/require"
 )
 
 type SessionPolicyClient struct {
@@ -17,3 +22,32 @@ func NewSessionPolicyClient(context *TestClientContext) *SessionPolicyClient {
 func (c *SessionPolicyClient) client() sdk.SessionPolicies {
 	return c.context.client.SessionPolicies
 }
+
+func (c *SessionPolicyClient) CreateSessionPolicy(t *testing.T, database *sdk.Database, schema *sdk.Schema) (*sdk.SessionPolicy, func()) {
+	t.Helper()
+	id := sdk.NewSchemaObjectIdentifier(database.Name, schema.Name, random.StringN(12))
+	return c.CreateSessionPolicyWithOptions(t, id, sdk.NewCreateSessionPolicyRequest(id))
+}
+
+func (c *SessionPolicyClient) CreateSessionPolicyWithOptions(t *testing.T, id sdk.SchemaObjectIdentifier, request *sdk.CreateSessionPolicyRequest) (*sdk.SessionPolicy, func()) {
+	t.Helper()
+	ctx := context.Background()
+
+	err := c.client().Create(ctx, request)
+	require.NoError(t, err)
+
+	sessionPolicy, err := c.client().ShowByID(ctx, id)
+	require.NoError(t, err)
+
+	return sessionPolicy, c.DropSessionPolicyFunc(t, id)
+}
+
+func (c *SessionPolicyClient) DropSessionPolicyFunc(t *testing.T, id sdk.SchemaObjectIdentifier) func() {
+	t.Helper()
+	ctx := context.Background()
+
+	return func() {
+		err := c.client().Drop(ctx, sdk.NewDropSessionPolicyRequest(id).WithIfExists(sdk.Bool(true)))
+		require.NoError(t, err)
+	}
+}