66 lines
3.1 KiB
Python
66 lines
3.1 KiB
Python
from .context import mango
|
|
|
|
from solana.publickey import PublicKey
|
|
|
|
|
|
def context_has_default_values(ctx):
|
|
assert ctx.program_id == PublicKey("mv3ekLzLbnVPNxjSKvqBpU3ZeZXPQdEC3bp5MDEBG68")
|
|
assert ctx.dex_program_id == PublicKey("9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin")
|
|
assert ctx.group_name == "mainnet.0"
|
|
assert ctx.group_id == PublicKey("4yJ2Vx3kZnmHTNCrHzdoj5nCwriF2kVhfKNvqC6gU8tr")
|
|
|
|
|
|
def test_context_default_exists():
|
|
assert mango.ContextBuilder.default() is not None
|
|
|
|
|
|
def test_context_default_values():
|
|
context_has_default_values(mango.ContextBuilder.default())
|
|
|
|
|
|
# Need to have more than one working cluster for this test.
|
|
# def test_new_from_cluster():
|
|
# context_has_default_values(mango.ContextBuilder.default())
|
|
# derived = mango.ContextBuilder.default().new_from_cluster("mainnet")
|
|
# assert derived.cluster == "mainnet"
|
|
# assert derived.cluster_url == "https://solana-api.projectserum.com"
|
|
# assert derived.program_id == PublicKey("mv3ekLzLbnVPNxjSKvqBpU3ZeZXPQdEC3bp5MDEBG68")
|
|
# assert derived.dex_program_id == PublicKey("9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin")
|
|
# assert derived.group_name == "mainnet.0"
|
|
# assert derived.group_id == PublicKey("4yJ2Vx3kZnmHTNCrHzdoj5nCwriF2kVhfKNvqC6gU8tr")
|
|
# context_has_default_values(mango.ContextBuilder.default())
|
|
|
|
|
|
def test_new_from_cluster_url():
|
|
context_has_default_values(mango.ContextBuilder.default())
|
|
derived = mango.ContextBuilder.default().new_from_cluster_url("https://some-dev-host")
|
|
assert derived.program_id == PublicKey("mv3ekLzLbnVPNxjSKvqBpU3ZeZXPQdEC3bp5MDEBG68")
|
|
assert derived.dex_program_id == PublicKey("9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin")
|
|
assert derived.group_name == "mainnet.0"
|
|
assert derived.group_id == PublicKey("4yJ2Vx3kZnmHTNCrHzdoj5nCwriF2kVhfKNvqC6gU8tr")
|
|
context_has_default_values(mango.ContextBuilder.default())
|
|
|
|
|
|
def test_new_from_group_name():
|
|
context_has_default_values(mango.ContextBuilder.default())
|
|
derived = mango.ContextBuilder.default().new_from_group_name("mainnet.0")
|
|
assert derived.program_id == PublicKey("mv3ekLzLbnVPNxjSKvqBpU3ZeZXPQdEC3bp5MDEBG68")
|
|
assert derived.dex_program_id == PublicKey("9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin")
|
|
|
|
# Should update both of these values on new group name.
|
|
assert derived.group_name == "mainnet.0"
|
|
assert derived.group_id == PublicKey("4yJ2Vx3kZnmHTNCrHzdoj5nCwriF2kVhfKNvqC6gU8tr")
|
|
context_has_default_values(mango.ContextBuilder.default())
|
|
|
|
|
|
def test_new_from_group_id():
|
|
context_has_default_values(mango.ContextBuilder.default())
|
|
derived = mango.ContextBuilder.default().new_from_group_id(PublicKey("4yJ2Vx3kZnmHTNCrHzdoj5nCwriF2kVhfKNvqC6gU8tr"))
|
|
assert derived.program_id == PublicKey("mv3ekLzLbnVPNxjSKvqBpU3ZeZXPQdEC3bp5MDEBG68")
|
|
assert derived.dex_program_id == PublicKey("9xQeWvG816bUx9EPjHmaT23yvVM2ZWbrrpZb9PusVFin")
|
|
|
|
# Should update both of these values on new group ID.
|
|
assert derived.group_name == "mainnet.0"
|
|
assert derived.group_id == PublicKey("4yJ2Vx3kZnmHTNCrHzdoj5nCwriF2kVhfKNvqC6gU8tr")
|
|
context_has_default_values(mango.ContextBuilder.default())
|