AvailabilityFactory

billypchan · May 31, 2021 · 136c16c · 136c16c
1 parent 1f8c80d
commit 136c16c
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/Source/Utilis/Protos/GenericMessage+Helper.swift b/Source/Utilis/Protos/GenericMessage+Helper.swift
@@ -530,7 +530,7 @@ public enum MentionFactory {
 
 // MARK: - Availability
 
-extension WireProtos.Availability {
+public enum AvailabilityFactory {
     public static func createAvailability(availability: Availability) -> WireProtos.Availability {
         return WireProtos.Availability.with {
             switch availability {

diff --git a/Tests/Source/Model/Messages/GenericMessageTests.swift b/Tests/Source/Model/Messages/GenericMessageTests.swift
@@ -38,7 +38,7 @@ class GenericMessageTests: XCTestCase {
             { return GenericMessage(content: Location(latitude: 1, longitude: 2)) },
             { return GenericMessage(content: MessageDelete(messageId: UUID.create())) },
             { return GenericMessage(content: WireProtos.Reaction.createReaction(emoji: "test", messageID: UUID.create())) },
-            { return GenericMessage(content: WireProtos.Availability.createAvailability(availability: .away)) }
+            { return GenericMessage(content: AvailabilityFactory.createAvailability(availability: .away)) }
         ]
 
         generators.forEach { generator in

diff --git a/Tests/Source/Model/Messages/ZMClientMessageTests.swift b/Tests/Source/Model/Messages/ZMClientMessageTests.swift
@@ -378,7 +378,7 @@ extension ClientMessageTests {
         let senderClientID = NSString.createAlphanumerical()
         let conversation = ZMConversation.insertNewObject(in: self.uiMOC)
         conversation.remoteIdentifier = UUID.create()
-        let availability = WireProtos.Availability.createAvailability(availability: .away)
+        let availability = AvailabilityFactory.createAvailability(availability: .away)
         let contentData = try? GenericMessage(content: availability, nonce: UUID.create()).serializedData()
         let data: NSDictionary = [
             "sender": senderClientID,

diff --git a/Tests/Source/Model/User/ZMUserTests+Swift.swift b/Tests/Source/Model/User/ZMUserTests+Swift.swift
@@ -466,7 +466,7 @@ extension ZMUserTests_Swift {
         // given
         let user = ZMUser.insert(in: self.uiMOC, name: "Foo")
         XCTAssertEqual(user.availability, .none)
-        let availability = WireProtos.Availability.createAvailability(availability: .away)
+        let availability = AvailabilityFactory.createAvailability(availability: .away)
         // when
         user.updateAvailability(from: GenericMessage(content: availability))