Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions crypto/configuration/network.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from datetime import datetime
from typing import Type, TypedDict, Union
from crypto.networks.devnet import Devnet
from crypto.networks.mainnet import Mainnet
from crypto.networks.testnet import Testnet

Expand All @@ -10,12 +9,12 @@ class NetworkType(TypedDict):
wif: int

network: NetworkType = {
'epoch': Devnet.epoch,
'version': Devnet.version,
'wif': Devnet.wif,
'epoch': Testnet.epoch,
'version': Testnet.version,
'wif': Testnet.wif,
}

def set_network(network_object: Union[Type[Mainnet], Type[Devnet], Type[Testnet]]) -> None:
def set_network(network_object: Union[Type[Mainnet], Type[Testnet]]) -> None:
"""Set what network you want to use in the crypto library

Args:
Expand Down
6 changes: 0 additions & 6 deletions crypto/networks/devnet.py

This file was deleted.

4 changes: 2 additions & 2 deletions crypto/networks/mainnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,5 +2,5 @@

class Mainnet(object):
epoch = datetime(2017, 3, 21, 13, 00, 00)
version = 23
wif = 170
version = 30
wif = 186
2 changes: 1 addition & 1 deletion crypto/networks/testnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,5 +2,5 @@

class Testnet(object):
epoch = datetime(2017, 3, 21, 13, 00, 00)
version = 23
version = 30
wif = 186
19 changes: 11 additions & 8 deletions tests/configuration/test_network.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from datetime import datetime

from crypto.configuration.network import get_network, set_custom_network, set_network
from crypto.networks.devnet import Devnet
from crypto.networks.testnet import Testnet
from crypto.networks.mainnet import Mainnet
from crypto.networks.testnet import Testnet

Expand All @@ -12,17 +12,19 @@ def test_get_network():


def test_set_network():
# test main net
# mainnet
set_network(Mainnet)
result = get_network()
assert result['version'] == 23
assert result['wif'] == 170
# test test net
assert result['version'] == 30
assert result['wif'] == 186

# testnet
set_network(Testnet)
result = get_network()
assert result['version'] == 23
assert result['version'] == 30
assert result['wif'] == 186
set_network(Devnet) # set back to devnet so other tests don't fail

set_network(Testnet) # set back to Testnet so other tests don't fail


def test_set_custom_network():
Expand All @@ -32,4 +34,5 @@ def test_set_custom_network():
assert result['version'] == 11
assert result['wif'] == 130
assert result['epoch'] == epoch_time
set_network(Devnet) # set back to devnet so other tests don't fail

set_network(Testnet) # set back to Testnet so other tests don't fail
6 changes: 3 additions & 3 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,16 @@
import json
import os
from crypto.configuration.network import set_network
from crypto.networks.devnet import Devnet
from crypto.networks.testnet import Testnet


@pytest.fixture(scope='session', autouse=True)
def configure_network():
"""
Configures the network to Devnet before running any tests.
Configures the network to Testnet before running any tests.
This fixture runs automatically once per test session.
"""
set_network(Devnet)
set_network(Testnet)

@pytest.fixture
def load_transaction_fixture():
Expand Down