compact method

kemsky · Nov 13, 2015 · 58dda9c · 58dda9c
1 parent ea2284e
commit 58dda9c
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 5 deletions.
diff --git a/source/com/kemsky/Stream.as b/source/com/kemsky/Stream.as
@@ -2,6 +2,7 @@ package com.kemsky
 {
     import com.kemsky.filters._;
     import com.kemsky.filters.defined;
+    import com.kemsky.filters.existing;
     import com.kemsky.support.*;
 
     import flash.utils.ByteArray;
@@ -182,8 +183,11 @@ package com.kemsky
         }
 
         /**
-         * Creates a new list for all items that are not strictly equal to undefined (item !== <i>undefined</i>).
+         * Creates a new list for all items that match <i>defined</i> or <i>existing</i> filters depending on undefinedOnly parameter.
+         * @param undefinedOnly use <i>defined</i> filter
          * @return A new list that contains all items from the original list that are not equal to <i>undefined</i>.
+         * @see com.kemsky.filters.defined
+         * @see com.kemsky.filters.existing
          * @example
          * <pre>
          *     var s:Stream = new Stream();
@@ -195,9 +199,16 @@ package com.kemsky
          * </pre>
          * @internal immutable
          */
-        public function compact():Stream
+        public function compact(undefinedOnly:Boolean = false):Stream
         {
-            return filter(defined(_));
+            if(undefinedOnly)
+            {
+                return filter(defined(_));
+            }
+            else
+            {
+                return filter(existing(_));
+            }
         }
 
         /**

diff --git a/testSrc/com/kemsky/TestStream.as b/testSrc/com/kemsky/TestStream.as
@@ -131,12 +131,23 @@ package com.kemsky
         {
             var s:Stream = $();
             s[0] = 1;
-            s[10] = 2;
+            s[3] = 2;
 
-            var c:Stream = s.compact();
+            var c:Stream = s.compact(true);
             assertEquals(c.length, 2);
             assertEquals(c.first, 1);
             assertEquals(c.second, 2);
+
+            var s1:Stream = $();
+            s1[0] = 1;
+            s1[1] = null;
+            s1[2] = NaN;
+            s1[3] = 2;
+
+            var c1:Stream = s1.compact();
+            assertEquals(c1.length, 2);
+            assertEquals(c1.first, 1);
+            assertEquals(c1.second, 2);
         }
 
         [Test]