Skip to content

Commit 9d6c962

Browse files
author
Ryan Rossiter
committed
Fix up generate_order() unit tests
1 parent 08255fe commit 9d6c962

File tree

1 file changed

+18
-26
lines changed

1 file changed

+18
-26
lines changed

tests/managers/ordering_tests.py

Lines changed: 18 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -281,23 +281,6 @@ def test_get_price_id_list_item_not_found(self):
281281
self.assertEqual("Item ITEM2 does not exist for package PACKAGE_KEYNAME",
282282
str(exc))
283283

284-
def test_generate_order(self):
285-
ord_mock = self.set_mock('SoftLayer_Product_Order', 'verifyOrder')
286-
ord_mock.return_value = {'id': 1234}
287-
pkg = 'PACKAGE_KEYNAME'
288-
complex_type = 'SoftLayer_Container_Foo'
289-
items = ['ITEM1', 'ITEM2']
290-
291-
mock_pkg, mock_preset, mock_get_ids = self._patch_for_generate()
292-
293-
order = self.ordering.verify_order(pkg, 'DALLAS13', items,
294-
complex_type=complex_type)
295-
296-
mock_pkg.assert_called_once_with(pkg, mask='id')
297-
mock_preset.assert_not_called()
298-
mock_get_ids.assert_called_once_with(pkg, items)
299-
self.assertEqual(ord_mock.return_value, order)
300-
301284
def test_generate_order_package_not_found(self):
302285
self._assert_package_error(self.ordering.generate_order,
303286
'PACKAGE_KEYNAME', 'DALLAS13',
@@ -318,31 +301,40 @@ def test_generate_order_with_preset(self):
318301
complex_type = 'SoftLayer_Container_Foo'
319302
items = ['ITEM1', 'ITEM2']
320303
preset = 'PRESET_KEYNAME'
321-
expected_order = {}
304+
expected_order = {'complexType': 'SoftLayer_Container_Foo',
305+
'location': 'DALLAS13',
306+
'packageId': 1234,
307+
'presetId': 5678,
308+
'prices': [{'id': 1111}, {'id': 2222}],
309+
'quantity': 1,
310+
'useHourlyPricing': True}
322311

323312
mock_pkg, mock_preset, mock_get_ids = self._patch_for_generate()
324313

325-
order = self.ordering.verify_order(pkg, 'DALLAS13', items,
326-
preset_keyname=preset,
327-
complex_type=complex_type)
314+
order = self.ordering.generate_order(pkg, 'DALLAS13', items,
315+
preset_keyname=preset,
316+
complex_type=complex_type)
328317

329318
mock_pkg.assert_called_once_with(pkg, mask='id')
330319
mock_preset.assert_called_once_with(pkg, preset)
331320
mock_get_ids.assert_called_once_with(pkg, items)
332321
self.assertEqual(expected_order, order)
333322

334323
def test_generate_order(self):
335-
ord_mock = self.set_mock('SoftLayer_Product_Order', 'placeOrder')
336-
ord_mock.return_value = {'id': 1234}
337324
pkg = 'PACKAGE_KEYNAME'
338325
items = ['ITEM1', 'ITEM2']
339326
complex_type = 'My_Type'
340-
expected_order = {}
327+
expected_order = {'complexType': 'My_Type',
328+
'location': 'DALLAS13',
329+
'packageId': 1234,
330+
'prices': [{'id': 1111}, {'id': 2222}],
331+
'quantity': 1,
332+
'useHourlyPricing': True}
341333

342334
mock_pkg, mock_preset, mock_get_ids = self._patch_for_generate()
343335

344-
order = self.ordering.verify_order(pkg, 'DALLAS13', items,
345-
complex_type=complex_type)
336+
order = self.ordering.generate_order(pkg, 'DALLAS13', items,
337+
complex_type=complex_type)
346338

347339
mock_pkg.assert_called_once_with(pkg, mask='id')
348340
mock_preset.assert_not_called()

0 commit comments

Comments
 (0)