"""Tests for things a member is likely to do that uses payments"""
import time
from django.urls import reverse
from selenium.webdriver.support.select import Select
from notifications.tests.common_functions import check_email_sent
from payments.views.core import get_balance
from payments.models import MemberTransaction
from payments import forms
from payments.tests.integration.common_functions import (
setup_auto_top_up,
check_balance_for_user,
check_last_transaction_for_user,
stripe_manual_payment_screen,
)
from tests.test_manager import CobaltTestManagerIntegration
[docs]
class MemberTransfer:
"""Member transfer related activities.
These tests cover scenarios where a member transfers money to another
member. This can trigger manual or auto top ups and different validation
rules.
"""
def __init__(self, manager: CobaltTestManagerIntegration):
self.manager = manager
self.client = self.manager.client
self.driver = self.manager.driver
# Log user in
self.manager.login_user(self.manager.test_user)
[docs]
def a1_member_transfer_with_sufficient_funds(self):
"""Transfer to another member with sufficient funds in account"""
# Set up betty to transfer to
betty = self.manager.get_user(username="101")
alan = self.manager.test_user # shorthand
##################################
# Check Alan's balance before
alan_expected_initial_balance = 400.0
check_balance_for_user(
manager=self.manager,
user=alan,
expected_balance=alan_expected_initial_balance,
test_name="Check initial balance for Alan",
test_description="This is the initial check of Alan's balance before we process the transfer.",
)
#################################
# Check Betty's balance before
betty_expected_initial_balance = 404.44
check_balance_for_user(
manager=self.manager,
user=betty,
expected_balance=betty_expected_initial_balance,
test_name="Check initial balance for Betty",
test_description="This is the initial check of Betty's balance before we process the transfer.",
)
################################
# Transfer from Alan to Betty
desc = "Al to Betty, test transfer"
amt = 54.45
# Test the form
form_data = {
"transfer_to": betty,
"amount": amt,
"description": desc,
}
form = forms.MemberTransfer(data=form_data)
self.manager.save_results(
status=form.is_valid(),
test_name="Check member Transfer Form - Alan to Betty",
output=form.errors,
test_description="Test the form is valid for this transfer. Doesn't actually do the transfer.",
)
#############################
# Test the view - actually do the transfer
view_data = {
"transfer_to": betty.id,
"amount": amt,
"description": desc,
}
url = reverse("payments:member_transfer")
response = self.client.post(url, view_data)
self.manager.save_results(
status=response.status_code,
test_name="Execute member transfer - Alan to Betty",
test_description="Use the view to actually perform a transfer from Alan to Betty. Alan should have sufficient funds for this to go through.",
)
#############################
# Check the emails
check_email_sent(
manager=self.manager,
test_name="Check that Alan received an email for the transfer",
test_description="Doing a transfer should generate an email to Alan to confirm it has gone through.",
subject_search="Transfer to Betty Bunting",
email_to="Alan",
)
check_email_sent(
manager=self.manager,
test_name="Check that Betty received an email for the transfer",
test_description="Doing a transfer should generate an email to Betty to let her know.",
subject_search="Transfer from Alan",
body_search=f"has transferred {amt}",
)
#############################
# Check after
# Betty side
check_last_transaction_for_user(
manager=self.manager,
user=betty,
tran_desc=desc,
tran_amt=amt,
test_name="Execute member transfer - Alan to Betty. Betty transaction",
other_member=alan,
test_description="Check that Betty's latest transaction is the transfer from Alan.",
)
#############################
# Alan side
check_last_transaction_for_user(
manager=self.manager,
user=alan,
tran_desc=desc,
tran_amt=-amt,
test_name="Execute member transfer - Alan to Betty. Alan transaction",
other_member=betty,
test_description="Check that Alan's latest transaction is the transfer to Betty.",
)
#############################
# Check Alan's balance after
check_balance_for_user(
manager=self.manager,
user=alan,
expected_balance=alan_expected_initial_balance - amt,
test_name="Execute member transfer - Alan to Betty. Alan balance",
test_description="Check that Alan's balance after the transfer to Betty is correct.",
)
##############################
# Check Betty's balance after
check_balance_for_user(
manager=self.manager,
user=betty,
expected_balance=betty_expected_initial_balance + amt,
test_name="Execute member transfer - Alan to Betty. Betty balance",
test_description="Check that Betty's balance after the transfer from Alan is correct.",
)
[docs]
def a2_member_transfer_with_insufficient_funds(self):
"""Member transfer action which triggers manual top up"""
colin = self.manager.get_user(username="102")
fiona = self.manager.get_user(username="105")
self.manager.login_user(colin)
# Check Colin
colin_expected_initial_balance = 408.88
check_balance_for_user(
manager=self.manager,
user=colin,
expected_balance=colin_expected_initial_balance,
test_name="Check initial balance for Colin",
test_description="This is the initial check of Colin's balance before we process the transfer.",
)
# Check Fiona
fiona_expected_initial_balance = 305.26
check_balance_for_user(
manager=self.manager,
user=fiona,
expected_balance=fiona_expected_initial_balance,
test_name="Check initial balance for Fiona",
test_description="This is the initial check of Fiona's balance before we process the transfer.",
)
#################
# Generated Selenium Code
##################
# Get transfer url
transfer_url = self.manager.base_url + reverse("payments:member_transfer")
# Connect to page
self.manager.driver.get(transfer_url)
# Select Fiona from recent list
select = Select(self.manager.selenium_wait_for_clickable("id-cobalt-recent"))
select.select_by_value("12")
# Wait for refresh
self.manager.selenium_wait_for_clickable("id_amount").send_keys("500")
self.manager.selenium_wait_for_clickable("id_description").send_keys(
"Colin to Fiona 500"
)
self.manager.selenium_wait_for_clickable("cobalt-button").click()
# Wait for credit card entry screen (Stripe manual) to appear
self.manager.selenium_wait_for("id_credit_card_header")
stripe_manual_payment_screen(self.manager)
[docs]
def a3_member_auto_top_up_enable(self):
"""Enable auto top up"""
alan = self.manager.alan
fiona = self.manager.fiona
# Log Alan in
self.manager.login_user(alan)
# set it up
setup_auto_top_up(self.manager)
self.manager.save_results(
status=True,
test_name="Turn on auto top up for Alan",
output="Hardcoded success. Tests follow.",
test_description="This step sets up auto top up using Selenium. Pass here just means that it didn't crash, subsequent steps check if it was really successful.",
)
###############################
# IMPORTANT!!!!
#
# Users seem to now be cached so we need to reload Alan again
################################
alan = self.manager.get_user(username="100")
##############################
# Check auto top up
test = alan.stripe_auto_confirmed == "On"
self.manager.save_results(
status=test,
test_name="Check auto top up flag turned on for Alan",
output=f"Expected stripe_auto_confirmed='On'. Found: {alan.stripe_auto_confirmed}.",
test_description="Looks at user object to see if auto top up has been enabled.",
)
##############################
# Check auto top up amount
expected_amount = 100.0
test = alan.auto_amount == expected_amount
self.manager.save_results(
status=test,
test_name="Check auto top up amount for Alan",
output=f"Expected ${expected_amount}, got ${alan.auto_amount}",
test_description="Looks at user object to see that auto top up amount is set to expected value.",
)
# This one doesn't work through the form, use selenium
# Get url
url = self.manager.base_url + reverse("payments:member_transfer")
# Connect to page
self.manager.driver.get(url)
# select fiona to pay to
self.manager.selenium_wait_for_select_and_pick_an_option(
"id-cobalt-recent", "Fiona Freckle"
)
# values
amt = 1000.0
desc = "Trigger Auto"
# enter amount
self.manager.selenium_wait_for_clickable("id_amount").send_keys(amt)
# enter description
self.manager.selenium_wait_for_clickable("id_description").send_keys(desc)
# submit
self.manager.selenium_wait_for_clickable("cobalt-button").click()
self.manager.save_results(
status=True,
test_name="Manual transfer to trigger auto top up - Alan to fiona",
test_description="This transaction should trigger an auto top up event for Alan.",
)
#############################
# Give Stripe time to call us back
time.sleep(5)
# Check after
# fiona side
fiona_tran = (
MemberTransaction.objects.filter(member=fiona)
.order_by("-created_date")
.first()
)
if fiona_tran.description == desc and float(fiona_tran.amount) == amt:
test_result = True
else:
test_result = False
result = f"Expected {amt} and '{desc}'. Got {fiona_tran.amount} and '{fiona_tran.description}'"
self.manager.save_results(
status=test_result,
test_name="Member transfer triggering auto top up - Alan to fiona. fiona transaction",
output=result,
test_description="Check fiona's latest transaction is the transfer from Alan.",
)
############################
alan_balance = get_balance(alan)
test_result = alan_balance == 345.55
self.manager.save_results(
status=test_result,
test_name="Manual transfer triggering auto top up - Alan to fiona. Alan's balance",
output=f"Expected $345.55. Got ${alan_balance}",
test_description="Check Alan's balance is as expected",
)
