[ADD]:Unit Test cases for Sale

bzr revid: nch@tinyerp.com-20090918100159-qffibrf0h1mg7l78

addons/sale/unit_test/__init__.py

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    OpenERP, Open Source Management Solution
+#    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
+#    $Id$
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+
+# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:

addons/sale/unit_test/test.py

@@ -0,0 +1,139 @@
+import unittest
+import pooler
+import netsvc
+from cStringIO import StringIO
+from osv import osv
+
+cr = None
+uid = None
+order_id = None
+
+class sale_order_test_case(unittest.TestCase):
+    def setUp(self):
+        try:
+            self.pool = pooler.get_pool(cr.dbname)
+            self.sale_order = self.pool.get('sale.order')
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+
+    def tearDown(self):
+        try:
+            self.pool = None
+            self.sale_order = None
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+
+    def test_1_Create(self):
+
+        try:
+            global order_id
+            model_obj = self.pool.get('ir.model.data')
+
+            ### SALE ORDER
+            shop = model_obj._get_id(cr, uid, 'sale', 'shop')
+            shop_id = model_obj.browse(cr, uid, shop).res_id
+            partner = model_obj._get_id(cr,uid, 'base', 'res_partner_9')
+            partner_id = model_obj.browse(cr, uid, partner,).res_id
+            partner_invoice = model_obj._get_id(cr, uid, 'base', 'res_partner_address_9')
+            partner_invoice_id = model_obj.browse(cr, uid, partner_invoice).res_id
+            pricelist_id = self.pool.get('res.partner').browse(cr, uid,partner_id).property_product_pricelist.id
+            order_id = self.sale_order.create(cr,uid,
+                            {'shop_id':shop_id,'pricelist_id':pricelist_id,'user_id':uid,
+                             'partner_id':partner_id,'partner_invoice_id':partner_invoice_id,
+                             'partner_shipping_id':partner_invoice_id,'partner_order_id':partner_invoice_id})
+            ### SALE ORDER LINE
+            product = model_obj._get_id(cr,uid, 'product', 'product_product_pc2')
+            product_id = model_obj.browse(cr, uid, product).res_id
+            product_uom = model_obj._get_id(cr, uid, 'product', 'product_uom_unit')
+            product_uom_id = model_obj.browse(cr, uid, product_uom).res_id
+            self.pool.get('sale.order.line').create(cr,uid,
+                            {'order_id':order_id,'name':'[PC2] Computer assembled on demand',
+                             'product_id':product_id,'product_uom':product_uom_id,'price_unit':600,
+                             'type':'make_to_order'})
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+    def test_2_ConfirmOrder(self):
+        try:
+            self.failUnless(order_id,"No Sale Order Created !")
+            wf_service = netsvc.LocalService("workflow")
+            res = wf_service.trg_validate(uid, 'sale.order',order_id, 'order_confirm', cr)
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+    def test_3_CreateInvoice(self):
+
+        try:
+            self.failUnless(order_id,"No Sale Order Created !")
+            wf_service = netsvc.LocalService("workflow")
+            res = wf_service.trg_validate(uid, 'sale.order',order_id, 'manual_invoice', cr)
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+    def test_4_CancelPicking(self):
+        try:
+            self.failUnless(order_id,"No Sale Order Created !")
+            picking_obj = self.pool.get('stock.picking')
+            pickings = picking_obj.search(cr,uid,[('sale_id','=',order_id)])
+            picking_obj.action_cancel(cr, uid, pickings)
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+    def test_5_PrintOrder(self):
+        try:
+            self.failUnless(order_id,"No Sale Order Created !")
+            report_service = netsvc.LocalService("report")
+            model_obj = self.pool.get('ir.model.data')
+            passwd = self.pool.get('res.users').browse(cr,uid,uid).password
+            report = model_obj._get_id(cr, uid, 'sale', 'report_sale_order')
+            report_id = model_obj.browse(cr, uid, report).res_id
+            report_service.report(cr.dbname, uid, passwd, 'sale.order', [order_id])
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+    def test_6_CancelOrder(self):
+        try:
+            self.failUnless(order_id,"No Sale Order Created !")
+            self.sale_order.action_cancel(cr, uid, [order_id])
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+    def test_7_Unlink(self):
+        try:
+            self.failUnless(order_id,"No Sale Order Created !")
+            self.sale_order.unlink(cr, uid, [order_id])
+        except osv.except_osv,e:
+            self.fail(e.name + e.value)
+        except Exception,e:
+            self.fail(e)
+
+
+def runTest(cursor=None, user=None):
+    global cr
+    global uid
+    cr = cursor
+    uid = user
+    out = StringIO()
+    suite = unittest.TestLoader().loadTestsFromTestCase(sale_order_test_case)
+    res = unittest.TextTestRunner(stream=out,verbosity=2).run(suite)
+    if res.wasSuccessful():
+        return (True,out.getvalue())
+    return (res,out.getvalue())