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")