# -*- coding: utf-8; -*-

import datetime
from unittest.mock import patch

from pyramid.httpexceptions import HTTPFound

from sideshow.testing import WebTestCase
from sideshow.web.views import products as mod
from sideshow.batch.neworder import NewOrderBatchHandler


class TestIncludeme(WebTestCase):

    def test_coverage(self):
        mod.includeme(self.pyramid_config)


class TestLocalProductView(WebTestCase):

    def make_view(self):
        return mod.LocalProductView(self.request)

    def test_configure_grid(self):
        model = self.app.model
        view = self.make_view()
        grid = view.make_grid(model_class=model.LocalProduct)
        self.assertNotIn('scancode', grid.linked_columns)
        self.assertNotIn('brand_name', grid.linked_columns)
        self.assertNotIn('description', grid.linked_columns)
        view.configure_grid(grid)
        self.assertIn('scancode', grid.linked_columns)
        self.assertIn('brand_name', grid.linked_columns)
        self.assertIn('description', grid.linked_columns)

    def test_configure_form(self):
        model = self.app.model
        view = self.make_view()

        # creating
        with patch.object(view, 'creating', new=True):
            form = view.make_form(model_class=model.LocalProduct)
            self.assertIn('external_id', form)
            view.configure_form(form)
            self.assertNotIn('external_id', form)

        user = model.User(username='barney')
        self.session.add(user)
        product = model.LocalProduct()
        self.session.add(product)
        self.session.commit()

        # viewing
        with patch.object(view, 'viewing', new=True):
            form = view.make_form(model_instance=product)
            self.assertNotIn('external_id', form.readonly_fields)
            self.assertNotIn('local_products.view.orders', form.grid_vue_context)
            view.configure_form(form)
            self.assertIn('external_id', form.readonly_fields)
            self.assertIn('local_products.view.orders', form.grid_vue_context)

    def test_make_orders_grid(self):
        model = self.app.model
        enum = self.app.enum
        view = self.make_view()

        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_id=42, created_by=user)
        product = model.LocalProduct()
        self.session.add(product)
        item = model.OrderItem(local_product=product,
                               order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                               status_code=enum.ORDER_ITEM_STATUS_INITIATED)
        order.items.append(item)
        self.session.add(order)
        self.session.commit()

        # no view perm
        grid = view.make_orders_grid(product)
        self.assertEqual(len(grid.actions), 0)

        # with view perm
        with patch.object(self.request, 'is_root', new=True):
            grid = view.make_orders_grid(product)
            self.assertEqual(len(grid.actions), 1)
            self.assertEqual(grid.actions[0].key, 'view')

    def test_make_new_order_batches_grid(self):
        model = self.app.model
        enum = self.app.enum
        handler = NewOrderBatchHandler(self.config)
        view = self.make_view()

        user = model.User(username='barney')
        self.session.add(user)
        batch = handler.make_batch(self.session, created_by=user)
        self.session.add(batch)
        product = model.LocalProduct()
        self.session.add(product)
        row = handler.make_row(local_product=product,
                               order_qty=1, order_uom=enum.ORDER_UOM_UNIT)
        handler.add_row(batch, row)
        self.session.commit()

        # no view perm
        grid = view.make_new_order_batches_grid(product)
        self.assertEqual(len(grid.actions), 0)

        # with view perm
        with patch.object(self.request, 'is_root', new=True):
            grid = view.make_new_order_batches_grid(product)
            self.assertEqual(len(grid.actions), 1)
            self.assertEqual(grid.actions[0].key, 'view')


