1 /*

     2  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    20  * or visit www.oracle.com if you need additional information or have any

    21  * questions.

    22  */

    23 

    24 /**

    25  * @test

    26  * @bug 8144733

    27  * @summary Verify that Scope.remove removes the Symbol also from already running iterations.

    28  * @modules jdk.compiler/com.sun.tools.javac.code

    29  *          jdk.compiler/com.sun.tools.javac.util

    30  */

    31 

    32 import java.util.ArrayList;

    33 import java.util.Arrays;

    34 import java.util.Collections;

    35 import java.util.Iterator;

    36 import java.util.List;

    37 import java.util.function.Function;

    38 

    39 import com.sun.tools.javac.code.Scope;

    40 import com.sun.tools.javac.code.Scope.WriteableScope;

    41 import com.sun.tools.javac.code.Symbol;

    42 import com.sun.tools.javac.code.Symbol.PackageSymbol;

    43 import com.sun.tools.javac.util.Context;

    44 import com.sun.tools.javac.util.Name;

    45 import com.sun.tools.javac.util.Names;

    46 

    47 public class IterateAndRemove {

    48     public static void main(String... args) {

    49         new IterateAndRemove().run();

    50     }

    51 

    52     void run() {

    53         Context ctx = new Context();

    54         Names names = Names.instance(ctx);

    55         Symbol root = new PackageSymbol(names.empty, null);

    56         Name one = names.fromString("1");

    57         PackageSymbol sym1 = new PackageSymbol(one, new PackageSymbol(names.fromString("a"), root));

    58         PackageSymbol sym2 = new PackageSymbol(one, new PackageSymbol(names.fromString("b"), root));

    59         PackageSymbol sym3 = new PackageSymbol(one, new PackageSymbol(names.fromString("c"), root));

    60         List<Symbol> symbols = Arrays.asList(sym1, sym2, sym3);

    61 

    62         List<Function<Scope, Iterable<Symbol>>> getters = Arrays.asList(

    63                 scope -> scope.getSymbols(),

    64                 scope -> scope.getSymbolsByName(one)

    65         );

    66         for (Function<Scope, Iterable<Symbol>> scope2Content : getters) {

    67             for (int removeAt : new int[] {0, 1, 2, 3}) {

    68                 for (Symbol removeWhat : new Symbol[] {sym1, sym2, sym3}) {

    69                     WriteableScope s = WriteableScope.create(root);

    70 

    71                     symbols.forEach(s :: enter);

    72 

    73                     Iterator<Symbol> it = scope2Content.apply(s).iterator();

    74                     List<PackageSymbol> actual = new ArrayList<>();

    75                     int count = 0;

    76 

    77                     while (true) {

    78                         if (count++ == removeAt)

    79                             s.remove(removeWhat);

    80                         if (!it.hasNext())

    81                             break;

    82                         actual.add((PackageSymbol) it.next());

    83                     }

    84 

    85                     List<Symbol> copy = new ArrayList<>(symbols);

    86 

    87                     Collections.reverse(copy);

    88 

    89                     count = 0;

    90 

    91                     while (true) {

    92                         if (count == removeAt && copy.indexOf(removeWhat) >= count)

    93                             copy.remove(removeWhat);

    94                         count++;

    95                         if (count >= copy.size())

    96                             break;

    97                     }

    98 

    99                     if (!copy.equals(actual)) {

   100                         throw new AssertionError("differs: actual: " + actual + "; expected: " + copy);

   101                     }

   102                 }

   103             }

   104         }

   105     }

   106 }