ccvpn/payments/tests/online/stripe.py

from django.contrib.staticfiles.testing import StaticLiveServerTestCase
from selenium.webdriver.firefox.webdriver import WebDriver

from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support.ui import Select
from selenium.webdriver.support.expected_conditions import presence_of_element_located
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

from django.conf import settings
from django.utils import timezone
from datetime import timedelta
from lambdainst.models import User
from payments.models import Payment, Subscription


class BaseOnlineTest(StaticLiveServerTestCase):
    # using a fixed port because you're supposed to forward a public ip to it
    # for the web hooks
    # $ ssh relaybox -R 12345:127.0.0.1:8000
    # (set ROOT_URL to the right public url and RUN_ONLINE_TESTS to True)
    port = 8000

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.selenium = WebDriver(firefox_binary="/usr/bin/firefox")
        cls.selenium.implicitly_wait(6)

        cls.root_url = settings.ROOT_URL
        cls.wait = WebDriverWait(cls.selenium, 6)

    @classmethod
    def tearDownClass(cls):
        cls.selenium.quit()
        super().tearDownClass()

    def _signup(self):
        self.selenium.get('%s%s' % (self.root_url, '/account/signup'))
        username_input = self.selenium.find_element_by_name("username")
        username_input.send_keys('test-user')
        password_input = self.selenium.find_element_by_name("password")
        password_input.send_keys('test-password')
        password_input = self.selenium.find_element_by_name("password2")
        password_input.send_keys('test-password')
        self.selenium.find_element_by_xpath('//input[@value="Sign up"]').click()


class OnlineStripeTests(BaseOnlineTest):
    fixtures = []

    def test_payment(self):
        self._signup()

        self.selenium.find_element_by_xpath('//label[@for="tab_onetime"]').click()
        self.wait.until(EC.visibility_of(self.selenium.find_element_by_xpath('//label[@for="tab_onetime"]/..//select[@name="method"]')))
        self.selenium.find_element_by_xpath('//label[@for="tab_onetime"]/..//select[@name="time"]/option[@value="3"]').click()
        self.selenium.find_element_by_xpath('//label[@for="tab_onetime"]/..//select[@name="method"]/option[@value="stripe"]').click()
        self.selenium.find_element_by_xpath('//label[@for="tab_onetime"]/..//input[@value="Buy Now"]').click()

        assert self.selenium.find_element_by_xpath('//span[text()="Test Mode"]')
        
        self.selenium.find_element_by_xpath('//input[@name="email"]').send_keys("test@ccrypto.org")
        self.selenium.find_element_by_xpath('//input[@name="cardNumber"]').send_keys("4242424242424242")
        self.selenium.find_element_by_xpath('//input[@name="cardExpiry"]').send_keys("6/66")
        self.selenium.find_element_by_xpath('//input[@name="cardCvc"]').send_keys("420")
        self.selenium.find_element_by_xpath('//input[@name="billingName"]').send_keys("Test User")
        self.selenium.find_element_by_xpath('//button[@type="submit"]').click()

        self.selenium.find_element_by_xpath('//h2[contains(text(),"Confirmed")]')

        user = User.objects.get(username="test-user")
        assert user.vpnuser.is_paid
        assert user.vpnuser.expiration >= (timezone.now() + timedelta(days=89))

    def test_subscription(self):
        self._signup()

        self.selenium.find_element_by_xpath('//label[@for="tab_subscr"]').click()
        self.wait.until(EC.visibility_of(self.selenium.find_element_by_xpath('//label[@for="tab_subscr"]/..//select[@name="method"]')))
        self.selenium.find_element_by_xpath('//label[@for="tab_subscr"]/..//select[@name="time"]/option[@value="12"]').click()
        self.selenium.find_element_by_xpath('//label[@for="tab_subscr"]/..//select[@name="method"]/option[@value="stripe"]').click()
        self.selenium.find_element_by_xpath('//label[@for="tab_subscr"]/..//input[@value="Subscribe"]').click()

        assert self.selenium.find_element_by_xpath('//span[text()="Test Mode"]')
        
        self.selenium.find_element_by_xpath('//input[@name="email"]').send_keys("test@ccrypto.org")
        self.selenium.find_element_by_xpath('//input[@name="cardNumber"]').send_keys("4242424242424242")
        self.selenium.find_element_by_xpath('//input[@name="cardExpiry"]').send_keys("6/66")
        self.selenium.find_element_by_xpath('//input[@name="cardCvc"]').send_keys("420")
        self.selenium.find_element_by_xpath('//input[@name="billingName"]').send_keys("Test User")
        self.selenium.find_element_by_xpath('//button[@type="submit"]').click()

        sub_status = self.selenium.find_element_by_xpath('//td[text()="Subscription"]//following-sibling::td')
        assert sub_status.text.startswith('ACTIVE')

        user = User.objects.get(username="test-user")
        assert user.vpnuser.is_paid
        assert user.vpnuser.expiration >= (timezone.now() + timedelta(days=359))

        sub_status.find_element_by_xpath('a[text()="cancel"]').click()
        self.selenium.find_element_by_xpath('//input[@value="Cancel Subscription"]').click()

        sub_status = self.selenium.find_element_by_xpath('//td[text()="Subscription"]//following-sibling::td')
        assert not sub_status.text.startswith('ACTIVE')

        user = User.objects.get(username="test-user")
        assert user.vpnuser.is_paid
        assert user.vpnuser.expiration >= (timezone.now() + timedelta(days=359))
        assert not user.vpnuser.get_subscription()