def test_name_schema_db_query_update(session): """Start with a blank database.""" # setup test_name = "my good company" name_json = {"name": test_name, "state": "NE", "conflict1": "conflict1", "conflict2": "conflict2", "conflict3": "conflict3", "consumptionDate": None, "designation": "LLC", "conflict1_num": "NR 0000001", "decision_text": "my descision", "conflict3_num": "NR 0000003", "conflict2_num": "NR 0000002", "choice": 1} name_schema = NameSchema() # create a name name = Name(name="temp") session.add(name) session.commit() name_id = name.id # update it from the json data vi the schema loader name_schema.load(name_json, instance=name, partial=False) session.add(name) session.commit() # get the name from the data base & assert it was updated name = None name = Name.query.filter_by(name=test_name).one_or_none() assert name.id is not None assert name.id == name_id assert name.name == test_name assert name_json == name_schema.dump(name).data
def test_name_schema_db_update(session): """Start with a blank database.""" # setup test_name = "my good company" name_json = { "name": "my good company", "state": "NE", "conflict1": "conflict1", "conflict2": "conflict2", "conflict3": "conflict3", "designation": "LLC", "conflict1_num": "NR 0000001", "decision_text": "my descision", "conflict3_num": "NR 0000003", "conflict2_num": "NR 0000002", "choice": 1 } name_schema = NameSchema() # create a name and add to the DB name = Name(name="temp") session.add(name) session.commit() # update the name and resave name_schema.load(name_json, instance=name, partial=False) session.add(name) session.commit() assert name.id is not None assert name.name == test_name assert name_json == name_schema.dump(name).data
def test_name_schema(): """Start with a blank database.""" name_json = { "name": "my good company", "state": "NE", "conflict1": "conflict1", "conflict2": "conflict2", "conflict3": "conflict3", "designation": "LLC", "conflict1_num": "NR 0000001", "decision_text": "my descision", "conflict3_num": "NR 0000003", "conflict2_num": "NR 0000002", "choice": 1 } name_schema = NameSchema() name = Name(name="my good company", state="NE", conflict1="conflict1", conflict2="conflict2", conflict3="conflict3", consumptionDate=None, corpNum=None, designation="LLC", conflict1_num="NR 0000001", decision_text="my descision", conflict3_num="NR 0000003", conflict2_num="NR 0000002", choice=1) name_dump = name_schema.dump(name).data assert name_dump == name_json
def test_name_update_via_save_to_db_method(session): """Test save via save_to_db() method, which does data transformations.""" # setup test_name = "my good company" name_json = {"name": test_name, "state": "NE", "conflict1": "conflict1", "conflict2": "conflict2", "conflict3": "", "consumptionDate": None, "designation": "LLC", "conflict1_num": "NR 0000001", "decision_text": "my descision", "conflict3_num": "", "conflict2_num": "A123456", "choice": 1} name_schema = NameSchema() # create a name name = Name(name="temp") session.add(name) session.commit() name_id = name.id # update it from the json data vi the schema loader name_schema.load(name_json, instance=name, partial=False) name.save_to_db() # get the name from the data base & assert it was updated with data changes name = None name = Name.query.filter_by(id=name_id).one_or_none() assert name.id is not None assert name.id == name_id assert name.name == test_name.upper() # name should be uppercase assert name.conflict1_num == "NR 0000001" # NR number conflicts should be unchanged assert name.conflict2_num == "A123456" # corp number conflicts should be unchanged assert name.conflict3_num in ("", None)
from namex.utils.util import cors_preflight from namex.analytics import SolrQueries, RestrictedWords, VALID_ANALYSIS as ANALYTICS_VALID_ANALYSIS import datetime # Register a local namespace for the requests api = Namespace( 'nameRequests', description='Name Request System - Core API for reviewing a Name Request') # Marshmallow schemas request_schema = RequestsSchema(many=False) request_schemas = RequestsSchema(many=True) request_header_schema = RequestsHeaderSchema(many=False) names_schema = NameSchema(many=False) names_schemas = NameSchema(many=True) nwpta_schema = PartnerNameSystemSchema(many=False) applicant_schema = ApplicantSchema(many=False) @api.errorhandler(AuthError) def handle_auth_error(ex): response = jsonify(ex.error) response.status_code = ex.status_code return response # noinspection PyUnresolvedReferences @cors_preflight("GET")