1 /*
2  * Copyright (c) 2018, 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 import java.util.Iterator;
25 import java.util.Set;
26 import java.util.TreeSet;
27 import org.testng.Assert;
28 import org.testng.annotations.Test;
29 
30 /**
31  * @test
32  * @bug 8137326
33  * @summary Test to verify the Comparable implementation for the StringBuffer class.
34  * @run testng Comparison
35  */
36 public class Comparison {
37     static char SEP = ':';
38 
39     static String[][] books = {
40         {"Biography", "Steve Jobs"},
41         {"Biography", "Elon Musk: Tesla, SpaceX, and the Quest for a Fantastic Future"},
42         {"Law", "Law 101: Everything You Need to Know About American Law, Fourth Edition"},
43         {"Law", "The Tools of Argument: How the Best Lawyers Think, Argue, and Win"},
44         {"History", "The History Book (Big Ideas Simply Explained)"},
45         {"History", "A People's History of the United States"},
46     };
47 
48     /**
49      * Verifies the Comparable implementation by comparing with two TreeSet that
50      * contain either StringBuffer or String.
51      */
52     @Test
compareWithString()53     public void compareWithString() {
54         Set<StringBuffer> sbSet = constructSBSet();
55         Set<String> sSet = constructStringSet();
56         Iterator<StringBuffer> iSB = sbSet.iterator();
57         Iterator<String> iS = sSet.iterator();
58         while (iSB.hasNext()) {
59             String temp1 = iSB.next().toString();
60             System.out.println(temp1);
61             String temp2 = iS.next();
62             System.out.println(temp2);
63 
64             Assert.assertTrue(temp1.equals(temp2), "Comparing item by item");
65         }
66 
67     }
68 
69     /**
70      * Compares between StringBuffers
71      */
72     @Test
testCompare()73     public void testCompare() {
74         StringBuffer sb1 = generateTestBuffer(65, 70, 97, 102);
75         StringBuffer sb2 = generateTestBuffer(65, 70, 97, 102);
76         StringBuffer sb3 = generateTestBuffer(65, 71, 97, 103);
77 
78         System.out.println(sb1.toString());
79         System.out.println(sb2.toString());
80         System.out.println(sb3.toString());
81         Assert.assertTrue(sb1.compareTo(sb2) == 0, "Compare sb1 and sb2");
82         Assert.assertFalse(sb1.compareTo(sb3) == 0, "Compare sb1 and sb3");
83     }
84 
85     /**
86      * Verifies that the comparison is from index 0 to length() - 1 of the two
87      * character sequences.
88      */
89     @Test
testModifiedSequence()90     public void testModifiedSequence() {
91         StringBuffer sb1 = generateTestBuffer(65, 70, 97, 102);
92         StringBuffer sb2 = generateTestBuffer(65, 70, 98, 103);
93 
94         // contain different character sequences
95         Assert.assertFalse(sb1.compareTo(sb2) == 0, "Compare the sequences before truncation");
96 
97         // the first 5 characters however are the same
98         sb1.setLength(5);
99         sb2.setLength(5);
100 
101         System.out.println(sb1.toString());
102         System.out.println(sb2.toString());
103 
104         Assert.assertTrue(sb1.compareTo(sb2) == 0, "Compare sb1 and sb2");
105         Assert.assertTrue(sb1.toString().compareTo(sb2.toString()) == 0, "Compare strings of sb1 and sb2");
106     }
107 
constructStringSet()108     private Set<String> constructStringSet() {
109         Set<String> sSet = new TreeSet<>();
110         for (String[] book : books) {
111             sSet.add(book[0] + SEP + book[1]);
112         }
113         return sSet;
114     }
115 
constructSBSet()116     private Set<StringBuffer> constructSBSet() {
117         Set<StringBuffer> sbSet = new TreeSet<>();
118         for (String[] book : books) {
119             sbSet.add(new StringBuffer(book[0]).append(SEP).append(book[1]));
120         }
121         return sbSet;
122     }
123 
generateTestBuffer(int from1, int to1, int from2, int to2)124     private static StringBuffer generateTestBuffer(int from1, int to1,
125             int from2, int to2) {
126         StringBuffer aBuffer = new StringBuffer(50);
127 
128         for (int i = from1; i < to1; i++) {
129             aBuffer.append((char)i);
130         }
131         for (int i = from2; i < to2; i++) {
132             aBuffer.append((char)i);
133         }
134         return aBuffer;
135     }
136 }
137