Python make_table Examples

Example #1

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_no_pk():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'integer'}]}
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, {}, [], bases, module, app_label)
    assert table._meta.pk.name == 'id'

Example #2

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_class_name():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'integer'}]}
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, {}, [], bases, module,
                       app_label)
    assert table.__name__ == 'TestTable'

Example #3

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_pk():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'integer'}]}
    constraint_json = {'primary_keys': [{'fields': ['integer']}]}
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, constraint_json, [], bases, module,
                       app_label)
    assert table._meta.pk.name == 'integer'

Example #4

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_not_null():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'integer'}]}
    constraint_json = {'not_null': [{'fields': ['integer']}]}
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, constraint_json, [], bases, module,
                       app_label)

    for field in table._meta.fields:
        if field.name == 'integer':
            assert not field.null

Example #5

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_index():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'integer'}]}
    index_json = [{'fields': ['integer']}]
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, {}, index_json, bases, module,
                       app_label)

    for field in table._meta.fields:
        if field.name == 'integer':
            assert field.db_index

Example #6

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_multi_pk():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'int1'},
                                                   {'type': 'integer',
                                                    'name': 'int2'}]}
    constraint_json = {'primary_keys': [{'fields': ['int1', 'int2']}]}
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, constraint_json, [], bases, module,
                       app_label)

    assert table._meta.pk.name == 'id'
    assert table._meta.unique_together == (('int1', 'int2'),)

    for field in table._meta.fields:
        if field.name in ['int1', 'int2']:
            assert not field.null

Example #7

File: test_make_table.py Project: chop-dbhi/data-models-django

def test_foreign_key():

    table_json = {'name': 'test_table', 'fields': [{'type': 'integer',
                                                    'name': 'integer'}]}
    constraint_json = {'foreign_keys': [{'source_table': 'test_table',
                                         'source_field': 'integer',
                                         'target_table': 'other',
                                         'target_field': 'field'}]}
    module = 'dmdj.tests'
    app_label = 'dmdj'
    bases = (Model,)
    table = make_table(table_json, constraint_json, [], bases, module,
                       app_label)

    for field in table._meta.fields:
        if field.name == 'integer':
            assert isinstance(field, ForeignKey)
            assert field.to_fields[0] == 'field'
            assert field.related_query_name() == 'test_table_integer_set'