Round-trip test: DomainRegistration <-> DomainRegistration'

wireapp · Feb 4, 2025 · 96afdab · 96afdab
1 parent aaf8b66
commit 96afdab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/libs/wire-api/src/Wire/API/EnterpriseLogin.hs b/libs/wire-api/src/Wire/API/EnterpriseLogin.hs
@@ -239,13 +239,17 @@ data DomainRegistration' = DomainRegistration'
     dnsVerificationToken :: Maybe DnsVerificationToken,
     authTokenHash :: Maybe Token
   }
+  deriving (Eq, Show, Generic)
+  deriving (Arbitrary) via (GenericUniform DomainRegistration')
 
 data DomainRegistrationSettings'
   = Locked'
   | PreAuthorized'
   | NoRegistration'
   | DomainForBackend HttpsUrl
   | DomainForLocalTeam TeamId (Maybe SAML.IdPId)
+  deriving (Eq, Show, Generic)
+  deriving (Arbitrary) via (GenericUniform DomainRegistrationSettings')
 
 -- | we need this if there is no entry in the table.
 defDomainRegistration :: Domain -> DomainRegistration

diff --git a/libs/wire-api/test/unit/Test/Wire/API/Roundtrip/Aeson.hs b/libs/wire-api/test/unit/Test/Wire/API/Roundtrip/Aeson.hs
@@ -36,6 +36,7 @@ import Wire.API.Conversation.Member qualified as Conversation.Member
 import Wire.API.Conversation.Role qualified as Conversation.Role
 import Wire.API.Conversation.Typing qualified as Conversation.Typing
 import Wire.API.CustomBackend qualified as CustomBackend
+import Wire.API.EnterpriseLogin qualified as EnterpriseLogin
 import Wire.API.Event.Conversation qualified as Event.Conversation
 import Wire.API.Event.Team qualified as Event.Team
 import Wire.API.Event.WebSocketProtocol qualified as EventWebSocketProtocol
@@ -349,7 +350,9 @@ tests =
       testRoundTrip @TeamsIntra.TeamStatus,
       testRoundTrip @TeamsIntra.TeamStatusUpdate,
       testRoundTrip @TeamsIntra.TeamData,
-      testRoundTrip @TeamsIntra.TeamName
+      testRoundTrip @TeamsIntra.TeamName,
+      testProperty "EnterpriseLogin.EnterpriseLogin'" $ \(dom, new) -> do
+        EnterpriseLogin.oldToNew (EnterpriseLogin.newToOld dom new) === Right (dom, new)
     ]
 
 testRoundTrip ::