flectra/addons/membership/tests/common.py

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

import datetime
from dateutil.relativedelta import relativedelta

from flectra.addons.account.tests.account_test_classes import AccountingTestCase


class TestMembershipCommon(AccountingTestCase):

    def setUp(self):
        super(TestMembershipCommon, self).setUp()

        # Usefull models
        Partner = self.env['res.partner']

        # Test memberships
        self.membership_1 = self.env['product.product'].create({
            'membership': True,
            'membership_date_from': datetime.date.today() + relativedelta(days=-2),
            'membership_date_to': datetime.date.today() + relativedelta(months=1),
            'name': 'Basic Limited',
            'type': 'service',
            'list_price': 100.00,
        })

        # Test people
        self.partner_1 = Partner.create({
            'name': 'Ignasse Reblochon',
        })
        self.partner_2 = Partner.create({
            'name': 'Martine Poulichette',
            'free_member': True,
        })