Merge pull request #30 from emirhanbattalbas/feature/tableview

Tableview's initine style selection added
mobillium · May 7, 2021 · e293fd4 · e293fd4
2 parents 42233cd + 9aed6f9
commit e293fd4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/Example/Tests/UITableViewBuilderTests.swift b/Example/Tests/UITableViewBuilderTests.swift
@@ -175,7 +175,15 @@ class UITableViewBuilderTests: XCTestCase {
 
         XCTAssertEqual(tableView.accessibilityIdentifier, accessibilityIdentifier)
     }
-
+
+    func testStyle() {
+        let style: UITableView.Style = .grouped
+        let tableView = UITableViewBuilder(style)
+            .build()
+
+        XCTAssertEqual(tableView.style, style)
+    }
+
     func testCellRegistration() {
         let cellIdentifier = "TestTableViewCell"
         let tableView = UITableViewBuilder()
@@ -193,7 +201,6 @@ class UITableViewBuilderTests: XCTestCase {
                 break
             }
         }
-
         XCTAssertTrue(isContainsIdentifier)
     }
 }
diff --git a/MobilliumBuilders/Classes/UITableViewBuilder.swift b/MobilliumBuilders/Classes/UITableViewBuilder.swift
@@ -8,11 +8,13 @@
 import UIKit
 
 public class UITableViewBuilder<T: UITableView> {
-
-    public let tableView = T()
-
-    public init() { }
-
+
+    public var tableView = T()
+
+    public init(_ style: UITableView.Style = .plain) {
+        tableView = T(frame: .zero, style: style)
+    }
+
     @discardableResult
     public func backgroundColor(_ backgroundColor: UIColor) -> Self {
         self.tableView.backgroundColor = backgroundColor