when(context.<CompleteBean>getEntityClass()).thenReturn(CompleteBean.class);
when(context.getEntityMeta()).thenReturn(meta);
when(cache.getIfPresent(cacheKeyCaptor.capture())).thenReturn(null);
when(generator.prepareUpdateFields(session, meta, pms, noOptions())).thenReturn(ps);
PreparedStatement actual = manager.getCacheForFieldsUpdate(session, cache, context, pms);
assertThat(actual).isSameAs(ps);
StatementCacheKey cacheKey = cacheKeyCaptor.getValue();
assertThat(cacheKey.<CompleteBean>getEntityClass()).isSameAs(CompleteBean.class);
assertThat(cacheKey.getType()).isEqualTo(CacheType.UPDATE_FIELDS);