implement truncate string

2 files changed, 16 insertions, 2 deletions

diff --git a/src/Q1/ReversedSentence.java b/src/Q1/ReversedSentence.java
index 13483d3..d218c8c 100644
--- a/src/Q1/ReversedSentence.java
+++ b/src/Q1/ReversedSentence.java
@@ -1,6 +1,5 @@
 package Q1;
 
-import java.util.ArrayList;
 import java.util.Scanner;
 
 public class ReversedSentence {
@@ -40,7 +39,7 @@ public class ReversedSentence {
   }
 
   public static String truncateSentence(String s) {
-    return s;
+    return s.substring(0, Math.min(s.length(), 80));
   }
 
   public static void main(String[] args) {
diff --git a/src/Q1/ReversedSentenceTest.java b/src/Q1/ReversedSentenceTest.java
index 26fc7a1..a7a3988 100644
--- a/src/Q1/ReversedSentenceTest.java
+++ b/src/Q1/ReversedSentenceTest.java
@@ -51,4 +51,19 @@ public class ReversedSentenceTest extends TestCase {
             new String(matrix[2]));
     assertEquals(expected, ReversedSentence.printChar2DArray(matrix));
   }
+
+  public void testTruncateSentence() {
+    String sentence = "mary had a little lamb";
+    assertEquals(sentence, ReversedSentence.truncateSentence(sentence));
+
+    String result = ReversedSentence.truncateSentence(createString(90, 'x'));
+    assertEquals(80, result.length());
+    assertEquals(createString(80, 'x'), result);
+  }
+
+  private String createString(int length, char character) {
+    StringBuffer buffer = new StringBuffer(length);
+    for (int i = 0; i < length; i++) buffer.append(character);
+    return buffer.toString();
+  }
 }