NotificationsServiceTest.java example

Explorer
uaa-master
package org.cloudfoundry.identity.uaa.login;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import static org.springframework.http.HttpMethod.POST;
import static org.springframework.http.HttpMethod.PUT;
import static org.springframework.test.web.client.match.MockRestRequestMatchers.jsonPath;
import static org.springframework.test.web.client.match.MockRestRequestMatchers.method;
import static org.springframework.test.web.client.match.MockRestRequestMatchers.requestTo;
import static org.springframework.test.web.client.response.MockRestResponseCreators.withBadRequest;
import static org.springframework.test.web.client.response.MockRestResponseCreators.withSuccess;

import org.cloudfoundry.identity.uaa.message.MessageType;
import org.cloudfoundry.identity.uaa.message.NotificationsService;
import org.cloudfoundry.identity.uaa.zone.IdentityZone;
import org.cloudfoundry.identity.uaa.zone.IdentityZoneHolder;
import org.cloudfoundry.identity.uaa.zone.MultitenancyFixture;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;
import org.springframework.test.web.client.MockRestServiceServer;
import org.springframework.web.client.HttpClientErrorException;
import org.springframework.web.client.RestTemplate;

public class NotificationsServiceTest {

    private RestTemplate notificationsTemplate;
    private MockRestServiceServer mockNotificationsServer;
    private RestTemplate uaaTemplate;
    private Map<MessageType, HashMap<String, Object>> notifications;
    private NotificationsService notificationsService;
    private Map<String, Object> response;

    @Before
    public void setUp(){
        notificationsTemplate = new RestTemplate();
        mockNotificationsServer = MockRestServiceServer.createServer(notificationsTemplate);

        uaaTemplate = Mockito.mock(RestTemplate.class);
        notifications = new HashMap<>();
        HashMap<String, Object> passwordResetNotification = new HashMap<>();

        passwordResetNotification.put("id", "kind-id-01");
        passwordResetNotification.put("description", "password reset");
        passwordResetNotification.put("critical", true);
        notifications.put(MessageType.PASSWORD_RESET, passwordResetNotification);

        notificationsService = new NotificationsService(notificationsTemplate, "http://notifications.example.com", notifications, true) {
            @Override
            protected void internalSendMessage(String email, MessageType messageType, String subject, String htmlContent) {
                assertEquals(IdentityZone.getUaa(), IdentityZoneHolder.get());
                super.internalSendMessage(email, messageType, subject, htmlContent);
            }
        };

        response = new HashMap<>();
        List<Map<String, String>> resources = new ArrayList<>();
        Map<String,String> userDetails = new HashMap<>();
        userDetails.put("id", "user-id-01");
        resources.add(userDetails);
        response.put("resources", resources);

        mockNotificationsServer.expect(requestTo("http://notifications.example.com/registration"))
            .andExpect(method(PUT))
            .andExpect(jsonPath("$.source_description").value("CF_Identity"))
            .andExpect(jsonPath("$.kinds[0].id").value("kind-id-01"))
            .andExpect(jsonPath("$.kinds[0].description").value("password reset"))
            .andExpect(jsonPath("$.kinds[0].critical").value(true))
            .andRespond(withSuccess());
    }

    @After
    public void clearZone() {
        IdentityZoneHolder.clear();
    }

    @Test
    public void testSendingMessageToEmailAddress() throws Exception {

        mockNotificationsServer.expect(requestTo("http://notifications.example.com/emails"))
            .andExpect(method(POST))
            .andExpect(jsonPath("$.kind_id").value("kind-id-01"))
            .andExpect(jsonPath("$.to").value("user@example.com"))
            .andExpect(jsonPath("$.subject").value("First message"))
            .andExpect(jsonPath("$.html").value("<p>Message</p>"))
            .andRespond(withSuccess());

        notificationsService.sendMessage("user@example.com", MessageType.PASSWORD_RESET, "First message", "<p>Message</p>");

        mockNotificationsServer.verify();
    }

    @Test
    public void testSendingMessageInAnotherZoneResets() throws Exception {
        IdentityZone zone = MultitenancyFixture.identityZone("zone", "zone");
        IdentityZoneHolder.set(zone);

        mockNotificationsServer.expect(requestTo("http://notifications.example.com/emails"))
            .andExpect(method(POST))
            .andExpect(jsonPath("$.to").value("user@example.com"))
            .andExpect(jsonPath("$.subject").value("First message"))
            .andExpect(jsonPath("$.html").value("<p>Message</p>"))
            .andRespond(withSuccess());

        notificationsService.sendMessage("user@example.com", MessageType.PASSWORD_RESET, "First message", "<p>Message</p>");

        mockNotificationsServer.verify();
        assertSame(zone, IdentityZoneHolder.get());
    }

    @Test
    public void testSendingMessageInAnotherZoneResetsWhenError() throws Exception {
        IdentityZone zone = MultitenancyFixture.identityZone("zone", "zone");
        IdentityZoneHolder.set(zone);

        mockNotificationsServer.expect(requestTo("http://notifications.example.com/emails"))
            .andExpect(method(POST))
            .andExpect(jsonPath("$.to").value("user@example.com"))
            .andExpect(jsonPath("$.subject").value("First message"))
            .andExpect(jsonPath("$.html").value("<p>Message</p>"))
            .andRespond(withBadRequest());
        try {
            notificationsService.sendMessage("user@example.com", MessageType.PASSWORD_RESET, "First message", "<p>Message</p>");
            fail();
        } catch (HttpClientErrorException x) {
        }

        mockNotificationsServer.verify();
        assertSame(zone, IdentityZoneHolder.get());
    }
}