Skip to content

Commit 7b3ffc7

Browse files
committed
fix: remove legacy tests
1 parent 98ce8ae commit 7b3ffc7

File tree

1 file changed

+3
-81
lines changed

1 file changed

+3
-81
lines changed

integration-tests/tests/test_integration.py

Lines changed: 3 additions & 81 deletions
Original file line numberDiff line numberDiff line change
@@ -359,21 +359,6 @@ def agent_publish_keypair(self, agent_keystore_path, sync_accounts):
359359
LOGGER.debug(f"Publisher {address.stdout.strip()} balance: {balance.stdout.strip()}")
360360
time.sleep(8)
361361

362-
@pytest.fixture
363-
def agent_keystore(self, agent_keystore_path, agent_publish_keypair):
364-
self.run(
365-
f"../scripts/init_key_store.sh localnet {agent_keystore_path}")
366-
367-
if USE_ACCUMULATOR:
368-
path = os.path.join(agent_keystore_path, "accumulator_program_key.json")
369-
370-
with open(path, 'w') as f:
371-
f.write(MESSAGE_BUFFER_PROGRAM)
372-
373-
if os.path.exists("keystore"):
374-
os.remove("keystore")
375-
os.symlink(agent_keystore_path, "keystore")
376-
377362
@pytest_asyncio.fixture
378363
async def initialize_message_buffer_program(self, funding_keypair, sync_key_path, sync_accounts):
379364

@@ -429,7 +414,7 @@ async def initialize_message_buffer_program(self, funding_keypair, sync_key_path
429414
await provider.send(tx, [parsed_funding_keypair])
430415

431416
@pytest.fixture
432-
def agent_config(self, agent_keystore, agent_keystore_path, tmp_path):
417+
def agent_config(self, agent_keystore_path, tmp_path):
433418
with open("agent_conf.toml") as config_file:
434419
agent_config = config_file.read()
435420

@@ -454,32 +439,7 @@ def agent_config(self, agent_keystore, agent_keystore_path, tmp_path):
454439
return path
455440

456441
@pytest.fixture
457-
def agent_legacy_config(self, agent_keystore, agent_keystore_path, tmp_path):
458-
"""
459-
Prepares a legacy v1.x.x config for testing agent-migrate-config
460-
"""
461-
with open("agent_conf.toml") as config_file:
462-
agent_config = config_file.read()
463-
464-
agent_config += f'\nkey_store.root_path = "{agent_keystore_path}"'
465-
466-
if USE_ACCUMULATOR:
467-
# Add accumulator setting to verify that it is inlined as well
468-
agent_config += f'\nkey_store.accumulator_key_path = "accumulator_program_key.json"'
469-
470-
LOGGER.debug(f"Built legacy agent config:\n{agent_config}")
471-
472-
path = os.path.join(tmp_path, "agent_conf_legacy.toml")
473-
474-
with open(path, 'w') as f:
475-
f.write(agent_config)
476-
477-
return path
478-
479-
480-
481-
@pytest.fixture
482-
def agent(self, sync_accounts, agent_keystore, tmp_path, initialize_message_buffer_program, agent_config):
442+
def agent(self, sync_accounts, tmp_path, initialize_message_buffer_program, agent_config):
483443
LOGGER.debug("Building agent binary")
484444
self.run("cargo build --release --bin agent")
485445

@@ -493,7 +453,7 @@ def agent(self, sync_accounts, agent_keystore, tmp_path, initialize_message_buff
493453
yield
494454

495455
@pytest.fixture
496-
def agent_hotload(self, sync_accounts, agent_keystore, agent_keystore_path, tmp_path, initialize_message_buffer_program, agent_config):
456+
def agent_hotload(self, sync_accounts, agent_keystore_path, tmp_path, initialize_message_buffer_program, agent_config):
497457
"""
498458
Spawns an agent without a publish keypair, used for keypair hotloading testing
499459
"""
@@ -723,44 +683,6 @@ async def test_publish_forever(self, client: PythAgentClient, tmp_path):
723683
await client.update_price(price_account, 47, 2, "trading")
724684
time.sleep(1)
725685

726-
@pytest.mark.asyncio
727-
async def test_agent_migrate_config(self,
728-
agent_keystore,
729-
agent_legacy_config,
730-
agent_migrate_config_binary,
731-
client_no_spawn: PythAgentClient,
732-
initialize_message_buffer_program,
733-
sync_accounts,
734-
tmp_path,
735-
):
736-
os.environ["RUST_BACKTRACE"] = "full"
737-
os.environ["RUST_LOG"] = "debug"
738-
739-
# Migrator must run successfully (run() raises on error)
740-
new_config = self.run(f"{agent_migrate_config_binary} -c {agent_legacy_config}").stdout.strip()
741-
742-
LOGGER.debug(f"Successfully migrated legacy config to:\n{new_config}")
743-
744-
# Overwrite legacy config with the migrated version.
745-
#
746-
# NOTE: assumes 'w' erases the file before access)
747-
with open(agent_legacy_config, 'w') as f:
748-
f.write(new_config)
749-
f.flush()
750-
751-
self.run("cargo build --release --bin agent")
752-
753-
log_dir = os.path.join(tmp_path, "agent_logs")
754-
755-
# We start the agent manually to pass it the updated legacy config
756-
with self.spawn(f"../target/release/agent --config {agent_legacy_config}", log_dir=log_dir):
757-
time.sleep(3)
758-
await client_no_spawn.connect()
759-
760-
# Continue with the simple test case, which must succeed
761-
await self.test_update_price_simple(client_no_spawn)
762-
await client_no_spawn.close()
763-
764686
@pytest.mark.asyncio
765687
async def test_agent_respects_market_hours(self, client: PythAgentClient):
766688
'''

0 commit comments

Comments
 (0)