Add new initialiser to help repositories

brokenhandsio · Apr 17, 2020 · f79c677 · f79c677
1 parent 4ca3b12
commit f79c677
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/Sources/SteamPress/Models/BlogPost.swift b/Sources/SteamPress/Models/BlogPost.swift
@@ -29,6 +29,18 @@ public final class BlogPost: Codable {
         self.lastEdited = nil
         self.published = published
     }
+
+    public init(blogID: Int? = nil, title: String, contents: String, authorID: Int, creationDate: Date, slugUrl: String,
+                published: Bool) {
+        self.blogID = blogID
+        self.title = title
+        self.contents = contents
+        self.author = authorID
+        self.created = creationDate
+        self.slugUrl = slugUrl
+        self.lastEdited = nil
+        self.published = published
+    }
 }
 
 // MARK: - BlogPost Utilities

diff --git a/Tests/SteamPressTests/BlogTests/PostTests.swift b/Tests/SteamPressTests/BlogTests/PostTests.swift
@@ -77,4 +77,9 @@ class PostTests: XCTestCase {
         XCTAssertEqual(tags.first?.name, tag1Name)
         XCTAssertEqual(tags.last?.name, tag2Name)
     }
+
+    func testExtraInitialiserWorks() throws {
+        let post = BlogPost(blogID: 1, title: "title", contents: "contents", authorID: 1, creationDate: Date(), slugUrl: "slug-url", published: true)
+        XCTAssertEqual(post.blogID, 1)
+    }
 }