test(BUX-175): add separate test or checking beef capability

BuxOrg · Sep 11, 2023 · e709f46 · e709f46
1 parent 585b456
commit e709f46
Showing 1 changed file with 28 additions and 2 deletions.
diff --git a/paymail_test.go b/paymail_test.go
@@ -109,6 +109,32 @@ func Test_hasP2P(t *testing.T) {
 	t.Parallel()
 
 	t.Run("no p2p capabilities", func(t *testing.T) {
+		capabilities := server.GenericCapabilities(paymail.DefaultBsvAliasVersion, false)
+		success, _, p2pDestinationURL, p2pSubmitTxURL := hasP2P(capabilities)
+		assert.Equal(t, false, success)
+		assert.Equal(t, "", p2pDestinationURL)
+		assert.Equal(t, "", p2pSubmitTxURL)
+	})
+
+	t.Run("valid p2p capabilities", func(t *testing.T) {
+		capabilities := server.GenericCapabilities(paymail.DefaultBsvAliasVersion, false)
+
+		// Add the P2P
+		capabilities.Capabilities[paymail.BRFCP2PTransactions] = "/receive-transaction/{alias}@{domain.tld}"
+		capabilities.Capabilities[paymail.BRFCP2PPaymentDestination] = "/p2p-payment-destination/{alias}@{domain.tld}"
+
+		success, _, p2pDestinationURL, p2pSubmitTxURL := hasP2P(capabilities)
+		assert.Equal(t, true, success)
+		assert.Equal(t, capabilities.Capabilities[paymail.BRFCP2PPaymentDestination], p2pDestinationURL)
+		assert.Equal(t, capabilities.Capabilities[paymail.BRFCP2PTransactions], p2pSubmitTxURL)
+	})
+}
+
+// Test_hasP2P_beefCapabilities will test the method hasP2P() but with BEEF capabilities
+func Test_hasP2P_beefCapabilities(t *testing.T) {
+	t.Parallel()
+
+	t.Run("no beef capabilities", func(t *testing.T) {
 		capabilities := server.GenericCapabilities(paymail.DefaultBsvAliasVersion, false)
 		success, useBeef, p2pDestinationURL, p2pSubmitTxURL := hasP2P(capabilities)
 		assert.Equal(t, false, success)
@@ -121,14 +147,14 @@ func Test_hasP2P(t *testing.T) {
 		capabilities := server.GenericCapabilities(paymail.DefaultBsvAliasVersion, false)
 
 		// Add the P2P
-		capabilities.Capabilities[paymail.BRFCP2PTransactions] = "/receive-transaction/{alias}@{domain.tld}"
 		capabilities.Capabilities[paymail.BRFCP2PPaymentDestination] = "/p2p-payment-destination/{alias}@{domain.tld}"
+		capabilities.Capabilities[paymail.BRFCBeefTransaction] = "/receive-beef-transaction/{alias}@{domain.tld}"
 
 		success, useBeef, p2pDestinationURL, p2pSubmitTxURL := hasP2P(capabilities)
 		assert.Equal(t, true, success)
 		assert.Equal(t, true, useBeef)
 		assert.Equal(t, capabilities.Capabilities[paymail.BRFCP2PPaymentDestination], p2pDestinationURL)
-		assert.Equal(t, capabilities.Capabilities[paymail.BRFCP2PTransactions], p2pSubmitTxURL)
+		assert.Equal(t, capabilities.Capabilities[paymail.BRFCBeefTransaction], p2pSubmitTxURL)
 	})
 }