fa3adb968b
Moves all tests under orm/services/audit_server to top level tests folder, making minimal necessary changes to get them working. Change-Id: Ia1ed4d2db2e0f75b110ba0d2c4f5658930a1b1ad
97 lines
4.7 KiB
Python
Executable File
97 lines
4.7 KiB
Python
Executable File
"""test_transaction module."""
|
|
|
|
from mock import patch
|
|
|
|
from orm.services.audit_trail_manager.audit_server.controllers.v1.transaction import QueryResult
|
|
from orm.services.audit_trail_manager.audit_server.model.transaction import Model as TransactionModel
|
|
from orm.services.audit_trail_manager.audit_server.model.transaction_query_result import \
|
|
Model as TransactionQueryResultModel
|
|
from orm.services.audit_trail_manager.audit_server.services import transaction as transaction_service
|
|
from orm.services.audit_trail_manager.audit_server.tests.controllers.v1.functional_test import FunctionalTest
|
|
|
|
|
|
class Test(FunctionalTest):
|
|
"""test transaction class."""
|
|
|
|
transaction_model = TransactionModel(timestamp=111, user_id="user_id_1",
|
|
application_id="application_id_1",
|
|
tracking_id="tracking_id_1",
|
|
external_id="external_id_1",
|
|
transaction_id="transaction_id_1",
|
|
transaction_type="transaction_type_1",
|
|
event_details="event_details_1",
|
|
resource_id="resource_id_1",
|
|
service_name="service_name_1")
|
|
|
|
@patch.object(transaction_service, 'get_transactions',
|
|
return_value=TransactionQueryResultModel(
|
|
[transaction_model]))
|
|
def test_get_all(self, mock_get_transactions):
|
|
"""test that get_one returns an appropriate json result."""
|
|
url = "/audit/transaction?q.timestamp_from=1111&q.timestamp_to=5555&" \
|
|
"q.user_id=user1&q.application_id=SSP&limit=15&marker=1"
|
|
output = self.get_json(url)
|
|
returned_transactions = output['transactions']
|
|
self.assertIsNotNone(returned_transactions)
|
|
self.assertEqual(len(returned_transactions), 1)
|
|
transaction = returned_transactions[0]
|
|
self._assert_returned_transaction(transaction)
|
|
|
|
@patch.object(transaction_service, 'get_transactions',
|
|
return_value=TransactionQueryResultModel(
|
|
[transaction_model]))
|
|
def test_get_all_with_empty_query(self, mock_get_transactions):
|
|
"""Test that get_one accepts an empty query and returns a response."""
|
|
url = "/audit/transaction?limit=15&marker=1"
|
|
output = self.get_json(url)
|
|
returned_transactions = output['transactions']
|
|
self.assertIsNotNone(returned_transactions)
|
|
self.assertEqual(len(returned_transactions), 1)
|
|
transaction = returned_transactions[0]
|
|
self._assert_returned_transaction(transaction)
|
|
|
|
@patch.object(transaction_service, 'add_transaction')
|
|
def test_post(self, mock_transaction_service):
|
|
"""Test that post is executed with no exceptions."""
|
|
url = "/audit/transaction"
|
|
body = {
|
|
"timestamp": 111,
|
|
"user_id": "user_1",
|
|
"application_id": "application_id_1",
|
|
"tracking_id": "tracking_id_1",
|
|
"external_id": "external_id_1",
|
|
"transaction_id": "transaction_id_1",
|
|
"transaction_type": "transaction_type_1",
|
|
"event_details": "event_details_1",
|
|
"resource_id": "resource_id_1",
|
|
"service_name": "service_name_1"
|
|
}
|
|
self.post_json(url, body)
|
|
|
|
def test_init_QueryResult(self):
|
|
"""test the init method."""
|
|
QueryResult(**{"prop1": "value1", "prop2": "value2"})
|
|
|
|
def _assert_returned_transaction(self, transaction):
|
|
"""Check the returned trasaction."""
|
|
self.assertEqual(transaction['timestamp'],
|
|
self.transaction_model.timestamp)
|
|
self.assertEqual(transaction['user_id'],
|
|
self.transaction_model.user_id)
|
|
self.assertEqual(transaction['application_id'],
|
|
self.transaction_model.application_id)
|
|
self.assertEqual(transaction['tracking_id'],
|
|
self.transaction_model.tracking_id)
|
|
self.assertEqual(transaction['external_id'],
|
|
self.transaction_model.external_id)
|
|
self.assertEqual(transaction['transaction_id'],
|
|
self.transaction_model.transaction_id)
|
|
self.assertEqual(transaction['transaction_type'],
|
|
self.transaction_model.transaction_type)
|
|
self.assertEqual(transaction['event_details'],
|
|
self.transaction_model.event_details)
|
|
self.assertEqual(transaction['resource_id'],
|
|
self.transaction_model.resource_id)
|
|
self.assertEqual(transaction['service_name'],
|
|
self.transaction_model.service_name)
|