diff --git a/tests/admin/conftest.py b/tests/admin/conftest.py index 5b0b67be..bb0e0f84 100644 --- a/tests/admin/conftest.py +++ b/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, + ) diff --git a/tests/admin/test_messages.py b/tests/admin/test_messages.py index edfe8115..3fb35cf8 100644 --- a/tests/admin/test_messages.py +++ b/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 diff --git a/tests/conftest.py b/tests/conftest.py index ebce3245..c8a42f3e 100644 --- a/tests/conftest.py +++ b/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, + )