Skip to content

Commit

Permalink
up-41: add schema tests
Browse files Browse the repository at this point in the history
Signed-off-by: toktar <renata.toktar@evernym.com>
  • Loading branch information
toktar committed Feb 19, 2021
1 parent c5c1c28 commit 859e8a6
Show file tree
Hide file tree
Showing 4 changed files with 65 additions and 6 deletions.
20 changes: 20 additions & 0 deletions indy_common/test/types/test_get_fee_schema.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from collections import OrderedDict

from indy_common.types import ClientGetFeeOperation
from plenum.common.messages.fields import ConstantField, LimitedLengthStringField

EXPECTED_ORDERED_FIELDS = OrderedDict([
("type", ConstantField),
("alias", LimitedLengthStringField),
])


def test_has_expected_fields():
actual_field_names = OrderedDict(ClientGetFeeOperation.schema).keys()
assert actual_field_names == EXPECTED_ORDERED_FIELDS.keys()


def test_has_expected_validators():
schema = dict(ClientGetFeeOperation.schema)
for field, validator in EXPECTED_ORDERED_FIELDS.items():
assert isinstance(schema[field], validator)
19 changes: 19 additions & 0 deletions indy_common/test/types/test_get_fees_schema.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
from collections import OrderedDict

from indy_common.types import ClientGetFeesOperation
from plenum.common.messages.fields import ConstantField

EXPECTED_ORDERED_FIELDS = OrderedDict([
("type", ConstantField),
])


def test_has_expected_fields():
actual_field_names = OrderedDict(ClientGetFeesOperation.schema).keys()
assert actual_field_names == EXPECTED_ORDERED_FIELDS.keys()


def test_has_expected_validators():
schema = dict(ClientGetFeesOperation.schema)
for field, validator in EXPECTED_ORDERED_FIELDS.items():
assert isinstance(schema[field], validator)
20 changes: 20 additions & 0 deletions indy_common/test/types/test_set_fees_schema.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
from collections import OrderedDict

from indy_common.types import ClientSetFeesOperation, SetFeesField
from plenum.common.messages.fields import ConstantField

EXPECTED_ORDERED_FIELDS = OrderedDict([
("type", ConstantField),
("fees", SetFeesField),
])


def test_has_expected_fields():
actual_field_names = OrderedDict(ClientSetFeesOperation.schema).keys()
assert actual_field_names == EXPECTED_ORDERED_FIELDS.keys()


def test_has_expected_validators():
schema = dict(ClientSetFeesOperation.schema)
for field, validator in EXPECTED_ORDERED_FIELDS.items():
assert isinstance(schema[field], validator)
12 changes: 6 additions & 6 deletions indy_common/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -467,14 +467,14 @@ class ClientGetRichSchemaObjectByMetadataOperation(MessageValidator):
)


class ClientGetFeeMsgOperation(MessageValidator):
class ClientGetFeeOperation(MessageValidator):
schema = (
(TXN_TYPE, ConstantField(GET_FEE)),
(FEES_ALIAS, LimitedLengthStringField(max_length=FEE_ALIAS_LENGTH)),
)


class ClientGetFeesMsgOperation(MessageValidator):
class ClientGetFeesOperation(MessageValidator):
schema = (
(TXN_TYPE, ConstantField(GET_FEES)),
)
Expand All @@ -491,7 +491,7 @@ def _specific_validation(self, val):
return "set_fees -- " + error


class ClientSetFeesMsgOperation(MessageValidator):
class ClientSetFeesOperation(MessageValidator):
schema = (
(TXN_TYPE, ConstantField(SET_FEES)),
(FEES, SetFeesField()),
Expand Down Expand Up @@ -528,9 +528,9 @@ class ClientOperationField(PClientOperationField):
RICH_SCHEMA_PRES_DEF: ClientRichSchemaPresDefOperation(),
GET_RICH_SCHEMA_OBJECT_BY_ID: ClientGetRichSchemaObjectByIdOperation(),
GET_RICH_SCHEMA_OBJECT_BY_METADATA: ClientGetRichSchemaObjectByMetadataOperation(),
GET_FEE: ClientGetFeeMsgOperation(),
GET_FEES: ClientGetFeesMsgOperation(),
SET_FEES: ClientSetFeesMsgOperation(),
GET_FEE: ClientGetFeeOperation(),
GET_FEES: ClientGetFeesOperation(),
SET_FEES: ClientSetFeesOperation(),
}

# TODO: it is a workaround because INDY-338, `operations` must be a class
Expand Down

0 comments on commit 859e8a6

Please sign in to comment.