def test_fill_example_table_data(data_fixture):
user = data_fixture.create_user()
database = data_fixture.create_database_application(user=user)
table_handler = TableHandler()
table_handler.create_table(user, database, fill_example=True, name='Table 1')
assert Table.objects.all().count() == 1
assert GridView.objects.all().count() == 1
assert TextField.objects.all().count() == 1
assert LongTextField.objects.all().count() == 1
assert BooleanField.objects.all().count() == 1
assert GridViewFieldOptions.objects.all().count() == 2
def test_create_database_table(send_mock, data_fixture):
user = data_fixture.create_user()
user_2 = data_fixture.create_user()
database = data_fixture.create_database_application(user=user)
handler = TableHandler()
handler.create_table(user=user, database=database, name="Test table")
assert Table.objects.all().count() == 1
assert TextField.objects.all().count() == 1
table = Table.objects.all().first()
assert table.name == "Test table"
assert table.order == 1
assert table.database == database
primary_field = TextField.objects.all().first()
assert primary_field.table == table
assert primary_field.primary
assert primary_field.name == "Name"
send_mock.assert_called_once()
assert send_mock.call_args[1]["table"].id == table.id
assert send_mock.call_args[1]["user"].id == user.id
with pytest.raises(UserNotInGroup):
handler.create_table(user=user_2, database=database, name="")
assert f"database_table_{table.id}" in connection.introspection.table_names(
)
model = table.get_model(attribute_names=True)
row = model.objects.create(name="Test")
assert row.name == "Test"
with pytest.raises(TypeError):
model.objects.create(does_not_exists=True)
assert model.objects.count() == 1
row = model.objects.get(id=row.id)
assert row.name == "Test"
def test_create_database_table(data_fixture):
user = data_fixture.create_user()
user_2 = data_fixture.create_user()
database = data_fixture.create_database_application(user=user)
handler = TableHandler()
handler.create_table(user=user, database=database, name='Test table')
assert Table.objects.all().count() == 1
assert TextField.objects.all().count() == 1
table = Table.objects.all().first()
assert table.name == 'Test table'
assert table.order == 1
assert table.database == database
primary_field = TextField.objects.all().first()
assert primary_field.table == table
assert primary_field.primary
assert primary_field.name == 'Name'
with pytest.raises(UserNotInGroupError):
handler.create_table(user=user_2, database=database, name='')
assert f'database_table_{table.id}' in connection.introspection.table_names(
)
model = table.get_model(attribute_names=True)
row = model.objects.create(name='Test')
assert row.name == 'Test'
with pytest.raises(TypeError):
model.objects.create(does_not_exists=True)
assert model.objects.count() == 1
row = model.objects.get(id=row.id)
assert row.name == 'Test'
def test_fill_example_table_data(data_fixture):
user = data_fixture.create_user()
database = data_fixture.create_database_application(user=user)
table_handler = TableHandler()
table = table_handler.create_table(user, database, fill_example=True,
name='Table 1')
assert Table.objects.all().count() == 1
assert GridView.objects.all().count() == 1
assert TextField.objects.all().count() == 1
assert LongTextField.objects.all().count() == 1
assert BooleanField.objects.all().count() == 1
assert GridViewFieldOptions.objects.all().count() == 2
model = table.get_model()
results = model.objects.all()
assert len(results) == 2
assert results[0].order == Decimal('1.00000000000000000000')
assert results[1].order == Decimal('2.00000000000000000000')
def test_fill_table_with_initial_data(data_fixture):
user = data_fixture.create_user()
database = data_fixture.create_database_application(user=user)
table_handler = TableHandler()
with pytest.raises(InvalidInitialTableData):
table_handler.create_table(user, database, name="Table 1", data=[])
with pytest.raises(InvalidInitialTableData):
table_handler.create_table(user, database, name="Table 1", data=[[]])
limit = settings.INITIAL_TABLE_DATA_LIMIT
settings.INITIAL_TABLE_DATA_LIMIT = 2
with pytest.raises(InitialTableDataLimitExceeded):
table_handler.create_table(user,
database,
name="Table 1",
data=[[], [], []])
settings.INITIAL_TABLE_DATA_LIMIT = limit
field_limit = settings.MAX_FIELD_LIMIT
settings.MAX_FIELD_LIMIT = 2
with pytest.raises(MaxFieldLimitExceeded):
table_handler.create_table(user,
database,
name="Table 1",
data=[["fields"] * 3, ["rows"] * 3])
settings.MAX_FIELD_LIMIT = field_limit
data = [
["A", "B", "C", "D"],
["1-1", "1-2", "1-3", "1-4", "1-5"],
["2-1", "2-2", "2-3"],
["3-1", "3-2"],
]
table = table_handler.create_table(user,
database,
name="Table 1",
data=data,
first_row_header=True)
text_fields = TextField.objects.filter(table=table)
assert text_fields[0].name == "A"
assert text_fields[1].name == "B"
assert text_fields[2].name == "C"
assert text_fields[3].name == "D"
assert text_fields[4].name == "Field 5"
assert GridView.objects.all().count() == 1
model = table.get_model()
results = model.objects.all()
assert results[0].order == Decimal("1.00000000000000000000")
assert results[1].order == Decimal("2.00000000000000000000")
assert results[2].order == Decimal("3.00000000000000000000")
assert getattr(results[0], f"field_{text_fields[0].id}") == "1-1"
assert getattr(results[0], f"field_{text_fields[1].id}") == "1-2"
assert getattr(results[0], f"field_{text_fields[2].id}") == "1-3"
assert getattr(results[0], f"field_{text_fields[3].id}") == "1-4"
assert getattr(results[0], f"field_{text_fields[4].id}") == "1-5"
assert getattr(results[1], f"field_{text_fields[0].id}") == "2-1"
assert getattr(results[1], f"field_{text_fields[1].id}") == "2-2"
assert getattr(results[1], f"field_{text_fields[2].id}") == "2-3"
assert getattr(results[1], f"field_{text_fields[3].id}") == ""
assert getattr(results[1], f"field_{text_fields[4].id}") == ""
assert getattr(results[2], f"field_{text_fields[0].id}") == "3-1"
assert getattr(results[2], f"field_{text_fields[1].id}") == "3-2"
assert getattr(results[2], f"field_{text_fields[2].id}") == ""
assert getattr(results[2], f"field_{text_fields[3].id}") == ""
assert getattr(results[2], f"field_{text_fields[4].id}") == ""
data = [
["1-1"],
["2-1", "2-2", "2-3"],
["3-1", "3-2"],
]
table = table_handler.create_table(user,
database,
name="Table 2",
data=data,
first_row_header=False)
text_fields = TextField.objects.filter(table=table)
assert text_fields[0].name == "Field 1"
assert text_fields[1].name == "Field 2"
assert text_fields[2].name == "Field 3"
assert GridView.objects.all().count() == 2
model = table.get_model()
results = model.objects.all()
assert getattr(results[0], f"field_{text_fields[0].id}") == "1-1"
assert getattr(results[0], f"field_{text_fields[1].id}") == ""
assert getattr(results[0], f"field_{text_fields[2].id}") == ""
assert getattr(results[1], f"field_{text_fields[0].id}") == "2-1"
assert getattr(results[1], f"field_{text_fields[1].id}") == "2-2"
assert getattr(results[1], f"field_{text_fields[2].id}") == "2-3"
assert getattr(results[2], f"field_{text_fields[0].id}") == "3-1"
assert getattr(results[2], f"field_{text_fields[1].id}") == "3-2"
field_limit = settings.MAX_FIELD_LIMIT
settings.MAX_FIELD_LIMIT = 5
data = [
["A", "B", "C", "D", "E"],
["1-1", "1-2", "1-3", "1-4", "1-5"],
]
table = table_handler.create_table(user,
database,
name="Table 3",
data=data,
first_row_header=True)
num_fields = table.field_set.count()
assert GridView.objects.all().count() == 3
assert num_fields == settings.MAX_FIELD_LIMIT
settings.MAX_FIELD_LIMIT = field_limit
def test_fill_table_with_initial_data(data_fixture):
user = data_fixture.create_user()
database = data_fixture.create_database_application(user=user)
table_handler = TableHandler()
with pytest.raises(InvalidInitialTableData):
table_handler.create_table(user, database, name='Table 1', data=[])
with pytest.raises(InvalidInitialTableData):
table_handler.create_table(user, database, name='Table 1', data=[[]])
limit = settings.INITIAL_TABLE_DATA_LIMIT
settings.INITIAL_TABLE_DATA_LIMIT = 2
with pytest.raises(InitialTableDataLimitExceeded):
table_handler.create_table(user, database, name='Table 1', data=[[], [], []])
settings.INITIAL_TABLE_DATA_LIMIT = limit
data = [
['A', 'B', 'C', 'D'],
['1-1', '1-2', '1-3', '1-4', '1-5'],
['2-1', '2-2', '2-3'],
['3-1', '3-2'],
]
table = table_handler.create_table(user, database, name='Table 1', data=data,
first_row_header=True)
text_fields = TextField.objects.filter(table=table)
assert text_fields[0].name == 'A'
assert text_fields[1].name == 'B'
assert text_fields[2].name == 'C'
assert text_fields[3].name == 'D'
assert text_fields[4].name == 'Field 5'
assert GridView.objects.all().count() == 1
model = table.get_model()
results = model.objects.all()
assert getattr(results[0], f'field_{text_fields[0].id}') == '1-1'
assert getattr(results[0], f'field_{text_fields[1].id}') == '1-2'
assert getattr(results[0], f'field_{text_fields[2].id}') == '1-3'
assert getattr(results[0], f'field_{text_fields[3].id}') == '1-4'
assert getattr(results[0], f'field_{text_fields[4].id}') == '1-5'
assert getattr(results[1], f'field_{text_fields[0].id}') == '2-1'
assert getattr(results[1], f'field_{text_fields[1].id}') == '2-2'
assert getattr(results[1], f'field_{text_fields[2].id}') == '2-3'
assert getattr(results[1], f'field_{text_fields[3].id}') == ''
assert getattr(results[1], f'field_{text_fields[4].id}') == ''
assert getattr(results[2], f'field_{text_fields[0].id}') == '3-1'
assert getattr(results[2], f'field_{text_fields[1].id}') == '3-2'
assert getattr(results[2], f'field_{text_fields[2].id}') == ''
assert getattr(results[2], f'field_{text_fields[3].id}') == ''
assert getattr(results[2], f'field_{text_fields[4].id}') == ''
data = [
['1-1'],
['2-1', '2-2', '2-3'],
['3-1', '3-2'],
]
table = table_handler.create_table(user, database, name='Table 2', data=data,
first_row_header=False)
text_fields = TextField.objects.filter(table=table)
assert text_fields[0].name == 'Field 1'
assert text_fields[1].name == 'Field 2'
assert text_fields[2].name == 'Field 3'
assert GridView.objects.all().count() == 2
model = table.get_model()
results = model.objects.all()
assert getattr(results[0], f'field_{text_fields[0].id}') == '1-1'
assert getattr(results[0], f'field_{text_fields[1].id}') == ''
assert getattr(results[0], f'field_{text_fields[2].id}') == ''
assert getattr(results[1], f'field_{text_fields[0].id}') == '2-1'
assert getattr(results[1], f'field_{text_fields[1].id}') == '2-2'
assert getattr(results[1], f'field_{text_fields[2].id}') == '2-3'
assert getattr(results[2], f'field_{text_fields[0].id}') == '3-1'
assert getattr(results[2], f'field_{text_fields[1].id}') == '3-2'