Source code for sqlobject.tests.test_default_style
"""
Test the default styles, to guarantee consistency.
"""
from sqlobject import ForeignKey, MixedCaseStyle, MixedCaseUnderscoreStyle, \
    SQLObject, StringCol, Style
# Hash of styles versus the database names resulting from 'columns' below.
columns = ["ABCUpper", "abc_lower", "ABCamelCaseColumn"]
styles = {
    Style: columns,
    MixedCaseUnderscoreStyle: ["abc_upper", "abc_lower",
                               "ab_camel_case_column"],
    MixedCaseStyle: ["ABCUpper", "Abc_lower", "ABCamelCaseColumn"],
}
# Hash of styles versus the database names
# resulting from a foreign key named 'FKey'.
fkey = ForeignKey("DefaultStyleTest", name="FKey")
fkeys = {
    Style: "FKeyID",
    MixedCaseUnderscoreStyle: "f_key_id",
    MixedCaseStyle: "FKeyID",
}
[docs]def make_columns():
    global columns
    columns = []
    for col_name in columns:
        columns.append(StringCol(name=col_name, length=10)) 
[docs]def do_col_test(DefaultStyleTest, style, dbnames):
    DefaultStyleTest.sqlmeta.style = style()
    for col, old_dbname in zip(columns, dbnames):
        DefaultStyleTest.sqlmeta.addColumn(col)
        try:
            new_dbname = DefaultStyleTest.sqlmeta.columns[col.name].dbName
            assert new_dbname == old_dbname
        finally:
            if col.name in DefaultStyleTest.sqlmeta.columns:
                DefaultStyleTest.sqlmeta.delColumn(col) 
[docs]def do_fkey_test(DefaultStyleTest, style, dbname):
    DefaultStyleTest.sqlmeta.style = style()
    DefaultStyleTest.sqlmeta.addColumn(fkey)
    try:
        assert list(DefaultStyleTest.sqlmeta.columns.keys())[0] == "FKeyID"
        assert list(DefaultStyleTest.sqlmeta.columns.values())[0].dbName == \
            
dbname
    finally:
        DefaultStyleTest.sqlmeta.delColumn(fkey) 
[docs]class DefaultStyleTest(SQLObject):
    pass 
[docs]def test_default_styles():
    make_columns()
    for style in styles:
        do_col_test(DefaultStyleTest, style, styles[style])
        do_fkey_test(DefaultStyleTest, style, fkeys[style])