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="alice@example.org", 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="alice@example.org", 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")