def test_get_lifecycle_state_details_can_handle_none_result(self):
mock_session = mock.MagicMock
mock_session.query = mock.MagicMock
mock_session.query.filter = mock.MagicMock
mock_session.query.group_by = mock.MagicMock
mock_session.query.all = mock.MagicMock
mock_session.query.all.return_value = [None]
inventory_index = InventoryIndex()
details = inventory_index.get_lifecycle_state_details(mock_session,
'abc')
self.assertEqual({}, details)
def populate_data(self):
self.engine = create_test_engine()
initialize(self.engine)
self.scoped_sessionmaker = db.create_scoped_sessionmaker(self.engine)
with self.scoped_sessionmaker() as session:
inventory_indices = [
InventoryIndex(id='one_day_old',
created_at_datetime=datetime(
2010, 12, 30, 8, 0, 0)),
InventoryIndex(id='seven_days_old',
created_at_datetime=datetime(
2010, 12, 24, 8, 0, 0)),
InventoryIndex(id='nine_days_old',
created_at_datetime=datetime(
2010, 12, 22, 8, 0, 0))
]
for i in inventory_indices:
session.add(i)
session.commit()
session.expunge_all()
inventory_resources = [
Inventory(id=1, full_name=1, inventory_index_id='one_day_old'),
Inventory(id=2, full_name=2, inventory_index_id='one_day_old'),
Inventory(id=3,
full_name=3,
inventory_index_id='seven_days_old'),
Inventory(id=4,
full_name=4,
inventory_index_id='seven_days_old'),
Inventory(id=5,
full_name=5,
inventory_index_id='nine_days_old'),
Inventory(id=6,
full_name=6,
inventory_index_id='nine_days_old'),
]
for i in inventory_resources:
session.add(i)
session.commit()
session.expunge_all()
return session