3182d90bbf
Now that we no longer support py27, we can use the standard library unittest.mock module instead of the third party mock lib. Change-Id: Ib2843c62ff29b269139981f067ae6afcab624799 Signed-off-by: Sean McGinnis <sean.mcginnis@gmail.com>
66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
# Copyright 2018 StackHPC Ltd.
|
|
#
|
|
# 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 hashlib
|
|
from unittest import mock
|
|
|
|
import monasca_api.db.fingerprint as fingerprint
|
|
from monasca_api.tests import base
|
|
|
|
|
|
class TestFingerprint(base.BaseTestCase):
|
|
|
|
@mock.patch('monasca_api.db.fingerprint.Fingerprint._get_metadata')
|
|
@mock.patch('monasca_api.db.fingerprint.Fingerprint._get_schema_raw')
|
|
def test_get_schema_raw_pre_alembic(self, mock_schema_raw, mock_metadata):
|
|
mock_schema_raw.return_value = 'dummy_schema_raw'
|
|
|
|
tables = mock.PropertyMock = {
|
|
'dummy_table': 'dummy_columns'
|
|
}
|
|
mock_metadata.return_value.tables = tables
|
|
|
|
# No Alembic revision ID exists in the DB so we look it up from the
|
|
# table of fingerprints. Since we use a dummy schema, we insert a dummy
|
|
# entry into the lookup table.
|
|
fingerprint._REVS[
|
|
hashlib.sha1(b'dummy_schema_raw').hexdigest()] = 'dummy_revision'
|
|
|
|
f = fingerprint.Fingerprint('mock_engine')
|
|
self.assertEqual(f.schema_raw, 'dummy_schema_raw')
|
|
self.assertEqual(f.sha1, hashlib.sha1(b'dummy_schema_raw').hexdigest())
|
|
self.assertEqual(f.revision, 'dummy_revision')
|
|
|
|
@mock.patch('monasca_api.db.fingerprint.Fingerprint._get_db_session')
|
|
@mock.patch('monasca_api.db.fingerprint.Fingerprint._get_metadata')
|
|
@mock.patch('monasca_api.db.fingerprint.Fingerprint._get_schema_raw')
|
|
def test_get_schema_raw_post_alembic(
|
|
self, mock_schema_raw, mock_metadata, mock_db_session):
|
|
mock_schema_raw.return_value = 'dummy_schema_raw'
|
|
|
|
tables = mock.PropertyMock = {
|
|
'alembic_version': 'dummy_version',
|
|
'dummy_table': 'dummy_columns'
|
|
}
|
|
mock_metadata.return_value.tables = tables
|
|
|
|
# Alembic sets the version in the DB, so we look it up from there
|
|
mock_db_session.return_value.query.return_value.one.return_value = (
|
|
'dummy_revision',)
|
|
|
|
f = fingerprint.Fingerprint('mock_engine')
|
|
self.assertEqual(f.schema_raw, 'dummy_schema_raw')
|
|
self.assertEqual(f.sha1, hashlib.sha1(b'dummy_schema_raw').hexdigest())
|
|
self.assertEqual(f.revision, 'dummy_revision')
|