File: Derby5730Test.java

package info (click to toggle)
derby 10.14.2.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 79,056 kB
  • sloc: java: 691,961; sql: 42,686; xml: 20,512; sh: 3,373; sed: 96; makefile: 60
file content (166 lines) | stat: -rw-r--r-- 6,148 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
/*
 * Derby - Class org.apache.derbyTesting.functionTests.tests.memory.Derby5730Test
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.derbyTesting.functionTests.tests.memory;

import java.io.IOException;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import junit.framework.Test;
import org.apache.derbyTesting.junit.BaseJDBCTestCase;
import org.apache.derbyTesting.junit.BaseTestCase;
import org.apache.derbyTesting.junit.BaseTestSuite;
import org.apache.derbyTesting.junit.JDBC;
import org.apache.derbyTesting.junit.SpawnedProcess;

/**
 * Regression test for DERBY-5730. Repeatedly boot and shut down a database.
 * In between call a function in the SYSFUN schema, a different one each time.
 * Without the fix, a reference to the currently running database instance will
 * be leaked for each iteration, and eventually an OutOfMemoryError is raised.
 */
public class Derby5730Test extends BaseTestCase {

    public Derby5730Test(String name) {
        super(name);
    }

    public static Test suite() {
        // The test case uses DriverManager, so require JDBC 3.0 or higher.
        if (JDBC.vmSupportsJDBC3()) {
            return new BaseTestSuite(Derby5730Test.class);
        }
        return new BaseTestSuite("Derby5730Test - skipped");
    }

    /**
     * Test case for DERBY-5730. The memory leak is only reproduced if the
     * test case runs with capped heap size and the SYSFUN functions have not
     * been called previously in the same JVM process. Spawn a new process to
     * satisfy those requirements.
     */
    public void testLeak() throws IOException {
        String[] cmd = {"-Xmx16M", getClass().getName()};
        SpawnedProcess sp = new SpawnedProcess(execJavaCmd(cmd), "DERBY-5730");
        if (sp.complete() != 0) {
            fail(sp.getFailMessage("Process failed"));
        }
    }

    private final static Integer ZERO = Integer.valueOf("0");
    private final static Integer ONE = Integer.valueOf("1");
    private final static Integer TWO = Integer.valueOf("2");

    /**
     * These are the functions in the SYSFUN schema. The second value in each
     * row tells how many arguments the function takes.
     */
    private final static Object[][] FUNCTIONS = {
        {"ACOS", ONE},
        {"ASIN", ONE},
        {"ATAN", ONE},
        {"ATAN2", TWO},
        {"COS", ONE},
        {"SIN", ONE},
        {"TAN", ONE},
        {"PI", ZERO},
        {"DEGREES", ONE},
        {"RADIANS", ONE},
        {"LN", ONE},
        {"LOG", ONE},
        {"LOG10", ONE},
        {"EXP", ONE},
        {"CEIL", ONE},
        {"CEILING", ONE},
        {"FLOOR", ONE},
        {"SIGN", ONE},
        {"RANDOM", ZERO},
        {"RAND", ONE},
        {"COT", ONE},
        {"COSH", ONE},
        {"SINH", ONE},
        {"TANH", ONE},
    };

    /**
     * Boot and repeatedly reboot a database, calling SYSFUN functions in
     * between. Eventually runs out of memory if DERBY-5730 is not fixed.
     * Must run with capped memory size (-Xmx16M) to expose the memory leak.
     */
    public static void main(String[] args) throws Exception {
        Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
        for (int i = 0; i < FUNCTIONS.length; i++) {
            Connection c = DriverManager.getConnection(
                    "jdbc:derby:memory:derby5730;create=true");
            prepareFunction(c,
                            (String) FUNCTIONS[i][0],
                            ((Integer) FUNCTIONS[i][1]).intValue());
            growDatabaseFootprint(c);
            c.close();
            try {
                DriverManager.getConnection(
                    "jdbc:derby:memory:derby5730;shutdown=true");
                fail("Shutdown should throw exception");
            } catch (SQLException sqle) {
                BaseJDBCTestCase.assertSQLState("08006", sqle);
            }
        }
    }

    /**
     * Prepare a call to a function. Close the statement once it is prepared.
     * Before the bug was fixed, preparing a call to a function in SYSFUN that
     * hadn't been prepared in the same JVM process, would leak a reference to
     * the current database instance.
     */
    private static void prepareFunction(Connection c, String name, int args)
            throws SQLException {
        StringBuffer sql = new StringBuffer("VALUES ");
        sql.append(name);
        sql.append('(');
        for (int i = 0; i < args; i++) {
            if (i > 0) sql.append(',');
            sql.append('?');
        }
        sql.append(')');

        String sqlText = sql.toString();

        System.out.println(sqlText);
        PreparedStatement ps = c.prepareStatement(sqlText);
        ps.close();
    }

    /**
     * Perform some database operations so that the internal structures of
     * the database instance (caches, for example) are filled, and the memory
     * footprint of the database instance grows. This is done to make the
     * test case run out of memory faster.
     */
    private static void growDatabaseFootprint(Connection c)
            throws SQLException {
        DatabaseMetaData dmd = c.getMetaData();
        JDBC.assertDrainResults(dmd.getColumns(null, "%", "%", "%"));
        JDBC.assertDrainResults(dmd.getTables(null, "%", "%", null));
    }
}