← Back to branch summary

~savoirfairelinux-openerp/openerp-travel/travel

« back to all changes in this revision

Viewing changes to travel_journey/tests/test_journey.py

  • Committer: Sandy Carter
  • Date: 2014-04-30 21:52:08 UTC
  • Revision ID: sandy.carter@savoirfairelinux.com-20140430215208-teiiem7t21wrfgve
[IMP] Increased travel_journey module test coverage to 70% (coverage)

Show diffs side-by-side

added added

removed removed

Lines of Context:
travel_journey/tests/test_journey.py
1
1
# -*- encoding: utf-8 -*-
2
 
################################################################################
 
2
###############################################################################
3
3
#
4
4
#    OpenERP, Open Source Management Solution
5
5
#    This module copyright (C) 2013 Savoir-faire Linux
18
18
#    You should have received a copy of the GNU Affero General Public License
19
19
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
20
#
21
 
################################################################################
 
21
###############################################################################
22
22
 
23
23
from openerp.tests.common import TransactionCase
24
 
from openerp.osv.osv import except_osv
 
24
from openerp.osv.orm import except_orm
 
25
from openerp.tools.misc import DEFAULT_SERVER_DATE_FORMAT
 
26
from openerp.tools.translate import _
25
27
 
26
28
 
27
29
class test_journey(TransactionCase):
60
62
            'destination': city_id,
61
63
            'departure': '2014-03-12',
62
64
            'passenger_id': self.passenger_id,
63
 
            'class_id': 1,
64
65
        }
65
66
 
66
67
    def test_create_journey(self):
84
85
            'cur_passenger_id': self.passenger_id,
85
86
        }, context=self.context)
86
87
        self.assertRaises(
87
 
            except_osv,
 
88
            except_orm,
88
89
            journey_import_model.data_import,
89
90
            self.cr, self.uid, [wizard_bad], context=self.context)
90
91
 
102
103
        }, context=self.context)
103
104
        journey_import_model.data_import(
104
105
            self.cr, self.uid, [wizard_good], context=self.context)
 
106
 
 
107
    def test_estimate_datetime(self):
 
108
        cr, uid, vals, context = self.cr, self.uid, self.vals, self.context
 
109
        journey_id = self.journey_model.create(
 
110
            cr, uid,
 
111
            dict(vals, departure=None, arrival='2014-05-01'),
 
112
            context=context)
 
113
        dt = self.journey_model._estimate_datetime(
 
114
            cr, uid, journey_id, 'date_start', context=context)
 
115
        date = self.journey_model._estimate_date(
 
116
            cr, uid, journey_id, 'date_start', context=context)
 
117
        time = self.journey_model._estimate_time(
 
118
            cr, uid, journey_id, 'date_start', context=context)
 
119
        self.assertEqual(dt[journey_id].strftime(DEFAULT_SERVER_DATE_FORMAT),
 
120
                         '2014-03-01')
 
121
        self.assertEqual(date[journey_id], '2014-03-01')
 
122
        self.assertEqual(time[journey_id], '00:00:00')
 
123
        journey_id = self.journey_model.create(cr, uid, vals, context=context)
 
124
        dt = self.journey_model._estimate_datetime(
 
125
            cr, uid, journey_id, 'date_stop', context=context)
 
126
        date = self.journey_model._estimate_date(
 
127
            cr, uid, journey_id, 'date_stop', context=context)
 
128
        time = self.journey_model._estimate_time(
 
129
            cr, uid, journey_id, 'date_stop', context=context)
 
130
        self.assertEqual(dt[journey_id].strftime(DEFAULT_SERVER_DATE_FORMAT),
 
131
                         '2014-03-12')
 
132
        self.assertEqual(date[journey_id], '2014-03-12')
 
133
        self.assertEqual(time[journey_id], '00:00:00')
 
134
 
 
135
    def test_on_change(self):
 
136
        cr, uid, vals, context = self.cr, self.uid, self.vals, self.context
 
137
        journey_id = self.journey_model.create(cr, uid, vals, context=context)
 
138
        new_city_id = self.city_model.create(self.cr, self.uid, {
 
139
            'city': 'new_test_city'
 
140
        }, context=self.context)
 
