You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.1 KiB
34 lines
1.1 KiB
from mock import Mock, patch
|
|
|
|
from hc.payments.models import Subscription
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
class BillingTestCase(BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(BillingTestCase, self).setUp()
|
|
self.sub = Subscription(user=self.alice)
|
|
self.sub.subscription_id = "test-id"
|
|
self.sub.customer_id = "test-customer-id"
|
|
self.sub.save()
|
|
|
|
@patch("hc.payments.views.braintree")
|
|
def test_it_works(self, mock_braintree):
|
|
|
|
m1 = Mock(id="abc123", amount=123)
|
|
m2 = Mock(id="def456", amount=456)
|
|
mock_braintree.Transaction.search.return_value = [m1, m2]
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
r = self.client.get("/billing/")
|
|
self.assertContains(r, "123")
|
|
self.assertContains(r, "def456")
|
|
|
|
def test_it_saves_company_details(self):
|
|
self.client.login(username="[email protected]", password="password")
|
|
r = self.client.post("/billing/", {"bill_to": "foo\nbar"})
|
|
|
|
self.assertEqual(r.status_code, 302)
|
|
self.profile.refresh_from_db()
|
|
self.assertEqual(self.profile.bill_to, "foo\nbar")
|