from openerp.tests.common import TransactionCase

class ComputeParentLevel(TransactionCase):
    def test_compute_parent_level0(self):
        workcenter = self.env['mrp.workcenter'].create({
            'name': 'New Assembly',
            'calendar_id': self.env.ref('resource.timesheet_group1').id,
            'capacity_per_cycle': 5,
            'time_cycle': 1,
            'time_start': 0.1,
            'time_stop': 0.1, 
            'time_efficiency': 0.87,
            'product_id': self.env.ref('product.product_assembly').id,
            'costs_hour': 0,
            'costs_hour_account_id': self.env.ref('mrp.account_assembly_hours').id,
            'costs_cycle': 0.05,
            'costs_cycle_account_id': self.env.ref('mrp.account_assembly_cycle').id,
            })
        workcenter_child = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_A')
        workcenter_child.write({'parent_id': workcenter.id})
        self.assertEqual(workcenter.parent_level_1_id, workcenter)
        
    def test_compute_parent_level1(self):
        workcenter_child = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_E')
        workcenter_parent = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_top')
        workcenter_child.write({'parent_id': workcenter_parent.id})
        self.assertEqual(workcenter_child.parent_level_1_id, workcenter_parent)

    def test_compute_parent_level2(self):
        workcenter_child = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_A')
        workcenter_parent1 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_top')
        workcenter_parent2 = self.env.ref('mrp.mrp_workcenter_0')
        
        workcenter_child.write({'parent_id': workcenter_parent2.id})
        import pdb; pdb.set_trace()
        self.assertEqual(workcenter_child.parent_level_2_id, workcenter_parent2)
        self.assertEqual(workcenter_child.parent_level_1_id, workcenter_parent1)

    def test_compute_parent_level3(self):
        workcenter_child = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_B')
        workcenter_parent1 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_top')
        workcenter_parent2 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_2')
        workcenter_parent3 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_A')
        workcenter_child.write({'parent_id': workcenter_parent3.id})
        self.assertEqual(workcenter_child.parent_level_3_id, workcenter_parent3)
        self.assertEqual(workcenter_child.parent_level_2_id, workcenter_parent2)
        self.assertEqual(workcenter_child.parent_level_1_id, workcenter_parent1)
        
    def test_compute_parent_level3(self):
        workcenter_child = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_C')
        workcenter_parent1 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_top')
        workcenter_parent2 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_2')
        workcenter_parent3 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_A')
        workcenter_parent4 = self.env.ref('mrp_workcenter_hierarchical.mrp_workcenter_B')
        workcenter_parent4.write({'parent_id': workcenter_parent3.id})
        workcenter_child.write({'parent_id': workcenter_parent4.id})
        self.assertEqual(workcenter_child.parent_id, workcenter_parent4)