1 files changed, 33 insertions, 0 deletions
diff --git a/WordPress/src/androidTest/java/org/wordpress/android/ui/posts/PostUtilsTest.java b/WordPress/src/androidTest/java/org/wordpress/android/ui/posts/PostUtilsTest.java
new file mode 100644
index 000000000..d251d9f28
--- /dev/null
+++ b/WordPress/src/androidTest/java/org/wordpress/android/ui/posts/PostUtilsTest.java
@@ -0,0 +1,33 @@
+package org.wordpress.android.ui.posts;
+
+import android.test.AndroidTestCase;
+
+public class PostUtilsTest extends AndroidTestCase {
+    public void testCollapseShortcodes() {
+        String postContent = "Text before first gallery [gallery number=\"one\"]"
+                + " text between galleries"
+                + " [gallery number=\"two\"]"
+                + " text after second gallery"
+                + " [unknown shortcode].";
+        String collapsedContent = PostUtils.collapseShortcodes(postContent);
+
+        // make sure [gallery] now exists and [gallery number] does not
+        assertTrue(collapsedContent.contains("[gallery]"));
+        assertFalse(collapsedContent.contains("[gallery number]"));
+
+        // make sure the unknown shortcode is intact
+        assertTrue(collapsedContent.contains("[unknown shortcode]"));
+    }
+
+    public void testShortcodeSpaces() {
+        String postContent = "[   gallery number=\"arst\"     /]";
+        String collapsedContent = PostUtils.collapseShortcodes(postContent);
+        assertEquals("[gallery]", collapsedContent);
+    }
+
+    public void testOpeningClosingShortcode() {
+        String postContent = "[recipe difficulty=\"easy\"]Put your recipe here.[/recipe]";
+        String collapsedContent = PostUtils.collapseShortcodes(postContent);
+        assertEquals("[recipe]Put your recipe here.[/recipe]", collapsedContent);
+    }
+}