Add tests for setGap with .column and .all directions

Author: TaekH

Tests/FlexLayoutTests/WidthSizeContentTests.swift

@@ -19,6 +19,7 @@ final class WidthSizeContentTests: XCTestCase {
     var rootFlexContainer: UIView!
     var aView: UIView!
     var bView: UIView!
+    var cView: UIView!
 
     override func setUp() {
         super.setUp()
@@ -31,6 +32,7 @@ final class WidthSizeContentTests: XCTestCase {
 
         aView = UIView()
         bView = UIView()
+        cView = UIView()
     }
 
     func test_basis_adjust_aView_size_and_position() {
@@ -162,7 +164,7 @@ final class WidthSizeContentTests: XCTestCase {
         XCTAssertEqual(aView.frame, CGRect(x: 0.0, y: 0.0, width: 400.0, height: 200.0))
     }
 
-    func test_gap() {
+    func test_gap_in_row_direction() {
         rootFlexContainer.flex.direction(.row).define { flex in
             flex.addItem(aView).height(100).width(100)
             flex.addItem(bView).height(100).width(100)
@@ -174,4 +176,35 @@ final class WidthSizeContentTests: XCTestCase {
         rootFlexContainer.flex.layout()
         XCTAssertEqual(bView.frame, CGRect(x: 150.0, y: 0.0, width: 100.0, height: 100.0))
     }
+
+    func test_gap_in_column_direction() {
+        rootFlexContainer.flex.direction(.column).define { flex in
+            flex.addItem(aView).height(100).width(100)
+            flex.addItem(bView).height(100).width(100)
+            flex.alignItems(.start)
+            
+            flex.setGap(.column, 50)
+        }
+
+        rootFlexContainer.flex.layout()
+        XCTAssertEqual(bView.frame, CGRect(x: 0.0, y: 150.0, width: 100.0, height: 100.0))
+    }
+    
+    func test_gap_in_all_direction() {
+        rootFlexContainer.flex.direction(.column).define { flex in
+            flex.addItem(aView).height(100).width(100)
+            flex.addItem(bView).height(100).width(100)
+            flex.addItem().direction(.row).define { flex in
+                flex.addItem(cView).height(100).width(100)
+            }
+            flex.alignItems(.start)
+            
+            flex.setGap(.all, 50)
+        }
+
+        rootFlexContainer.flex.layout()
+
+        XCTAssertEqual(bView.frame, CGRect(x: 0.0, y: 150.0, width: 100.0, height: 100.0))
+        XCTAssertEqual(cView.frame, CGRect(x: 150.0, y: 0.0, width: 100.0, height: 100.0))
+    }
 }