142 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			142 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import boto3
 | 
						|
import pytest
 | 
						|
import sure  # noqa # pylint: disable=unused-import
 | 
						|
 | 
						|
from botocore.exceptions import ClientError
 | 
						|
from moto import mock_managedblockchain
 | 
						|
from . import helpers
 | 
						|
 | 
						|
 | 
						|
@mock_managedblockchain
 | 
						|
def test_create_2_invitations():
 | 
						|
    conn = boto3.client("managedblockchain", region_name="us-east-1")
 | 
						|
 | 
						|
    # Create network
 | 
						|
    response = conn.create_network(
 | 
						|
        Name="testnetwork1",
 | 
						|
        Framework="HYPERLEDGER_FABRIC",
 | 
						|
        FrameworkVersion="1.2",
 | 
						|
        FrameworkConfiguration=helpers.default_frameworkconfiguration,
 | 
						|
        VotingPolicy=helpers.default_votingpolicy,
 | 
						|
        MemberConfiguration=helpers.default_memberconfiguration,
 | 
						|
    )
 | 
						|
    network_id = response["NetworkId"]
 | 
						|
    member_id = response["MemberId"]
 | 
						|
 | 
						|
    # Create proposal
 | 
						|
    response = conn.create_proposal(
 | 
						|
        NetworkId=network_id,
 | 
						|
        MemberId=member_id,
 | 
						|
        Actions=helpers.multiple_policy_actions,
 | 
						|
    )
 | 
						|
    proposal_id = response["ProposalId"]
 | 
						|
 | 
						|
    # Get proposal details
 | 
						|
    response = conn.get_proposal(NetworkId=network_id, ProposalId=proposal_id)
 | 
						|
    response["Proposal"]["NetworkId"].should.equal(network_id)
 | 
						|
    response["Proposal"]["Status"].should.equal("IN_PROGRESS")
 | 
						|
 | 
						|
    # Vote yes
 | 
						|
    response = conn.vote_on_proposal(
 | 
						|
        NetworkId=network_id,
 | 
						|
        ProposalId=proposal_id,
 | 
						|
        VoterMemberId=member_id,
 | 
						|
        Vote="YES",
 | 
						|
    )
 | 
						|
 | 
						|
    # Get the invitation
 | 
						|
    response = conn.list_invitations()
 | 
						|
    response["Invitations"].should.have.length_of(2)
 | 
						|
    response["Invitations"][0]["NetworkSummary"]["Id"].should.equal(network_id)
 | 
						|
    response["Invitations"][0]["Status"].should.equal("PENDING")
 | 
						|
    response["Invitations"][1]["NetworkSummary"]["Id"].should.equal(network_id)
 | 
						|
    response["Invitations"][1]["Status"].should.equal("PENDING")
 | 
						|
 | 
						|
 | 
						|
@mock_managedblockchain
 | 
						|
def test_reject_invitation():
 | 
						|
    conn = boto3.client("managedblockchain", region_name="us-east-1")
 | 
						|
 | 
						|
    # Create network
 | 
						|
    response = conn.create_network(
 | 
						|
        Name="testnetwork1",
 | 
						|
        Framework="HYPERLEDGER_FABRIC",
 | 
						|
        FrameworkVersion="1.2",
 | 
						|
        FrameworkConfiguration=helpers.default_frameworkconfiguration,
 | 
						|
        VotingPolicy=helpers.default_votingpolicy,
 | 
						|
        MemberConfiguration=helpers.default_memberconfiguration,
 | 
						|
    )
 | 
						|
    network_id = response["NetworkId"]
 | 
						|
    member_id = response["MemberId"]
 | 
						|
 | 
						|
    # Create proposal
 | 
						|
    response = conn.create_proposal(
 | 
						|
        NetworkId=network_id, MemberId=member_id, Actions=helpers.default_policy_actions
 | 
						|
    )
 | 
						|
    proposal_id = response["ProposalId"]
 | 
						|
 | 
						|
    # Get proposal details
 | 
						|
    response = conn.get_proposal(NetworkId=network_id, ProposalId=proposal_id)
 | 
						|
    response["Proposal"]["NetworkId"].should.equal(network_id)
 | 
						|
    response["Proposal"]["Status"].should.equal("IN_PROGRESS")
 | 
						|
 | 
						|
    # Vote yes
 | 
						|
    response = conn.vote_on_proposal(
 | 
						|
        NetworkId=network_id,
 | 
						|
        ProposalId=proposal_id,
 | 
						|
        VoterMemberId=member_id,
 | 
						|
        Vote="YES",
 | 
						|
    )
 | 
						|
 | 
						|
    # Get the invitation
 | 
						|
    response = conn.list_invitations()
 | 
						|
    response["Invitations"][0]["NetworkSummary"]["Id"].should.equal(network_id)
 | 
						|
    response["Invitations"][0]["Status"].should.equal("PENDING")
 | 
						|
    invitation_id = response["Invitations"][0]["InvitationId"]
 | 
						|
 | 
						|
    # Reject - thanks but no thanks
 | 
						|
    response = conn.reject_invitation(InvitationId=invitation_id)
 | 
						|
 | 
						|
    # Check the invitation status
 | 
						|
    response = conn.list_invitations()
 | 
						|
    response["Invitations"][0]["InvitationId"].should.equal(invitation_id)
 | 
						|
    response["Invitations"][0]["Status"].should.equal("REJECTED")
 | 
						|
 | 
						|
 | 
						|
@mock_managedblockchain
 | 
						|
def test_reject_invitation_badinvitation():
 | 
						|
    conn = boto3.client("managedblockchain", region_name="us-east-1")
 | 
						|
 | 
						|
    # Create network - need a good network
 | 
						|
    response = conn.create_network(
 | 
						|
        Name="testnetwork1",
 | 
						|
        Framework="HYPERLEDGER_FABRIC",
 | 
						|
        FrameworkVersion="1.2",
 | 
						|
        FrameworkConfiguration=helpers.default_frameworkconfiguration,
 | 
						|
        VotingPolicy=helpers.default_votingpolicy,
 | 
						|
        MemberConfiguration=helpers.default_memberconfiguration,
 | 
						|
    )
 | 
						|
    network_id = response["NetworkId"]
 | 
						|
    member_id = response["MemberId"]
 | 
						|
 | 
						|
    response = conn.create_proposal(
 | 
						|
        NetworkId=network_id, MemberId=member_id, Actions=helpers.default_policy_actions
 | 
						|
    )
 | 
						|
 | 
						|
    proposal_id = response["ProposalId"]
 | 
						|
 | 
						|
    response = conn.vote_on_proposal(
 | 
						|
        NetworkId=network_id,
 | 
						|
        ProposalId=proposal_id,
 | 
						|
        VoterMemberId=member_id,
 | 
						|
        Vote="YES",
 | 
						|
    )
 | 
						|
 | 
						|
    with pytest.raises(ClientError) as ex:
 | 
						|
        conn.reject_invitation(InvitationId="in-ABCDEFGHIJKLMNOP0123456789")
 | 
						|
    err = ex.value.response["Error"]
 | 
						|
    err["Code"].should.equal("ResourceNotFoundException")
 | 
						|
    err["Message"].should.contain(
 | 
						|
        "InvitationId in-ABCDEFGHIJKLMNOP0123456789 not found."
 | 
						|
    )
 |