@SuppressWarnings("unchecked")
@Test
public void testLookupValueSetCategories() {

	// Arrange
	ValueSetQueryDto queryMock = mock(ValueSetQueryDto.class);
	@SuppressWarnings("rawtypes")
	Set valueSetCategorySetMock = mock(Set.class);
	when(
			restTemplateMock.getForObject(eq(EXPECTED_REST_URL),
					eq(ValueSetQueryDto.class), isA(Map.class)))
			.thenReturn(queryMock);
	when(queryMock.getVsCategoryCodes())
			.thenReturn(valueSetCategorySetMock);
	// Act
	@SuppressWarnings("rawtypes")
	Set response = sut
			.lookupValueSetCategories(CODE_MOCK, CODE_SYSTEM_MOCK);
	// Assert
	assertEquals(valueSetCategorySetMock, response);
}