Skip to content

Commit

Permalink
update import paths and add __init__.py to make it testable
Browse files Browse the repository at this point in the history
  • Loading branch information
youngjoon-lee committed Jul 19, 2024
1 parent 3bf199b commit bf41a36
Show file tree
Hide file tree
Showing 19 changed files with 37 additions and 33 deletions.
File renamed without changes.
File renamed without changes.
Empty file.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import blspy

from mixnet.utils import random_bytes
from deprecated.mixnet_v1.utils import random_bytes

BlsPrivateKey: TypeAlias = blspy.PrivateKey
BlsPublicKey: TypeAlias = blspy.G1Element
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@
from contextlib import suppress
from typing import Self

from mixnet.config import MixClientConfig, MixnetTopology
from mixnet.node import PacketQueue
from mixnet.packet import PacketBuilder
from mixnet.poisson import poisson_interval_sec
from deprecated.mixnet_v1.config import MixClientConfig, MixnetTopology
from deprecated.mixnet_v1.node import PacketQueue
from deprecated.mixnet_v1.packet import PacketBuilder
from deprecated.mixnet_v1.poisson import poisson_interval_sec


class MixClient:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
)
from pysphinx.node import Node

from mixnet.bls import BlsPrivateKey, BlsPublicKey
from mixnet.fisheryates import FisherYates
from deprecated.mixnet_v1.bls import BlsPrivateKey, BlsPublicKey
from deprecated.mixnet_v1.fisheryates import FisherYates


@dataclass
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,13 @@
from contextlib import suppress
from typing import Self, TypeAlias

from mixnet.client import MixClient
from mixnet.config import MixnetConfig, MixnetTopology, MixnetTopologyConfig
from mixnet.node import MixNode
from deprecated.mixnet_v1.client import MixClient
from deprecated.mixnet_v1.config import (
MixnetConfig,
MixnetTopology,
MixnetTopologyConfig,
)
from deprecated.mixnet_v1.node import MixNode

EntropyQueue: TypeAlias = "asyncio.Queue[bytes]"

Expand Down
4 changes: 2 additions & 2 deletions deprecated/mixnet-v1/node.py → deprecated/mixnet_v1/node.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,8 @@
UnknownHeaderTypeError,
)

from mixnet.config import MixNodeConfig, NodeAddress
from mixnet.poisson import poisson_interval_sec
from deprecated.mixnet_v1.config import MixNodeConfig, NodeAddress
from deprecated.mixnet_v1.poisson import poisson_interval_sec

PacketQueue: TypeAlias = "asyncio.Queue[Tuple[NodeAddress, SphinxPacket]]"
PacketPayloadQueue: TypeAlias = (
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from pysphinx.payload import Payload
from pysphinx.sphinx import SphinxPacket

from mixnet.config import MixnetTopology, MixNodeInfo
from deprecated.mixnet_v1.config import MixnetTopology, MixNodeInfo


class MessageFlag(Enum):
Expand Down
File renamed without changes.
File renamed without changes
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,13 @@

import numpy

from mixnet.client import MixClient
from mixnet.poisson import poisson_mean_interval_sec
from mixnet.test_utils import (
from deprecated.mixnet_v1.client import MixClient
from deprecated.mixnet_v1.poisson import poisson_mean_interval_sec
from deprecated.mixnet_v1.test_utils import (
init_mixnet_config,
with_test_timeout,
)
from mixnet.utils import random_bytes
from deprecated.mixnet_v1.utils import random_bytes


class TestMixClient(IsolatedAsyncioTestCase):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from unittest import TestCase

from mixnet.fisheryates import FisherYates
from deprecated.mixnet_v1.fisheryates import FisherYates


class TestFisherYates(TestCase):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import asyncio
from unittest import IsolatedAsyncioTestCase

from mixnet.mixnet import Mixnet
from mixnet.test_utils import init_mixnet_config
from deprecated.mixnet_v1.mixnet import Mixnet
from deprecated.mixnet_v1.test_utils import init_mixnet_config


class TestMixnet(IsolatedAsyncioTestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@
import numpy
from pysphinx.sphinx import SphinxPacket

from mixnet.node import MixNode, NodeAddress, PacketQueue
from mixnet.packet import PacketBuilder
from mixnet.poisson import poisson_interval_sec, poisson_mean_interval_sec
from mixnet.test_utils import (
from deprecated.mixnet_v1.node import MixNode, NodeAddress, PacketQueue
from deprecated.mixnet_v1.packet import PacketBuilder
from deprecated.mixnet_v1.poisson import poisson_interval_sec, poisson_mean_interval_sec
from deprecated.mixnet_v1.test_utils import (
init_mixnet_config,
with_test_timeout,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,15 @@

from pysphinx.sphinx import ProcessedFinalHopPacket, SphinxPacket

from mixnet.config import MixNodeInfo
from mixnet.packet import (
from deprecated.mixnet_v1.config import MixNodeInfo
from deprecated.mixnet_v1.packet import (
Fragment,
MessageFlag,
MessageReconstructor,
PacketBuilder,
)
from mixnet.test_utils import init_mixnet_config
from mixnet.utils import random_bytes
from deprecated.mixnet_v1.test_utils import init_mixnet_config
from deprecated.mixnet_v1.utils import random_bytes


class TestPacket(TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,17 @@

from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey

from mixnet.bls import generate_bls
from mixnet.config import (
from deprecated.mixnet_v1.bls import generate_bls
from deprecated.mixnet_v1.config import (
MixClientConfig,
MixNodeConfig,
MixnetConfig,
MixNodeInfo,
MixnetTopology,
MixnetTopologyConfig,
MixnetTopologySize,
MixNodeConfig,
MixNodeInfo,
)
from mixnet.utils import random_bytes
from deprecated.mixnet_v1.utils import random_bytes


def with_test_timeout(t):
Expand Down
File renamed without changes.

0 comments on commit bf41a36

Please sign in to comment.