feat: add media type for random meme model

daoseng33 · Oct 8, 2024 · e45c201 · e45c201
1 parent 0f7deb1
commit e45c201
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/Sources/HumorAPIService/General/Utility.swift b/Sources/HumorAPIService/General/Utility.swift
@@ -15,4 +15,16 @@ public struct Utility {
         }
         return data
     }
+
+    /// type string example: "video/mp4", "image/jpeg"
+    static func getMediaType(with typeString: String) -> MemeMediaType {
+        // get "video" or "image"
+        let components = typeString.components(separatedBy: "/")
+        guard let firstPart = components.first,
+                let type = MemeMediaType(rawValue: firstPart) else {
+            return .image
+        }
+
+        return type
+    }
 }
diff --git a/Sources/HumorAPIService/Models/RandomMeme.swift b/Sources/HumorAPIService/Models/RandomMeme.swift
@@ -16,6 +16,9 @@ public class RandomMeme: Object, Decodable, TimeSensitive, Favoritable {
     }
     @Persisted public var urlString: String
     @Persisted public var type: String
+    public var mediaType: MemeMediaType {
+        return Utility.getMediaType(with: type)
+    }
     @Persisted public var createdAt: Date = Date()
     @Persisted public var isFavorite: Bool = false
 

diff --git a/Tests/HumorAPIServiceTests/MemeAPITests.swift b/Tests/HumorAPIServiceTests/MemeAPITests.swift
@@ -22,6 +22,7 @@ struct MemeAPITests: @unchecked Sendable {
         case .success(let random):
             assert(random.id == 831819)
             assert(random.type == "image/png")
+            assert(random.mediaType == .image)
 
         case .failure(_):
             break
@@ -35,6 +36,7 @@ struct MemeAPITests: @unchecked Sendable {
         case .success(let random):
             assert(random.id == 12142)
             assert(random.type == "video/mp4")
+            assert(random.mediaType == .video)
 
         case .failure(_):
             break