Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
purplesmoke05 committed Feb 2, 2024
1 parent 03c1825 commit 64e4b30
Showing 1 changed file with 0 additions and 97 deletions.
97 changes: 0 additions & 97 deletions tests/test_batch_indexer_token_holders.py
Original file line number Diff line number Diff line change
Expand Up @@ -2532,100 +2532,3 @@ async def test_error_3(
)
).all()
assert len(_records) == 0

# <Error_4>
# If DB session fails in phase sinking events, batch logs exception message.
@pytest.mark.asyncio
async def test_error_4(
self,
main_func,
db: Session,
ibet_security_token_escrow_contract,
personal_info_contract,
caplog: pytest.LogCaptureFixture,
):
user_1 = config_eth_account("user1")
issuer_address = user_1["address"]
issuer_private_key = decode_keyfile_json(
raw_keyfile_json=user_1["keyfile_json"], password="password".encode("utf-8")
)
user_2 = config_eth_account("user2")
user_address_1 = user_2["address"]
user_pk_1 = decode_keyfile_json(
raw_keyfile_json=user_2["keyfile_json"], password="password".encode("utf-8")
)
user_3 = config_eth_account("user3")
user_address_2 = user_3["address"]
user_pk_2 = decode_keyfile_json(
raw_keyfile_json=user_3["keyfile_json"], password="password".encode("utf-8")
)

# Issuer issues share token.
token_contract = await deploy_share_token_contract(
issuer_address,
issuer_private_key,
personal_info_contract.address,
tradable_exchange_contract_address=ibet_security_token_escrow_contract.address,
transfer_approval_required=False,
)
token_address_1 = token_contract.address
token_1 = Token()
token_1.type = TokenType.IBET_SHARE.value
token_1.token_address = token_address_1
token_1.issuer_address = issuer_address
token_1.abi = token_contract.abi
token_1.tx_hash = "tx_hash"
token_1.version = TokenVersion.V_22_12
db.add(token_1)

PersonalInfoContractTestUtils.register(
personal_info_contract.address,
user_address_1,
user_pk_1,
[issuer_address, ""],
)
PersonalInfoContractTestUtils.register(
personal_info_contract.address,
user_address_2,
user_pk_2,
[issuer_address, ""],
)
PersonalInfoContractTestUtils.register(
personal_info_contract.address,
issuer_address,
issuer_private_key,
[issuer_address, ""],
)

STContractUtils.transfer(
token_contract.address,
issuer_address,
issuer_private_key,
[user_address_1, 20000],
)
STContractUtils.transfer(
token_contract.address,
user_address_1,
user_pk_1,
[ibet_security_token_escrow_contract.address, 10000],
)

# Insert collection record with above token and current block number
list_id = str(uuid.uuid4())
block_number = web3.eth.block_number
_token_holders_list = token_holders_list(
token_contract.address, block_number, list_id
)
db.add(_token_holders_list)
db.commit()

with patch(
"batch.indexer_token_holders.INDEXER_SYNC_INTERVAL", None
), patch.object(
AsyncSession, "close", side_effect=SQLAlchemyError()
), pytest.raises(
TypeError
):
await main_func()
assert 1 == caplog.text.count("A database error has occurred")
caplog.clear()

0 comments on commit 64e4b30

Please sign in to comment.