102 lines
3.7 KiB
Python
102 lines
3.7 KiB
Python
|
# -*- coding: utf-8; -*-
|
||
|
|
|
||
|
|
import datetime
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from wuttaweb.forms.schema import WuttaMoney
|
||
|
|
|
||
|
|
from sideshow.testing import WebTestCase
|
||
|
|
from sideshow.web.views.batch import neworder as mod
|
||
|
|
from sideshow.web.forms.schema import PendingCustomerRef
|
||
|
|
from sideshow.batch.neworder import NewOrderBatchHandler
|
||
|
|
|
||
|
|
|
||
|
|
class TestIncludeme(WebTestCase):
|
||
|
|
|
||
|
|
def test_coverage(self):
|
||
|
|
mod.includeme(self.pyramid_config)
|
||
|
|
|
||
|
|
|
||
|
|
class TestNewOrderBatchView(WebTestCase):
|
||
|
|
|
||
|
|
def make_view(self):
|
||
|
|
return mod.NewOrderBatchView(self.request)
|
||
|
|
|
||
|
|
def test_get_batch_handler(self):
|
||
|
|
view = self.make_view()
|
||
|
|
handler = view.get_batch_handler()
|
||
|
|
self.assertIsInstance(handler, NewOrderBatchHandler)
|
||
|
|
|
||
|
|
def test_configure_grid(self):
|
||
|
|
model = self.app.model
|
||
|
|
view = self.make_view()
|
||
|
|
grid = view.make_grid(model_class=model.NewOrderBatch)
|
||
|
|
self.assertNotIn('total_price', grid.renderers)
|
||
|
|
view.configure_grid(grid)
|
||
|
|
self.assertIn('total_price', grid.renderers)
|
||
|
|
|
||
|
|
def test_configure_form(self):
|
||
|
|
model = self.app.model
|
||
|
|
enum = self.app.enum
|
||
|
|
view = self.make_view()
|
||
|
|
handler = view.batch_handler
|
||
|
|
|
||
|
|
user = model.User(username='barney')
|
||
|
|
self.session.add(user)
|
||
|
|
customer = model.PendingCustomer(status=enum.PendingCustomerStatus.PENDING,
|
||
|
|
created_by=user)
|
||
|
|
self.session.add(customer)
|
||
|
|
batch = handler.make_batch(self.session, pending_customer=customer, created_by=user)
|
||
|
|
self.session.add(batch)
|
||
|
|
self.session.commit()
|
||
|
|
|
||
|
|
# viewing
|
||
|
|
with patch.object(view, 'viewing', new=True):
|
||
|
|
form = view.make_form(model_instance=batch)
|
||
|
|
view.configure_form(form)
|
||
|
|
schema = form.get_schema()
|
||
|
|
self.assertIsInstance(schema['pending_customer'].typ, PendingCustomerRef)
|
||
|
|
self.assertIsInstance(schema['total_price'].typ, WuttaMoney)
|
||
|
|
|
||
|
|
def test_configure_row_grid(self):
|
||
|
|
model = self.app.model
|
||
|
|
view = self.make_view()
|
||
|
|
grid = view.make_grid(model_class=model.NewOrderBatchRow)
|
||
|
|
self.assertNotIn('total_price', grid.renderers)
|
||
|
|
view.configure_row_grid(grid)
|
||
|
|
self.assertIn('total_price', grid.renderers)
|
||
|
|
|
||
|
|
def test_get_xref_buttons(self):
|
||
|
|
self.pyramid_config.add_route('orders.view', '/orders/{uuid}')
|
||
|
|
model = self.app.model
|
||
|
|
enum = self.app.enum
|
||
|
|
view = self.make_view()
|
||
|
|
handler = view.batch_handler
|
||
|
|
|
||
|
|
user = model.User(username='barney')
|
||
|
|
self.session.add(user)
|
||
|
|
customer = model.PendingCustomer(status=enum.PendingCustomerStatus.PENDING,
|
||
|
|
created_by=user)
|
||
|
|
self.session.add(customer)
|
||
|
|
|
||
|
|
# 1st batch has no order
|
||
|
|
batch = handler.make_batch(self.session, pending_customer=customer, created_by=user)
|
||
|
|
self.session.add(batch)
|
||
|
|
self.session.flush()
|
||
|
|
buttons = view.get_xref_buttons(batch)
|
||
|
|
self.assertEqual(len(buttons), 0)
|
||
|
|
|
||
|
|
# 2nd batch is executed; has order
|
||
|
|
batch = handler.make_batch(self.session, pending_customer=customer, created_by=user,
|
||
|
|
executed=datetime.datetime.now(), executed_by=user)
|
||
|
|
self.session.add(batch)
|
||
|
|
self.session.flush()
|
||
|
|
order = model.Order(order_id=batch.id, created_by=user)
|
||
|
|
self.session.add(order)
|
||
|
|
self.session.flush()
|
||
|
|
with patch.object(view, 'Session', return_value=self.session):
|
||
|
|
# nb. this also requires perm
|
||
|
|
with patch.object(self.request, 'is_root', new=True):
|
||
|
|
buttons = view.get_xref_buttons(batch)
|
||
|
|
self.assertEqual(len(buttons), 1)
|