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)
|