first commit

modules/production_work_timesheet/tests/__init__.py

@@ -0,0 +1,2 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.

modules/production_work_timesheet/tests/test_module.py

@@ -0,0 +1,109 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+import datetime
+
+from trytond.modules.company.tests import (
+    CompanyTestMixin, create_company, create_employee, set_company)
+from trytond.pool import Pool
+from trytond.tests.test_tryton import ModuleTestCase, with_transaction
+
+
+class ProductionWorkTimesheetTestCase(CompanyTestMixin, ModuleTestCase):
+    'Test Production Work Timesheet module'
+    module = 'production_work_timesheet'
+
+    def create_work(self, production_state='draft'):
+        pool = Pool()
+        Work = pool.get('production.work')
+        Operation = pool.get('production.routing.operation')
+        WorkCenter = pool.get('production.work.center')
+        Production = pool.get('production')
+
+        work_center = WorkCenter(name='Work Center')
+        work_center.save()
+
+        operation = Operation(name='Operation')
+        operation.timesheet_available = True
+        operation.save()
+
+        production = Production(state=production_state)
+        production.save()
+
+        work = Work(
+            production=production,
+            operation=operation,
+            work_center=work_center)
+        return work
+
+    @with_transaction()
+    def test_set_timesheet_work(self):
+        'Test _set_timesheet_work'
+        pool = Pool()
+        Work = pool.get('production.work')
+        TimesheetWork = pool.get('timesheet.work')
+        Date = pool.get('ir.date')
+
+        company = create_company()
+        with set_company(company):
+            # Test on create
+            work = self.create_work()
+            work.save()
+            self.assertEqual(len(work.timesheet_works), 1)
+
+            # Test on write
+            work.operation.timesheet_available = False
+            work.operation.save()
+            work.sequence = 1  # trigger write
+            work.save()
+            self.assertFalse(work.timesheet_works)
+
+            # Test delete
+            work = self.create_work()
+            work.save()
+            timesheet_work, = work.timesheet_works
+            timesheet_work_id = timesheet_work.id
+            Work.delete([work])
+            self.assertFalse(
+                TimesheetWork.search([('id', '=', timesheet_work_id)]))
+
+            # Test create as done
+            work = self.create_work(production_state='done')
+            work.save()
+            timesheet_work, = work.timesheet_works
+            self.assertEqual(timesheet_work.timesheet_end_date, Date.today())
+
+            # Set write as done
+            work = self.create_work()
+            work.save()
+            work.production.state = 'done'
+            work.production.save()
+            work.state = 'done'  # Trigger a write to update state
+            work.save()
+            timesheet_work, = work.timesheet_works
+            self.assertEqual(timesheet_work.timesheet_end_date, Date.today())
+
+    @with_transaction()
+    def test_timesheet_lines(self):
+        'Test timesheet_lines'
+        pool = Pool()
+        TimesheetLine = pool.get('timesheet.line')
+
+        company = create_company()
+        with set_company(company):
+            work = self.create_work()
+            work.save()
+
+            employee = create_employee(company)
+
+            timesheet_line = TimesheetLine(
+                employee=employee,
+                duration=datetime.timedelta(1))
+
+            work.timesheet_lines = [timesheet_line]
+            work.save()
+
+            self.assertEqual(len(work.timesheet_lines), 1)
+
+
+del ModuleTestCase