141
        res = self.journey_model.on_change_return(
 
142
            cr, uid, journey_id, key='return_destination',
 
143
            location=new_city_id, context=context)
 
144
        self.assertEqual(res['value']['return_destination'], new_city_id)
 
145
        res = self.journey_model.on_change_times(
 
146
            cr, uid, journey_id, departure='2014-03-01', arrival='2014-03-12',
 
147
            return_trip=False, context=context)
 
148
        self.assertEqual(res, {})
 
149
        res = self.journey_model.on_change_times(
 
150
            cr, uid, journey_id, departure='2014-03-12', arrival='2014-03-01',
 
151
            return_trip=False, context=context)
 
152
        self.assertIn('warning', res)
 
153
 
 
154
    def test_checks(self):
 
155
        cr, uid, vals, context = self.cr, self.uid, self.vals, self.context
 
156
        journey_id = self.journey_model.create(cr, uid, vals, context=context)
 
157
        self.assertTrue(self.journey_model.check_date_exists(
 
158
            cr, uid, journey_id, context=context))
 
159
        self.assertTrue(self.journey_model.check_date_exists_return(
 
160
            cr, uid, journey_id, context=context))
 
161
        self.assertTrue(self.journey_model.check_date(
 
162
            cr, uid, journey_id, context=context))
 
163
        self.assertTrue(self.journey_model.check_date_return(
 
164
            cr, uid, journey_id, context=context))
 
165
        self.assertTrue(self.journey_model.check_uom(
 
166
            cr, uid, journey_id, context=context))
 
167
 
 
168
    def test_company_get(self):
 
169
        cr, uid, vals, context = self.cr, self.uid, self.vals, self.context
 
170
        journey_id = self.journey_model.create(cr, uid, vals, context=context)
 
171
        self.assertEqual(self.journey_model.company_get(
 
172
            cr, uid, [journey_id], context=context), _("N/A"))
 
173
 
 
174
    def test_gets(self):
 
175
        cr, uid, vals, context = self.cr, self.uid, self.vals, self.context
 
176
        journey_id = self.journey_model.create(cr, uid, vals, context=context)
 
177
        self.assertEqual(self.journey_model.origin_get(
 
178
            cr, uid, journey_id, context=context).id, vals['origin'])
 
179
        self.assertEqual(self.journey_model.destination_get(
 
180
            cr, uid, journey_id, context=context).id, vals['destination'])
 
181
        self.assertEqual(self.journey_model.departure_date_get(
 
182
            cr, uid, journey_id, context=context), vals['departure'])
 
183
        self.assertEqual(self.journey_model.arrival_date_get(
 
184
            cr, uid, journey_id, context=context), '2014-03-12')
 
185
        self.assertEqual(self.journey_model.departure_time_get(
 
186
            cr, uid, journey_id, context=context), '00:00:00')
 
187
        self.assertEqual(self.journey_model.arrival_time_get(
 
188
            cr, uid, journey_id, context=context), '00:00:00')
 
189
 
 
190
    def test_inv_estimate_date(self):
 
191
        cr, uid, vals, context = self.cr, self.uid, self.vals, self.context
 
192
        journey_id = self.journey_model.create(cr, uid, vals, context=context)
 
193
        self.journey_model._inv_estimate_date(
 
194
            cr, uid, journey_id,
 
195
            field_name='date_start', val='2014-03-05',
 
196
            arg=None, context=context)
 
197
        self.journey_model._inv_estimate_date(
 
198
            cr, uid, journey_id,
 
199
            field_name='date_stop', val='2014-03-10',
 
200
            arg=None, context=context)
 
201
        journey_id = self.journey_model.create(
 
202
            cr, uid,
 
203
            dict(vals, departure=None, arrival='2014-05-01'),
 
204
            context=context)
 
205
        self.journey_model._inv_estimate_date(
 
206
            cr, uid, journey_id,
 
207
            field_name='date_start', val='2014-03-05',
 
208
            arg=None, context=context)
 
209
        self.journey_model._inv_estimate_date(
 
210
            cr, uid, journey_id,
 
211
            field_name='date_stop', val='2014-03-10',
 
212
            arg=None, context=context)