def test_backref(self):
column_obj = ColumnObject(
get_grt_column('test',
'tables',
'INTEGER',
comment="remote_side=remote_test;use_alter=True"))
column_ref = get_grt_column('ref', 'table_refs', 'INTEGER')
self.assertIsNone(column_obj.getBackref())
column_obj.setForeignKey(
get_grt_foreignKey('fk_test', referencedColumns=[column_ref]))
self.assertEquals(
' tableRef = relationship("TableRef", foreign_keys=[test], backref="tables", remote_side=[remote_test])',
column_obj.getBackref())
self.assertEquals(
' test = Column(INTEGER, ForeignKey("table_refs.ref", name="fk_test", use_alter=True, onupdate="SET NULL"))',
str(column_obj))
column_obj.foreign_key.deleteRule = 'CASCADE'
self.assertEquals(
' test = Column(\n'
' INTEGER, ForeignKey("table_refs.ref", name="fk_test", use_alter=True, ondelete="CASCADE", onupdate="SET NULL")\n'
' )', str(column_obj))
def test_backref_uselist(self):
column_obj = ColumnObject(
get_grt_column('test',
'tables',
'INTEGER',
comment="backrefuselist=False"))
column_ref = get_grt_column('ref', 'table_refs', 'INTEGER')
column_obj.setForeignKey(
get_grt_foreignKey('fk_test', referencedColumns=[column_ref]))
self.assertEquals(
' tableRef = relationship("TableRef", foreign_keys=[test], backref=backref("tables", uselist=False))',
column_obj.getBackref())
def test_abstract(self):
id_col = get_grt_column('id',
'table_test',
'INT(16)',
isNotNull=1,
autoIncrement=1)
table = get_grt_table(
'table_test',
columns=[id_col],
indices=[get_grt_index('i_test', columns=[id_col])],
comment="abstract=True")
self.assertEquals(
'class TableTest(object):\n'
'\n'
' __table_args__ = (\n'
' {\'mysql_charset\': \'utf8\', \'sqlite_autoincrement\': True}\n'
' )\n'
'\n'
' id = Column(INTEGER, nullable=False, autoincrement=True, primary_key=True) # pylint: disable=invalid-name\n'
'\n'
' def __repr__(self):\n'
' return self.__str__()\n'
'\n'
' def __str__(self):\n'
' return "<TableTest(%(id)s)>" % self.__dict__',
str(TableObject(table)))
def test_backref_ignore(self):
column_obj = ColumnObject(
get_grt_column('test',
'tables',
'INTEGER',
comment="relation=False"))
column_ref = get_grt_column('ref', 'table_refs', 'INTEGER')
self.assertIsNone(column_obj.getBackref())
column_obj.setForeignKey(
get_grt_foreignKey('fk_test', referencedColumns=[column_ref]))
self.assertEquals(
' # relation for test.ForeignKey ignored as configured in column comment',
column_obj.getBackref())
def test_basic(self):
column_obj = ColumnObject(
get_grt_column('test_column',
'test_table',
'VARCHAR(45)',
comment='alias=test'))
self.assertEquals(' test = Column("test_column", VARCHAR(45))',
str(column_obj))
def test_name_switch_primary_key(self):
column = get_grt_column('id_something', 'table_test', 'INTEGER')
column.owner.indices.append(get_grt_index('i_test', columns=[column]))
column_obj = ColumnObject(column, primary=True)
self.assertEquals('id', column_obj.name)
self.assertEquals(
' id = Column("id_something", INTEGER, autoincrement=False, primary_key=True) # pylint: disable=invalid-name',
str(column_obj))
def test_basics(self):
id_col = get_grt_column('id',
'table_test',
'INT(16)',
isNotNull=1,
autoIncrement=1)
name_col = get_grt_column('name',
'table_test',
'VARCHAR(145)',
isNotNull=1,
comment="toprint=True")
description_col = get_grt_column('description', 'table_test', 'BLOB')
table = get_grt_table(
'table_test',
columns=[id_col, name_col, description_col],
indices=[get_grt_index('i_test', columns=[id_col])])
self.assertEquals(
'class TableTest(DECLARATIVE_BASE):\n'
'\n'
' __tablename__ = \'table_test\'\n'
' __table_args__ = (\n'
' {\'mysql_charset\': \'utf8\', \'sqlite_autoincrement\': True}\n'
' )\n'
'\n'
' id = Column(INTEGER, nullable=False, autoincrement=True, primary_key=True) # pylint: disable=invalid-name\n'
' name = Column(VARCHAR(145), nullable=False)\n'
' description = Column(BLOB)\n'
'\n'
' def __repr__(self):\n'
' return self.__str__()\n'
'\n'
' def __str__(self):\n'
' return "<TableTest(%(id)s, %(name)s)>" % self.__dict__',
str(TableObject(table)))
def test_datetime(self):
column_obj = ColumnObject(
get_grt_column('test_column',
'test_table',
'DATETIME',
defaultValue='CURRENT_TIMESTAMP',
comment='alias=test'))
self.assertEquals(
' test = Column("test_column", DATETIME, default=datetime.datetime.utcnow)',
str(column_obj))
column_obj._column.defaultValue = 'CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'
self.assertEquals(
' test = Column("test_column", DATETIME, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow)',
str(column_obj))
def test_few_options(self):
column = get_grt_column('test',
'test',
'INTEGER',
defaultValue='1234567890',
isNotNull=1,
autoIncrement=1)
column_obj = ColumnObject(column,
primary=True,
unique=True,
index=True)
self.assertEquals(
' test = Column(\n'
' INTEGER, nullable=False, autoincrement=True, primary_key=True, unique=True, index=True, default=1234567890\n'
' )', str(column_obj))
def test_indices(self):
id_col = get_grt_column('id',
'table_test',
'INT(16)',
isNotNull=1,
autoIncrement=1)
name_col = get_grt_column('name',
'table_test',
'VARCHAR(145)',
isNotNull=1,
comment="toprint=True")
description_col = get_grt_column('description', 'table_test', 'BLOB')
unique_1 = get_grt_column('unique_1', 'table_test', 'INT(16)')
unique_2 = get_grt_column('unique_2', 'table_test', 'INT(16)')
index_1 = get_grt_column('index_1', 'table_test', 'INT(16)')
index_2 = get_grt_column('index_2', 'table_test', 'INT(16)')
table = get_grt_table('table_test',
columns=[
id_col, name_col, description_col, unique_1,
unique_2, index_1, index_2
],
indices=[
get_grt_index('i_primary', columns=[id_col]),
get_grt_index('i_unique_single',
'UNIQUE',
columns=[name_col]),
get_grt_index('i_index_single',
'INDEX',
columns=[description_col]),
get_grt_index('i_unique_multi',
'UNIQUE',
columns=[unique_1, unique_2]),
get_grt_index('i_index_multi',
'INDEX',
columns=[index_1, index_2]),
])
self.maxDiff = None
self.assertEquals(
'class TableTest(DECLARATIVE_BASE):\n'
'\n'
' __tablename__ = \'table_test\'\n'
' __table_args__ = (\n'
' {\'mysql_charset\': \'utf8\', \'sqlite_autoincrement\': True},\n'
' UniqueConstraint("unique_1", "unique_2", name="i_unique_multi"),\n'
' Index("index_1", "index_2", name="i_index_multi")\n'
' )\n'
'\n'
' id = Column(INTEGER, nullable=False, autoincrement=True, primary_key=True) # pylint: disable=invalid-name\n'
' name = Column(VARCHAR(145), nullable=False, unique=True)\n'
' description = Column(BLOB, index=True)\n'
' unique_1 = Column(INTEGER)\n'
' unique_2 = Column(INTEGER)\n'
' index_1 = Column(INTEGER)\n'
' index_2 = Column(INTEGER)\n'
'\n'
' def __repr__(self):\n'
' return self.__str__()\n'
'\n'
' def __str__(self):\n'
' return "<TableTest(%(id)s, %(name)s)>" % self.__dict__',
str(TableObject(table)))
def test_with_foreignkeys(self):
id_col = get_grt_column('id',
'table_test',
'INT(16)',
isNotNull=1,
autoIncrement=1)
id_other_ref = get_grt_column('id',
'table_test_other',
'INT(16)',
isNotNull=1)
id_other = get_grt_column('id_other',
'table_test',
'INT(16)',
isNotNull=1)
id_other2 = get_grt_column('id_other2',
'table_test',
'INT(16)',
isNotNull=1,
comment="relation=False")
id_other3 = get_grt_column('id_other3',
'table_test',
'INT(16)',
isNotNull=1,
comment="backref=False")
id_other4 = get_grt_column('id_other4',
'table_test',
'INT(16)',
isNotNull=1,
comment="remote_side='alias'")
id_other5 = get_grt_column('id_other5',
'table_test',
'INT(16)',
isNotNull=1,
comment="backrefname=newbr")
id_other6 = get_grt_column('id_other6',
'table_test',
'INT(16)',
isNotNull=1,
comment="uselist=False")
table = get_grt_table(
'table_test',
columns=[
id_col, id_other, id_other2, id_other3, id_other4, id_other5,
id_other6
],
indices=[get_grt_index('i_test', columns=[id_col])],
foreignKeys=[
get_grt_foreignKey('fk_id_other',
columns=[id_other],
referencedColumns=[id_other_ref]),
get_grt_foreignKey('fk_id_other2',
columns=[id_other2],
referencedColumns=[id_other_ref]),
get_grt_foreignKey('fk_id_other3',
columns=[id_other3],
referencedColumns=[id_other_ref]),
get_grt_foreignKey('fk_id_other4',
columns=[id_other4],
referencedColumns=[id_other_ref]),
get_grt_foreignKey('fk_id_other5',
columns=[id_other5],
referencedColumns=[id_other_ref]),
get_grt_foreignKey('fk_id_other6',
columns=[id_other6],
referencedColumns=[id_other_ref]),
])
self.maxDiff = None
self.assertEquals(
'class TableTest(DECLARATIVE_BASE):\n'
'\n'
' __tablename__ = \'table_test\'\n'
' __table_args__ = (\n'
' {\'mysql_charset\': \'utf8\', \'sqlite_autoincrement\': True}\n'
' )\n'
'\n'
' id = Column(INTEGER, nullable=False, autoincrement=True, primary_key=True) # pylint: disable=invalid-name\n'
' id_other = Column(\n'
' INTEGER, ForeignKey("table_test_other.id", name="fk_id_other", onupdate="SET NULL"), nullable=False\n'
' )\n'
' id_other2 = Column(\n'
' INTEGER, ForeignKey("table_test_other.id", name="fk_id_other2", onupdate="SET NULL"), nullable=False\n'
' )\n'
' id_other3 = Column(\n'
' INTEGER, ForeignKey("table_test_other.id", name="fk_id_other3", onupdate="SET NULL"), nullable=False\n'
' )\n'
' id_other4 = Column(\n'
' INTEGER, ForeignKey("table_test_other.id", name="fk_id_other4", onupdate="SET NULL"), nullable=False\n'
' )\n'
' id_other5 = Column(\n'
' INTEGER, ForeignKey("table_test_other.id", name="fk_id_other5", onupdate="SET NULL"), nullable=False\n'
' )\n'
' id_other6 = Column(\n'
' INTEGER, ForeignKey("table_test_other.id", name="fk_id_other6", onupdate="SET NULL"), nullable=False\n'
' )\n'
'\n'
' tableTestOther = relationship("TableTestOther", foreign_keys=[id_other], backref="tableTest")\n'
' # relation for id_other2.ForeignKey ignored as configured in column comment\n'
' tableTestOther = relationship("TableTestOther", foreign_keys=[id_other3])\n'
' tableTestOther = relationship(\n'
' "TableTestOther", foreign_keys=[id_other4], backref="tableTest", remote_side=[\'alias\']\n'
' )\n'
' tableTestOther = relationship("TableTestOther", foreign_keys=[id_other5], backref="newbr")\n'
' tableTestOther = relationship("TableTestOther", foreign_keys=[id_other6], backref="tableTest", uselist=False)\n'
'\n'
' def __repr__(self):\n'
' return self.__str__()\n'
'\n'
' def __str__(self):\n'
' return "<TableTest(%(id)s)>" % self.__dict__',
str(TableObject(table)))
