def test_that_when_set_identity_pool_roles_with_only_unauth_role_specified_the_set_identity_pool_roles_method_only_set_the_unauth_role( self, ): """ Tests that given a valid pool id, and only other given role is the UnauthenticatedRole, the unauth role for the pool is set and the auth role is cleared. """ self.conn.set_identity_pool_roles.return_value = None with patch.dict( boto_cognitoidentity.__salt__, { "boto_iam.describe_role": MagicMock( return_value={"arn": "my_unauth_role_arn"} ) }, ): expected_roles = dict(unauthenticated="my_unauth_role_arn") result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", UnauthenticatedRole="my_unauth_role", **conn_parameters ) mock_calls = self.conn.mock_calls self.assertTrue(result.get("set")) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get("Roles"), expected_roles)
def test_that_when_set_identity_pool_roles_with_both_roles_specified_the_set_identity_pool_role_method_set_both_roles( self, ): """ Tests setting of both roles to valid given roles """ self.conn.set_identity_pool_roles.return_value = None with patch.dict( boto_cognitoidentity.__salt__, { "boto_iam.describe_role": MagicMock( return_value={"arn": "my_unauth_role_arn"} ) }, ): expected_roles = dict( authenticated="arn:aws:iam:my_auth_role", unauthenticated="my_unauth_role_arn", ) result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", AuthenticatedRole="arn:aws:iam:my_auth_role", UnauthenticatedRole="my_unauth_role", **conn_parameters ) mock_calls = self.conn.mock_calls self.assertTrue(result.get("set")) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get("Roles"), expected_roles)
def test_that_when_set_identity_pool_roles_given_invalid_auth_role_the_set_identity_pool_method_returns_set_false_and_error(boto_conn): ''' Tests error handling for invalid auth role ''' with patch.dict(boto_cognitoidentity.__salt__, {'boto_iam.describe_role': MagicMock(return_value=False)}): result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', AuthenticatedRole='no_such_auth_role', **pytest.conn_parameters) mock_calls = boto_conn.mock_calls assert result.get('set') is False assert 'no_such_auth_role' in result.get('error', '') assert len(mock_calls) == 0
def test_that_when_set_identity_pool_roles_with_no_roles_specified_the_set_identity_pool_roles_method_unset_the_roles(boto_conn): ''' Tests that given a valid pool id, and no other roles given, the role for the pool is cleared. ''' boto_conn.set_identity_pool_roles.return_value = None result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', **pytest.conn_parameters) mock_calls = boto_conn.mock_calls assert result.get('set') is True assert len(mock_calls) == 1 assert mock_calls[0][2].get('Roles') == {}
def test_that_when_set_identity_pool_roles_with_invalid_pool_id_the_set_identity_pool_roles_method_returns_set_false_and_error(boto_conn): ''' Tests that given an invalid pool id, we properly handle error generated from set_identity_pool_roles ''' boto_conn.set_identity_pool_roles.side_effect = exceptions.ClientError(error_content, 'set_identity_pool_roles') result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='no_such_pool_id', **pytest.conn_parameters) mock_calls = boto_conn.mock_calls assert result.get('set') is False assert result.get('error', {}).get('message') == error_message.format('set_identity_pool_roles') assert len(mock_calls) == 1
def test_that_when_set_identity_pool_roles_with_invalid_pool_id_the_set_identity_pool_roles_method_returns_set_false_and_error(self): ''' Tests that given an invalid pool id, we properly handle error generated from set_identity_pool_roles ''' self.conn.set_identity_pool_roles.side_effect = ClientError(error_content, 'set_identity_pool_roles') result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='no_such_pool_id', **conn_parameters) mock_calls = self.conn.mock_calls self.assertIs(result.get('set'), False) self.assertEqual(result.get('error', {}).get('message'), error_message.format('set_identity_pool_roles')) self.assertEqual(len(mock_calls), 1)
def test_that_when_set_identity_pool_roles_with_no_roles_specified_the_set_identity_pool_roles_method_unset_the_roles(self): ''' Tests that given a valid pool id, and no other roles given, the role for the pool is cleared. ''' self.conn.set_identity_pool_roles.return_value = None result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', **conn_parameters) mock_calls = self.conn.mock_calls self.assertTrue(result.get('set')) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get('Roles'), {})
def test_that_when_set_identity_pool_roles_given_invalid_unauth_role_the_set_identity_pool_method_returns_set_false_and_error(self): ''' Tests error handling for invalid unauth role ''' with patch.dict(boto_cognitoidentity.__salt__, {'boto_iam.describe_role': MagicMock(return_value=False)}): result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', AuthenticatedRole='arn:aws:iam:my_auth_role', UnauthenticatedRole='no_such_unauth_role', **conn_parameters) mock_calls = self.conn.mock_calls self.assertIs(result.get('set'), False) self.assertIn('no_such_unauth_role', result.get('error', '')) self.assertEqual(len(mock_calls), 0)
def test_that_when_set_identity_pool_roles_with_invalid_pool_id_the_set_identity_pool_roles_method_returns_set_false_and_error( self ): """ Tests that given an invalid pool id, we properly handle error generated from set_identity_pool_roles """ self.conn.set_identity_pool_roles.side_effect = ClientError(error_content, "set_identity_pool_roles") result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId="no_such_pool_id", **conn_parameters) mock_calls = self.conn.mock_calls self.assertIs(result.get("set"), False) self.assertEqual(result.get("error", {}).get("message"), error_message.format("set_identity_pool_roles")) self.assertEqual(len(mock_calls), 1)
def test_that_when_set_identity_pool_roles_with_no_roles_specified_the_set_identity_pool_roles_method_unset_the_roles( self ): """ Tests that given a valid pool id, and no other roles given, the role for the pool is cleared. """ self.conn.set_identity_pool_roles.return_value = None result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId="some_id", **conn_parameters) mock_calls = self.conn.mock_calls self.assertTrue(result.get("set")) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get("Roles"), {})
def test_that_when_set_identity_pool_roles_with_no_roles_specified_the_set_identity_pool_roles_method_unset_the_roles( self, ): """ Tests that given a valid pool id, and no other roles given, the role for the pool is cleared. """ self.conn.set_identity_pool_roles.return_value = None result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", **conn_parameters) mock_calls = self.conn.mock_calls assert result.get("set") assert len(mock_calls) == 1 assert mock_calls[0][2].get("Roles") == {}
def test_that_when_set_identity_pool_roles_with_both_roles_specified_the_set_identity_pool_role_method_set_both_roles(self): ''' Tests setting of both roles to valid given roles ''' self.conn.set_identity_pool_roles.return_value = None with patch.dict(boto_cognitoidentity.__salt__, {'boto_iam.describe_role': MagicMock(return_value={'arn': 'my_unauth_role_arn'})}): expected_roles = dict(authenticated='arn:aws:iam:my_auth_role', unauthenticated='my_unauth_role_arn') result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', AuthenticatedRole='arn:aws:iam:my_auth_role', UnauthenticatedRole='my_unauth_role', **conn_parameters) mock_calls = self.conn.mock_calls self.assertTrue(result.get('set')) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get('Roles'), expected_roles)
def test_that_when_set_identity_pool_roles_with_only_unauth_role_specified_the_set_identity_pool_roles_method_only_set_the_unauth_role(self): ''' Tests that given a valid pool id, and only other given role is the UnauthenticatedRole, the unauth role for the pool is set and the auth role is cleared. ''' self.conn.set_identity_pool_roles.return_value = None with patch.dict(boto_cognitoidentity.__salt__, {'boto_iam.describe_role': MagicMock(return_value={'arn': 'my_unauth_role_arn'})}): expected_roles = dict(unauthenticated='my_unauth_role_arn') result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', UnauthenticatedRole='my_unauth_role', **conn_parameters) mock_calls = self.conn.mock_calls self.assertTrue(result.get('set')) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get('Roles'), expected_roles)
def test_that_when_set_identity_pool_roles_with_only_auth_role_specified_the_set_identity_pool_roles_method_only_set_the_auth_role(boto_conn): ''' Tests that given a valid pool id, and only other given role is the AuthenticatedRole, the auth role for the pool is set and unauth role is cleared. ''' boto_conn.set_identity_pool_roles.return_value = None with patch.dict(boto_cognitoidentity.__salt__, {'boto_iam.describe_role': MagicMock(return_value={'arn': 'my_auth_role_arn'})}): expected_roles = dict(authenticated='my_auth_role_arn') result = boto_cognitoidentity.set_identity_pool_roles(IdentityPoolId='some_id', AuthenticatedRole='my_auth_role', **pytest.conn_parameters) mock_calls = boto_conn.mock_calls assert result.get('set') is True assert len(mock_calls) == 1 assert mock_calls[0][2].get('Roles') == expected_roles
def test_that_when_set_identity_pool_roles_with_invalid_pool_id_the_set_identity_pool_roles_method_returns_set_false_and_error( self, ): """ Tests that given an invalid pool id, we properly handle error generated from set_identity_pool_roles """ self.conn.set_identity_pool_roles.side_effect = ClientError( error_content, "set_identity_pool_roles") result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="no_such_pool_id", **conn_parameters) mock_calls = self.conn.mock_calls assert result.get("set") is False assert result.get("error", {}).get("message") == error_message.format( "set_identity_pool_roles") assert len(mock_calls) == 1
def test_that_when_set_identity_pool_roles_given_invalid_auth_role_the_set_identity_pool_method_returns_set_false_and_error( self ): """ Tests error handling for invalid auth role """ with patch.dict(boto_cognitoidentity.__salt__, {"boto_iam.describe_role": MagicMock(return_value=False)}): result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", AuthenticatedRole="no_such_auth_role", **conn_parameters ) mock_calls = self.conn.mock_calls self.assertIs(result.get("set"), False) self.assertIn("no_such_auth_role", result.get("error", "")) self.assertEqual(len(mock_calls), 0)
def test_that_when_set_identity_pool_roles_given_invalid_auth_role_the_set_identity_pool_method_returns_set_false_and_error( self, ): """ Tests error handling for invalid auth role """ with patch.dict( boto_cognitoidentity.__salt__, {"boto_iam.describe_role": MagicMock(return_value=False)}, ): result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", AuthenticatedRole="no_such_auth_role", **conn_parameters) mock_calls = self.conn.mock_calls self.assertIs(result.get("set"), False) self.assertIn("no_such_auth_role", result.get("error", "")) self.assertEqual(len(mock_calls), 0)
def test_that_when_set_identity_pool_roles_given_invalid_unauth_role_the_set_identity_pool_method_returns_set_false_and_error( self, ): """ Tests error handling for invalid unauth role """ with patch.dict( boto_cognitoidentity.__salt__, {"boto_iam.describe_role": MagicMock(return_value=False)}, ): result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", AuthenticatedRole="arn:aws:iam:my_auth_role", UnauthenticatedRole="no_such_unauth_role", **conn_parameters) mock_calls = self.conn.mock_calls assert result.get("set") is False assert "no_such_unauth_role" in result.get("error", "") assert len(mock_calls) == 0
def test_that_when_set_identity_pool_roles_with_only_unauth_role_specified_the_set_identity_pool_roles_method_only_set_the_unauth_role( self ): """ Tests that given a valid pool id, and only other given role is the UnauthenticatedRole, the unauth role for the pool is set and the auth role is cleared. """ self.conn.set_identity_pool_roles.return_value = None with patch.dict( boto_cognitoidentity.__salt__, {"boto_iam.describe_role": MagicMock(return_value={"arn": "my_unauth_role_arn"})}, ): expected_roles = dict(unauthenticated="my_unauth_role_arn") result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", UnauthenticatedRole="my_unauth_role", **conn_parameters ) mock_calls = self.conn.mock_calls self.assertTrue(result.get("set")) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get("Roles"), expected_roles)
def test_that_when_set_identity_pool_roles_with_both_roles_specified_the_set_identity_pool_role_method_set_both_roles( self ): """ Tests setting of both roles to valid given roles """ self.conn.set_identity_pool_roles.return_value = None with patch.dict( boto_cognitoidentity.__salt__, {"boto_iam.describe_role": MagicMock(return_value={"arn": "my_unauth_role_arn"})}, ): expected_roles = dict(authenticated="arn:aws:iam:my_auth_role", unauthenticated="my_unauth_role_arn") result = boto_cognitoidentity.set_identity_pool_roles( IdentityPoolId="some_id", AuthenticatedRole="arn:aws:iam:my_auth_role", UnauthenticatedRole="my_unauth_role", **conn_parameters ) mock_calls = self.conn.mock_calls self.assertTrue(result.get("set")) self.assertEqual(len(mock_calls), 1) self.assertEqual(mock_calls[0][2].get("Roles"), expected_roles)