neutron/neutron/tests/unit/db/test_segments_db.py
Ann Kamyshnikova 12b0e16692 Change passing session to context in segments db functions
We have a number of functions that expect to get session as one
of the arguments. Passing session is not correct and this prevents
using new enginefacade which expects context to be passed and
session will be injected in current context.

NeutronLibImpact

Partially-Implements blueprint: enginefacade-switch

Change-Id: Ie1c54138ceaf9ac6f0674ad2786d4aaea9c80f73
2016-12-23 16:47:41 +04:00

30 lines
1011 B
Python

# Copyright (c) 2016 IBM Corp.
#
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import mock
from neutron.db import segments_db
from neutron.tests import base
class TestSegmentsDb(base.BaseTestCase):
def test_get_networks_segments_with_empty_networks(self):
context = mock.MagicMock()
net_segs = segments_db.get_networks_segments(context, [])
self.assertFalse(context.session.query.called)
self.assertEqual({}, net_segs)