6f4f7b5287
This commit adds a new set of tests to assert that our sqlalchemy ORM models match the state of the db post migration. Recently we've seen how far the 2 have drifted apart over time and even after several patches to correct it they still don't match. This commit corrects the existing issues as well as adds tests to assert this is correct moving forward. The model sync tests with sqlite are currently skipped because the migration sets the id columns to an integer type on sqlite but the models expect biginteger. This is due to biginteger not existing on sqlite, so functionally there is no difference but the tests would fail because of the type mismatch. Change-Id: I1c25f331fe1143328fef9a7d46fdcec350a037c8
166 lines
6.4 KiB
Python
166 lines
6.4 KiB
Python
# Copyright 2014 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 datetime
|
|
import uuid
|
|
|
|
from oslo_db.sqlalchemy import models # noqa
|
|
import six
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.ext import declarative
|
|
|
|
BASE = declarative.declarative_base()
|
|
|
|
|
|
class SubunitBase(models.ModelBase):
|
|
"""Base class for Subunit Models."""
|
|
__table_args__ = {'mysql_engine': 'InnoDB'}
|
|
__table_initialized__ = False
|
|
|
|
def save(self, session=None):
|
|
from subunit2sql.db import api as db_api
|
|
super(SubunitBase, self).save(session or db_api.get_session())
|
|
|
|
def keys(self):
|
|
return list(self.__dict__.keys())
|
|
|
|
def values(self):
|
|
return self.__dict__.values()
|
|
|
|
def items(self):
|
|
return self.__dict__.items()
|
|
|
|
def to_dict(self):
|
|
d = self.__dict__.copy()
|
|
d.pop("_sa_instance_state")
|
|
return d
|
|
|
|
|
|
class Test(BASE, SubunitBase):
|
|
__tablename__ = 'tests'
|
|
__table_args__ = (sa.Index('ix_test_ids', 'id', 'test_id',
|
|
mysql_length={'test_id': 30}),
|
|
sa.Index('ix_tests_test_id', 'test_id',
|
|
mysql_length=30))
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
test_id = sa.Column(sa.String(256),
|
|
nullable=False)
|
|
run_count = sa.Column(sa.Integer())
|
|
success = sa.Column(sa.Integer())
|
|
failure = sa.Column(sa.Integer())
|
|
run_time = sa.Column(sa.Float())
|
|
|
|
|
|
class Run(BASE, SubunitBase):
|
|
__tablename__ = 'runs'
|
|
__table_args__ = (sa.Index('ix_run_at', 'run_at'),
|
|
sa.Index('ix_run_uuid', 'uuid'))
|
|
uuid = sa.Column(sa.String(36),
|
|
default=lambda: six.text_type(uuid.uuid4()))
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
skips = sa.Column(sa.Integer())
|
|
fails = sa.Column(sa.Integer())
|
|
passes = sa.Column(sa.Integer())
|
|
run_time = sa.Column(sa.Float())
|
|
artifacts = sa.Column(sa.Text())
|
|
run_at = sa.Column(sa.DateTime,
|
|
default=datetime.datetime.utcnow)
|
|
|
|
|
|
class TestRun(BASE, SubunitBase):
|
|
__tablename__ = 'test_runs'
|
|
__table_args__ = (sa.Index('ix_test_id_status', 'test_id', 'status'),
|
|
sa.Index('ix_test_id_start_time', 'test_id',
|
|
'start_time'),
|
|
sa.Index('ix_test_runs_test_id', 'test_id'),
|
|
sa.Index('ix_test_runs_run_id', 'run_id'),
|
|
sa.Index('ix_test_runs_start_time', 'start_time'),
|
|
sa.Index('ix_test_runs_stop_time', 'stop_time'),
|
|
sa.UniqueConstraint('test_id', 'run_id',
|
|
name='uq_test_runs'))
|
|
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
test_id = sa.Column(sa.BigInteger)
|
|
run_id = sa.Column(sa.BigInteger)
|
|
status = sa.Column(sa.String(256))
|
|
start_time = sa.Column(sa.DateTime())
|
|
start_time_microsecond = sa.Column(sa.Integer(), default=0)
|
|
stop_time = sa.Column(sa.DateTime())
|
|
stop_time_microsecond = sa.Column(sa.Integer(), default=0)
|
|
test = sa.orm.relationship(Test, backref=sa.orm.backref('test_run_test'),
|
|
foreign_keys=test_id,
|
|
primaryjoin=test_id == Test.id)
|
|
run = sa.orm.relationship(Run, backref=sa.orm.backref('test_run_run'),
|
|
foreign_keys=run_id,
|
|
primaryjoin=run_id == Run.id)
|
|
|
|
|
|
class RunMetadata(BASE, SubunitBase):
|
|
__tablename__ = 'run_metadata'
|
|
__table_args__ = (sa.Index('ix_run_key_value', 'key', 'value'),
|
|
sa.Index('ix_run_id', 'run_id'),
|
|
sa.UniqueConstraint('run_id', 'key', 'value',
|
|
name='uq_run_metadata'))
|
|
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
key = sa.Column(sa.String(255))
|
|
value = sa.Column(sa.String(255))
|
|
run_id = sa.Column(sa.BigInteger)
|
|
run = sa.orm.relationship(Run, backref='run', foreign_keys=run_id,
|
|
primaryjoin=run_id == Run.id)
|
|
|
|
|
|
class TestRunMetadata(BASE, SubunitBase):
|
|
__tablename__ = 'test_run_metadata'
|
|
__table_args__ = (sa.Index('ix_test_run_key_value', 'key', 'value'),
|
|
sa.Index('ix_test_run_id', 'test_run_id'),
|
|
sa.UniqueConstraint('test_run_id', 'key', 'value',
|
|
name='uq_test_run_metadata'))
|
|
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
key = sa.Column(sa.String(255))
|
|
value = sa.Column(sa.String(255))
|
|
test_run_id = sa.Column(sa.BigInteger)
|
|
test_run = sa.orm.relationship(TestRun,
|
|
backref=sa.orm.backref('test_run_meta'),
|
|
foreign_keys=test_run_id,
|
|
primaryjoin=test_run_id == TestRun.id)
|
|
|
|
|
|
class TestMetadata(BASE, SubunitBase):
|
|
__tablename__ = 'test_metadata'
|
|
__table_args__ = (sa.Index('ix_test_key_value', 'key', 'value'),
|
|
sa.Index('ix_test_id', 'test_id'),
|
|
sa.UniqueConstraint('test_id', 'key', 'value',
|
|
name='uq_test_metadata'))
|
|
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
key = sa.Column(sa.String(255))
|
|
value = sa.Column(sa.String(255))
|
|
test_id = sa.Column(sa.BigInteger)
|
|
test = sa.orm.relationship(Test, backref='test', foreign_keys=test_id,
|
|
primaryjoin=test_id == Test.id)
|
|
|
|
|
|
class Attachments(BASE, SubunitBase):
|
|
__tablename__ = 'attachments'
|
|
__table_args__ = (sa.Index('ix_attach_test_run_id', 'test_run_id'),)
|
|
id = sa.Column(sa.BigInteger, primary_key=True)
|
|
test_run_id = sa.Column(sa.BigInteger)
|
|
label = sa.Column(sa.String(255))
|
|
attachment = sa.Column(sa.LargeBinary())
|
|
test_run = sa.orm.relationship(TestRun, backref='test_run_attach',
|
|
foreign_keys=test_run_id,
|
|
primaryjoin=test_run_id == TestRun.id)
|