mirror of
https://github.com/signalapp/Signal-Server
synced 2026-04-21 06:28:04 +01:00
Add the timestamp using the add method.
This commit is contained in:
committed by
Jon Chambers
parent
5b984d924f
commit
aa10f63d9f
@@ -0,0 +1,31 @@
|
||||
package org.whispersystems.textsecuregcm.filters;
|
||||
|
||||
import org.glassfish.jersey.message.internal.HeaderUtils;
|
||||
import org.junit.Test;
|
||||
import org.whispersystems.textsecuregcm.util.TimestampHeaderUtil;
|
||||
|
||||
import javax.ws.rs.container.ContainerRequestContext;
|
||||
import javax.ws.rs.container.ContainerResponseContext;
|
||||
import javax.ws.rs.core.MultivaluedHashMap;
|
||||
import javax.ws.rs.core.MultivaluedMap;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
public class TimestampResponseFilterTest {
|
||||
|
||||
@Test
|
||||
public void testFilter() {
|
||||
final ContainerRequestContext requestContext = mock(ContainerRequestContext.class);
|
||||
final ContainerResponseContext responseContext = mock(ContainerResponseContext.class);
|
||||
|
||||
final MultivaluedMap<String, Object> headers = HeaderUtils.createOutbound();
|
||||
|
||||
when(responseContext.getHeaders()).thenReturn(headers);
|
||||
|
||||
new TimestampResponseFilter().filter(requestContext, responseContext);
|
||||
|
||||
assertTrue(headers.containsKey(TimestampHeaderUtil.TIMESTAMP_HEADER));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user