import pytest from flask_atlas_auth.claims import registered_claims from .helpers import TestClaim class TestAuthClaims: def test_registered_claims(self): assert TestClaim in registered_claims def test_id(self): claim = TestClaim(TestClaim.Values.test_value1) assert claim.id == (claim.path, claim.value) def test_eq(self): claim1 = TestClaim(TestClaim.Values.test_value1) claim2 = TestClaim(TestClaim.Values.test_value1) assert claim1 == claim2 claim1 = TestClaim(TestClaim.Values.test_value1) claim2 = TestClaim(TestClaim.Values.test_value2) assert claim1 != claim2 def test_hash(self): claim = TestClaim(TestClaim.Values.test_value1) assert hash(claim) == hash(claim.id) def test_from_dict_success(self): claim = TestClaim.from_dict( {TestClaim.value_field: TestClaim.Values.test_value1.value} ) assert claim.value == TestClaim.Values.test_value1 def test_from_dict_failure(self): with pytest.raises(ValueError): TestClaim.from_dict({TestClaim.value_field: "test_wrong_value"})