def test_get_document_count(self):
mock_info = MagicMock(return_value={"num_docs": "10"})
redisearch._get_redisearch_index_client = MagicMock(return_value=MagicMock(info=mock_info))
count = redisearch.get_document_count("banana")
self.assertEqual(count, 10)
redisearch._get_redisearch_index_client.assert_called_once_with("banana")
mock_info.assert_called_once_with()
def create_contract_v1(body: dict) -> dict:
"""Deploy a new contract on the chain
Args:
body: The parsed body, supplied by user. This is used to create a data model
Returns:
DTO of the contract at rest which is being created
"""
# Before anything else, check to see if this chain has too many contracts
if redisearch.get_document_count(
redisearch.Indexes.smartcontract.value) >= MAX_CONTRACT_LIMIT:
raise exceptions.ContractLimitExceeded(MAX_CONTRACT_LIMIT)
# Create model and validate fields
_log.info(f"Creating data model for {body['txn_type']}")
contract = smart_contract_model.new_contract_from_user(body)
# Check that this transaction type isn't already taken
try:
transaction_type_dao.get_registered_transaction_type(contract.txn_type)
raise exceptions.TransactionTypeConflict(
f"Transaction type {contract.txn_type} already registered")
except exceptions.NotFound:
pass
# Start build task
job_processor.begin_task(
contract, task_type=smart_contract_model.ContractActions.CREATE)
try:
# Register new transaction type for smart contract
_log.info("Registering new smart contract transaction type")
smart_contract_dao.add_smart_contract_index(contract)
transaction_type_dao.register_smart_contract_transaction_type(
contract, body.get("custom_indexes"))
return contract.export_as_at_rest()
except Exception: # Try to cleanup if contract doesn't create successfully
_log.exception(
"Error creating contract index or transaction type. Reverting")
job_processor.begin_task(
contract, task_type=smart_contract_model.ContractActions.DELETE)
raise
