ory-am/hydra

// Copyright © 2022 Ory Corp
// SPDX-License-Identifier: Apache-2.0

package x

import (
    "testing"

    "github.com/gobuffalo/pop/v6"
)

func DeleteHydraRows(t *testing.T, c *pop.Connection) {
    for _, tb := range []string{
        "hydra_oauth2_access",
        "hydra_oauth2_refresh",
        "hydra_oauth2_code",
        "hydra_oauth2_oidc",
        "hydra_oauth2_pkce",
        "hydra_oauth2_flow",
        "hydra_oauth2_authentication_session",
        "hydra_oauth2_obfuscated_authentication_session",
        "hydra_oauth2_logout_request",
        "hydra_oauth2_jti_blacklist",
        "hydra_oauth2_trusted_jwt_bearer_issuer",
        "hydra_jwk",
        "hydra_client",
    } {
        if err := c.RawQuery("DELETE FROM " + tb).Exec(); err != nil {
            t.Logf(`Unable to delete rows in table "%s": %s`, tb, err)
        }
    }
}

func CleanSQLPop(t *testing.T, c *pop.Connection) {
    t.Logf("Cleaning up database: %s", c.Dialect.Name())
    for _, tb := range []string{
        "hydra_oauth2_access",
        "hydra_oauth2_refresh",
        "hydra_oauth2_code",
        "hydra_oauth2_oidc",
        "hydra_oauth2_pkce",
        "hydra_oauth2_flow",
        "hydra_oauth2_authentication_session",
        "hydra_oauth2_obfuscated_authentication_session",
        "hydra_oauth2_logout_request",
        "hydra_oauth2_jti_blacklist",
        "hydra_oauth2_trusted_jwt_bearer_issuer",
        "hydra_jwk",
        "hydra_client",
        // Migrations
        "hydra_oauth2_authentication_consent_migration",
        "hydra_client_migration",
        "hydra_oauth2_migration",
        "hydra_jwk_migration",
        "networks",
        "schema_migration",
    } {
        if err := c.RawQuery("DROP TABLE IF EXISTS " + tb).Exec(); err != nil {
            t.Fatalf(`Unable to clean up table "%s": %s`, tb, err)
        }
    }
    t.Logf("Successfully cleaned up database: %s", c.Dialect.Name())
}