Skip to content

Commit 6ad8724

Browse files
ArturWieczorekArturWieczorek
committed
test(db-sync config): Add db-sync test for governance option
1 parent e7298ef commit 6ad8724

File tree

3 files changed

+92
-12
lines changed

3 files changed

+92
-12
lines changed

‎cardano_node_tests/tests/test_dbsync_config.py‎

Lines changed: 79 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from cardano_node_tests.utils import cluster_nodes
1515
from cardano_node_tests.utils import configuration
1616
from cardano_node_tests.utils import dbsync_service_manager as db_sync
17+
from cardano_node_tests.utils import dbsync_queries
1718
from cardano_node_tests.utils import dbsync_utils
1819
from cardano_node_tests.utils import helpers
1920

@@ -143,7 +144,6 @@ def basic_tx_out(
143144
"""Test `tx_out` option."""
144145
db_config = db_sync_manager.get_config_builder()
145146

146-
# Test tx_out : enable
147147
db_sync_manager.restart_with_config(
148148
custom_config=db_config.with_tx_out(
149149
value=db_sync.TxOutMode.ENABLE, force_tx_in=False, use_address_table=False
@@ -158,7 +158,6 @@ def basic_tx_out(
158158
}
159159
)
160160

161-
# Test tx_out : disable
162161
db_sync_manager.restart_with_config(
163162
custom_config=db_config.with_tx_out(
164163
value=db_sync.TxOutMode.DISABLE, force_tx_in=True, use_address_table=True
@@ -177,6 +176,84 @@ def basic_tx_out(
177176

178177
yield basic_tx_out
179178

179+
def governance(
180+
db_sync_manager: db_sync.DBSyncManager,
181+
):
182+
"""Test `governance` option."""
183+
db_config = db_sync_manager.get_config_builder()
184+
185+
db_sync_manager.restart_with_config(
186+
custom_config=db_config.with_governance(
187+
value=db_sync.SettingState.ENABLE
188+
)
189+
)
190+
# Off-chain vote data is inserted into the database a few minutes after the restart of db-sync
191+
def _query_func():
192+
off_chain_vote_data = next(
193+
iter(dbsync_queries.query_off_chain_vote_data()), None
194+
)
195+
if off_chain_vote_data is None:
196+
msg = f"no any off-chain vota data record found id db-sync database"
197+
raise dbsync_utils.DbSyncNoResponseError(msg)
198+
return off_chain_vote_data
199+
dbsync_utils.retry_query(query_func=_query_func, timeout=360)
200+
201+
check_dbsync_state(
202+
expected_state={
203+
db_sync.Table.COMMITTEE_DE_REGISTRATION: TableCondition.NOT_EMPTY,
204+
db_sync.Table.COMMITTEE_MEMBER: TableCondition.NOT_EMPTY,
205+
db_sync.Table.COMMITTEE_REGISTRATION: TableCondition.NOT_EMPTY,
206+
db_sync.Table.COMMITTEE: TableCondition.NOT_EMPTY,
207+
db_sync.Table.CONSTITUTION: TableCondition.NOT_EMPTY,
208+
db_sync.Table.DELEGATION_VOTE: TableCondition.NOT_EMPTY,
209+
db_sync.Table.DREP_DISTR: TableCondition.NOT_EMPTY,
210+
db_sync.Table.DREP_REGISTRATION: TableCondition.NOT_EMPTY,
211+
db_sync.Table.EPOCH_STATE: TableCondition.NOT_EMPTY,
212+
db_sync.Table.GOV_ACTION_PROPOSAL: TableCondition.NOT_EMPTY,
213+
db_sync.Table.OFF_CHAIN_VOTE_DATA: TableCondition.NOT_EMPTY,
214+
db_sync.Table.OFF_CHAIN_VOTE_DREP_DATA: TableCondition.NOT_EMPTY,
215+
db_sync.Table.OFF_CHAIN_VOTE_EXTERNAL_UPDATE: TableCondition.NOT_EMPTY,
216+
db_sync.Table.OFF_CHAIN_VOTE_FETCH_ERROR: TableCondition.NOT_EMPTY,
217+
db_sync.Table.OFF_CHAIN_VOTE_GOV_ACTION_DATA: TableCondition.NOT_EMPTY,
218+
db_sync.Table.OFF_CHAIN_VOTE_REFERENCE: TableCondition.NOT_EMPTY,
219+
db_sync.Table.VOTING_ANCHOR: TableCondition.NOT_EMPTY,
220+
db_sync.Table.VOTING_PROCEDURE: TableCondition.NOT_EMPTY,
221+
db_sync.Table.TREASURY_WITHDRAWAL: TableCondition.NOT_EMPTY,
222+
}
223+
)
224+
225+
# Test governance : disable
226+
db_sync_manager.restart_with_config(
227+
custom_config=db_config.with_governance(
228+
value=db_sync.SettingState.DISABLE
229+
)
230+
)
231+
check_dbsync_state(
232+
expected_state={
233+
db_sync.Table.COMMITTEE_DE_REGISTRATION: TableCondition.EMPTY,
234+
db_sync.Table.COMMITTEE_MEMBER: TableCondition.EMPTY,
235+
db_sync.Table.COMMITTEE_REGISTRATION: TableCondition.EMPTY,
236+
db_sync.Table.COMMITTEE: TableCondition.EMPTY,
237+
db_sync.Table.CONSTITUTION: TableCondition.EMPTY,
238+
db_sync.Table.DELEGATION_VOTE: TableCondition.EMPTY,
239+
db_sync.Table.DREP_DISTR: TableCondition.EMPTY,
240+
db_sync.Table.DREP_REGISTRATION: TableCondition.EMPTY,
241+
db_sync.Table.EPOCH_STATE: TableCondition.EMPTY,
242+
db_sync.Table.GOV_ACTION_PROPOSAL: TableCondition.EMPTY,
243+
db_sync.Table.OFF_CHAIN_VOTE_DATA: TableCondition.EMPTY,
244+
db_sync.Table.OFF_CHAIN_VOTE_DREP_DATA: TableCondition.EMPTY,
245+
db_sync.Table.OFF_CHAIN_VOTE_EXTERNAL_UPDATE: TableCondition.EMPTY,
246+
db_sync.Table.OFF_CHAIN_VOTE_FETCH_ERROR: TableCondition.EMPTY,
247+
db_sync.Table.OFF_CHAIN_VOTE_GOV_ACTION_DATA: TableCondition.EMPTY,
248+
db_sync.Table.OFF_CHAIN_VOTE_REFERENCE: TableCondition.EMPTY,
249+
db_sync.Table.VOTING_ANCHOR: TableCondition.EMPTY,
250+
db_sync.Table.VOTING_PROCEDURE: TableCondition.EMPTY,
251+
db_sync.Table.TREASURY_WITHDRAWAL: TableCondition.EMPTY,
252+
}
253+
)
254+
255+
yield governance
256+
180257
def tx_cbor_value_enable(
181258
db_sync_manager: db_sync.DBSyncManager,
182259
):

‎cardano_node_tests/utils/dbsync_queries.py‎

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1465,7 +1465,7 @@ def query_treasury_withdrawal(txhash: str) -> tp.Generator[TreasuryWithdrawalDBR
14651465
yield TreasuryWithdrawalDBRow(*result)
14661466

14671467

1468-
def query_off_chain_vote_data(data_hash: str) -> tp.Generator[OffChainVoteDataDBRow, None, None]:
1468+
def query_off_chain_vote_data(data_hash: tp.Optional[str] = None) -> tp.Generator[OffChainVoteDataDBRow, None, None]:
14691469
"""Query the off chain vote data in db-sync."""
14701470
query = (
14711471
"SELECT"
@@ -1482,11 +1482,16 @@ def query_off_chain_vote_data(data_hash: str) -> tp.Generator[OffChainVoteDataDB
14821482
"LEFT JOIN off_chain_vote_gov_action_data gov ON data.id = gov.off_chain_vote_data_id "
14831483
"LEFT JOIN off_chain_vote_reference ref ON data.id = ref.off_chain_vote_data_id "
14841484
"LEFT JOIN voting_anchor va ON data.voting_anchor_id = va.id "
1485-
"WHERE data.hash = %s "
1486-
"ORDER BY va.id, ref.id, auth.id, updt.id;"
14871485
)
14881486

1489-
with execute(query=query, vars=(rf"\x{data_hash}",)) as cur:
1487+
query_params = ()
1488+
if data_hash is not None:
1489+
query += "WHERE data.hash = %s "
1490+
query_params = (rf"\x{data_hash}",)
1491+
1492+
query += "ORDER BY va.id, ref.id, auth.id, updt.id;"
1493+
1494+
with execute(query=query, vars=query_params) as cur:
14901495
while (result := cur.fetchone()) is not None:
14911496
yield OffChainVoteDataDBRow(*result)
14921497

‎cardano_node_tests/utils/dbsync_service_manager.py‎

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -358,14 +358,12 @@ def recreate_database(self) -> None:
358358
This is a DESTRUCTIVE operation!
359359
It will delete existing data and recreate the database from scratch.
360360
"""
361-
scripts_dir = cluster_scripts.get_testnet_variant_scriptdir(
362-
testnet_variant=configuration.TESTNET_VARIANT
363-
)
364-
if not scripts_dir:
365-
err = f"Testnet variant '{configuration.TESTNET_VARIANT}' scripts directory not found."
361+
common_scripts_dir = cluster_scripts.COMMON_DIR
362+
if not common_scripts_dir:
363+
err = f"Common scripts directory '{common_scripts_dir}' not found."
366364
raise RuntimeError(err)
367365

368-
db_script_template = scripts_dir / "postgres-setup.sh"
366+
db_script_template = common_scripts_dir / "postgres-setup.sh"
369367
if not db_script_template.exists() and configuration.BOOTSTRAP_DIR:
370368
db_script_template = pl.Path(configuration.BOOTSTRAP_DIR) / "postgres-setup.sh"
371369

0 commit comments

Comments
 (0)