jpa/eclipselink.jpa.test.jse/src/it/java/org/eclipse/persistence/jpa/test/ddl/TestDDLGenLineTerminator.java

/*
 * Copyright (c) 2015, 2020 Oracle and/or its affiliates. All rights reserved.
 * Copyright (c) 2015, 2019 IBM Corporation. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0,
 * or the Eclipse Distribution License v. 1.0 which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
 */

// Contributors:
//     01/13/2015 - Rick Curtis
//       - 438871 : Add support for writing statement terminator character(s) when generating ddl to script.
package org.eclipse.persistence.jpa.test.ddl;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.util.HashMap;
import java.util.Map;

import jakarta.persistence.EntityManagerFactory;

import org.eclipse.persistence.config.PersistenceUnitProperties;
import org.eclipse.persistence.internal.jpa.EntityManagerFactoryImpl;
import org.eclipse.persistence.jpa.test.basic.model.Dog;
import org.eclipse.persistence.jpa.test.basic.model.Person;
import org.eclipse.persistence.jpa.test.framework.DDLGen;
import org.eclipse.persistence.jpa.test.framework.Emf;
import org.eclipse.persistence.jpa.test.framework.EmfRunner;
import org.eclipse.persistence.jpa.test.framework.PUPropertiesProvider;
import org.eclipse.persistence.jpa.test.framework.Property;
import org.eclipse.persistence.platform.database.DatabasePlatform;
import org.junit.Assert;
import org.junit.Assume;
import org.junit.Test;
import org.junit.runner.RunWith;

@RunWith(EmfRunner.class)
public class TestDDLGenLineTerminator implements PUPropertiesProvider {
    private static File terminatedScript, unterminatedScript;

    static {
        try {
            terminatedScript = File.createTempFile("TestDDLGen-terminated", ".sql.txt");
            unterminatedScript = File.createTempFile("TestDDLGen-unterminated", ".sql.txt");
            unterminatedScript.deleteOnExit();
        } catch (Exception e) {
            Assert.fail("Unable to run test due to " + e);
        }
    }

    @Emf(name = "terminated", createTables = DDLGen.NONE, classes = { Person.class, Dog.class },
            properties = {
                    @Property(name = "jakarta.persistence.schema-generation.scripts.action", value = "create"),
                    @Property(name = PersistenceUnitProperties.SCHEMA_GENERATION_SCRIPT_TERMINATE_STATEMENTS,
                            value = "true"), })
    private EntityManagerFactory terminatedEmf;

    @Emf(name = "unterminated", createTables = DDLGen.NONE, classes = { Person.class, Dog.class },
            properties = { @Property(name = "jakarta.persistence.schema-generation.scripts.action", value = "create") })
    private EntityManagerFactory unterminatedEmf;

    @Override
    public Map<String, Object> getAdditionalPersistenceProperties(String puName) {
        Map<String, Object> res = new HashMap<String, Object>();
        String file;
        if (puName.equals("terminated")) {
            file = terminatedScript.getAbsolutePath();
        } else {
            file = unterminatedScript.getAbsolutePath();
        }
        res.put(PersistenceUnitProperties.SCHEMA_GENERATION_SCRIPTS_CREATE_TARGET, file);
        res.put(PersistenceUnitProperties.SCHEMA_GENERATION_SCRIPTS_DROP_TARGET, file);
        return res;
    }

    @Test
    public void testGeneratedScripts() throws Exception {
        DatabasePlatform platform = ((EntityManagerFactoryImpl) terminatedEmf).getDatabaseSession().getPlatform();
        String terminatorToken = platform.getStoredProcedureTerminationToken();
        Assume.assumeFalse("Platform " + platform + " is not supported for this test", terminatorToken.isEmpty());

        BufferedReader reader = new BufferedReader(new FileReader(terminatedScript));
        String line = reader.readLine();
        while (line != null) {
            Assert.assertTrue(line.endsWith(terminatorToken));
            line = reader.readLine();
        }
        reader.close();

        reader = new BufferedReader(new FileReader(unterminatedScript));
        line = reader.readLine();
        while (line != null) {
            Assert.assertFalse(line.endsWith(terminatorToken));
            line = reader.readLine();
        }
        reader.close();
    }
}