Add test for payment summary to msg (#1609)

tests/admin/conftest.py

@@ -184,3 +184,18 @@ def squeak_detail_msg(
     yield squeak_admin_pb2.SqueakDetailEntry(
         serialized_squeak_hex=squeak_bytes.hex(),
     )
+
+
+@pytest.fixture
+def payment_summary_msg(
+        num_received_payments,
+        num_sent_payments,
+        total_amount_received_msat,
+        total_amount_sent_msat,
+):
+    yield squeak_admin_pb2.PaymentSummary(
+        num_received_payments=num_received_payments,
+        num_sent_payments=num_sent_payments,
+        amount_earned_msat=total_amount_received_msat,
+        amount_spent_msat=total_amount_sent_msat,
+    )

tests/admin/test_messages.py

@@ -23,6 +23,7 @@ from squeaknode.admin.messages import message_to_peer_address
 from squeaknode.admin.messages import message_to_received_payment
 from squeaknode.admin.messages import message_to_sent_payment
 from squeaknode.admin.messages import message_to_squeak_entry
+from squeaknode.admin.messages import payment_summary_to_message
 from squeaknode.admin.messages import peer_address_to_message
 from squeaknode.admin.messages import received_offer_to_message
 from squeaknode.admin.messages import received_payment_to_message
@@ -118,3 +119,16 @@ def test_squeak_detail_to_message(squeak, squeak_detail_msg):
     msg = squeak_to_detail_message(squeak)
 
     assert msg == squeak_detail_msg
+
+
+def test_payment_summary_to_message(
+        received_payment_summary,
+        sent_payment_summary,
+        payment_summary_msg,
+):
+    msg = payment_summary_to_message(
+        received_payment_summary,
+        sent_payment_summary,
+    )
+
+    assert msg == payment_summary_msg

tests/conftest.py

@@ -32,10 +32,12 @@ from squeaknode.core.peer_address import Network
 from squeaknode.core.peer_address import PeerAddress
 from squeaknode.core.received_offer import ReceivedOffer
 from squeaknode.core.received_payment import ReceivedPayment
+from squeaknode.core.received_payment_summary import ReceivedPaymentSummary
 from squeaknode.core.secret_keys import add_tweak
 from squeaknode.core.secret_keys import generate_tweak
 from squeaknode.core.sent_offer import SentOffer
 from squeaknode.core.sent_payment import SentPayment
+from squeaknode.core.sent_payment_summary import SentPaymentSummary
 from squeaknode.core.squeak_entry import SqueakEntry
 from squeaknode.core.squeak_peer import SqueakPeer
 from squeaknode.core.squeaks import get_hash
@@ -414,3 +416,45 @@ def offer(
         host=lightning_address.host,
         port=lightning_address.port,
     )
+
+
+@pytest.fixture
+def num_received_payments():
+    yield 574
+
+
+@pytest.fixture
+def num_sent_payments():
+    yield 189
+
+
+@pytest.fixture
+def total_amount_received_msat():
+    yield 3259084
+
+
+@pytest.fixture
+def total_amount_sent_msat():
+    yield 198374
+
+
+@pytest.fixture
+def received_payment_summary(
+        num_received_payments,
+        total_amount_received_msat,
+):
+    yield ReceivedPaymentSummary(
+        num_received_payments=num_received_payments,
+        total_amount_received_msat=total_amount_received_msat,
+    )
+
+
+@pytest.fixture
+def sent_payment_summary(
+        num_sent_payments,
+        total_amount_sent_msat,
+):
+    yield SentPaymentSummary(
+        num_sent_payments=num_sent_payments,
+        total_amount_sent_msat=total_amount_sent_msat,
+    )