def test_pq_query(self):
"""Test JSON to the PolicyQuery object."""
policyquery = PolicyQuery(user=-1,
from_table='users',
where={'network_id': 'dmlb2001'},
columns=['last_name', 'first_name'])
result = policyquery.get_results()
self.assertEqual(result[0]['last_name'], u'Brown\u00e9 Jr')
def test_valid_metadata(self):
"""Test the valid metadata method for valid metadata."""
policyquery = PolicyQuery(user=10)
md_obj = metadata_decode("""[
{ "destinationTable": "Transactions.submitter", "value": 10 },
{ "destinationTable": "Transactions.project", "value": "1234a" },
{ "destinationTable": "Transactions.instrument", "value": 54 }
]""")
result = policyquery.valid_metadata(md_obj)
self.assertTrue('status' in result)
self.assertEqual(result['status'], 'success')
def test_get_user(self):
"""Test getting a user and converting it to and from ID/networkID."""
policyquery = PolicyQuery(user=10)
self.assertEqual(policyquery.get_user(), 10)
def test_pq_fromjson(self):
"""Test JSON to the PolicyQuery object."""
policyquery = PolicyQuery.fromjson(
'{ "user": "******", "from": "mytable" }')
self.assertEqual(policyquery.pq_data.from_table, 'mytable')
def test_pq_json(self):
"""Test converting the PolicyQuery to json."""
policyquery = PolicyQuery('dmlb2001')
self.assertEqual(
policyquery.tojson(),
'{"user": 10, "columns": null, "where": null, "from": null}')
def test_pq_init(self):
"""Test the constructor for PolicyQuery."""
policyquery = PolicyQuery('dmlb2001')
self.assertTrue(policyquery)
self.assertEqual(policyquery.get_user(), 10)