class TestPendingProductView(WebTestCase):

    def make_view(self):
        return mod.PendingProductView(self.request)

    def test_configure_grid(self):
        model = self.app.model
        view = self.make_view()
        # nb. mostly just getting coverage here
        grid = view.make_grid(model_class=model.PendingProduct)
        self.assertNotIn('scancode', grid.linked_columns)
        self.assertNotIn('brand_name', grid.linked_columns)
        self.assertNotIn('description', grid.linked_columns)
        view.configure_grid(grid)
        self.assertIn('scancode', grid.linked_columns)
        self.assertIn('brand_name', grid.linked_columns)
        self.assertIn('description', grid.linked_columns)

    def test_configure_form(self):
        model = self.app.model
        enum = self.app.enum
        view = self.make_view()

        # creating
        with patch.object(view, 'creating', new=True):
            form = view.make_form(model_class=model.PendingProduct)
            view.configure_form(form)
            self.assertNotIn('status', form)
            self.assertNotIn('created', form)
            self.assertNotIn('created_by', form)

        user = model.User(username='barney')
        self.session.add(user)
        product = model.PendingProduct(status=enum.PendingProductStatus.PENDING,
                                         created_by=user)
        self.session.add(product)
        self.session.commit()

        # viewing
        with patch.object(view, 'viewing', new=True):
            form = view.make_form(model_instance=product)
            view.configure_form(form)
            self.assertIn('status', form)
            self.assertIn('created', form)
            self.assertIn('created_by', form)

    def test_make_orders_grid(self):
        model = self.app.model
        enum = self.app.enum
        view = self.make_view()

        user = model.User(username='barney')
        self.session.add(user)
        order = model.Order(order_id=42, customer_id=42, created_by=user)
        product = model.PendingProduct(status=enum.PendingProductStatus.PENDING,
                                         created_by=user)
        self.session.add(product)
        item = model.OrderItem(pending_product=product,
                               order_qty=1, order_uom=enum.ORDER_UOM_UNIT,
                               status_code=enum.ORDER_ITEM_STATUS_INITIATED)
        order.items.append(item)
        self.session.add(order)
        self.session.commit()

        # no view perm
        grid = view.make_orders_grid(product)
        self.assertEqual(len(grid.actions), 0)

        # with view perm
        with patch.object(self.request, 'is_root', new=True):
            grid = view.make_orders_grid(product)
            self.assertEqual(len(grid.actions), 1)
            self.assertEqual(grid.actions[0].key, 'view')

    def test_make_new_order_batches_grid(self):
        model = self.app.model
        enum = self.app.enum
        handler = NewOrderBatchHandler(self.config)
        view = self.make_view()

        user = model.User(username='barney')
        self.session.add(user)
        batch = handler.make_batch(self.session, created_by=user)
        self.session.add(batch)
        product = model.PendingProduct(status=enum.PendingProductStatus.PENDING,
                                         created_by=user)
        self.session.add(product)
        row = handler.make_row(pending_product=product,
                               order_qty=1, order_uom=enum.ORDER_UOM_UNIT)
        handler.add_row(batch, row)
        self.session.commit()

        # no view perm
        grid = view.make_new_order_batches_grid(product)
        self.assertEqual(len(grid.actions), 0)

        # with view perm
        with patch.object(self.request, 'is_root', new=True):
            grid = view.make_new_order_batches_grid(product)
            self.assertEqual(len(grid.actions), 1)
            self.assertEqual(grid.actions[0].key, 'view')

    def test_delete_instance(self):
        self.pyramid_config.add_route('pending_products.view', '/pending/products/{uuid}')
        model = self.app.model
        enum = self.app.enum
        handler = NewOrderBatchHandler(self.config)
        view = self.make_view()

        user = model.User(username='barney')
        self.session.add(user)

        # 1st product is standalone, will be deleted
        product = model.PendingProduct(status=enum.PendingProductStatus.PENDING,
                                       created_by=user)
        self.session.add(product)
        self.session.flush()
        self.assertEqual(self.session.query(model.PendingProduct).count(), 1)
        view.delete_instance(product)
        self.session.flush()
        self.assertEqual(self.session.query(model.PendingProduct).count(), 0)

        # 2nd product is attached to new order batch, will not be deleted
        batch = handler.make_batch(self.session, created_by=user)
        self.session.add(batch)
        product = model.PendingProduct(status=enum.PendingProductStatus.PENDING,
                                         created_by=user)
        self.session.add(product)
        row = handler.make_row(pending_product=product,
                               order_qty=1, order_uom=enum.ORDER_UOM_UNIT)
        handler.add_row(batch, row)
        self.session.flush()
        self.assertEqual(self.session.query(model.PendingProduct).count(), 1)
        self.assertRaises(HTTPFound, view.delete_instance, product)
        self.session.flush()
        self.assertEqual(self.session.query(model.PendingProduct).count(), 1)

        # but after batch is executed, 2nd product can be deleted
        batch.executed = datetime.datetime.now()
        batch.executed_by = user
        self.session.flush()
        self.assertEqual(self.session.query(model.PendingProduct).count(), 1)
        view.delete_instance(product)
        self.session.flush()
        self.assertEqual(self.session.query(model.PendingProduct).count(), 0)