@Test
public void testSetAdCount() {

  assertThat(state.adGroups[0].count).isEqualTo(C.LENGTH_UNSET);
  state = state.withAdCount(/* adGroupIndex= */ 0, /* adCount= */ 1);
  assertThat(state.adGroups[0].count).isEqualTo(1);
}