def test_serialize(self, session):
connection_model = Connection(
conn_id='mysql_default',
conn_type='mysql',
host='mysql',
login='******',
schema='testschema',
port=80,
)
session.add(connection_model)
session.commit()
connection_model = session.query(Connection).first()
deserialized_connection = connection_collection_item_schema.dump(
connection_model)
self.assertEqual(
deserialized_connection,
{
'connection_id': "mysql_default",
'conn_type': 'mysql',
'host': 'mysql',
'login': '******',
'schema': 'testschema',
'port': 80,
},
)
def get_connection(connection_id, session):
"""
Get a connection entry
"""
connection = session.query(Connection).filter(Connection.conn_id == connection_id).one_or_none()
if connection is None:
raise NotFound("Connection not found")
return connection_collection_item_schema.dump(connection)