/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.flink.runtime.checkpoint;

import org.apache.flink.core.testutils.CommonTestUtils;
import org.junit.Test;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class CheckpointStatsSnapshotTest {

	/**
	 * Tests that the snapshot is actually serializable.
	 */
	@Test
	public void testIsJavaSerializable() throws Exception {
		CheckpointStatsCounts counts = new CheckpointStatsCounts();
		counts.incrementInProgressCheckpoints();
		counts.incrementInProgressCheckpoints();
		counts.incrementInProgressCheckpoints();
		counts.incrementCompletedCheckpoints();
		counts.incrementFailedCheckpoints();
		counts.incrementRestoredCheckpoints();

		CompletedCheckpointStatsSummary summary = new CompletedCheckpointStatsSummary();
		summary.updateSummary(createCompletedCheckpointsStats(12398, 9919, 12));
		summary.updateSummary(createCompletedCheckpointsStats(2221, 3333, 9122));

		CheckpointStatsHistory history = new CheckpointStatsHistory(1);
		RestoredCheckpointStats restored = new RestoredCheckpointStats(1, CheckpointProperties.forStandardCheckpoint(), 99119, null);

		CheckpointStatsSnapshot snapshot = new CheckpointStatsSnapshot(
			counts,
			summary,
			history,
			restored);

		CheckpointStatsSnapshot copy = CommonTestUtils.createCopySerializable(snapshot);

		assertEquals(counts.getNumberOfCompletedCheckpoints(), copy.getCounts().getNumberOfCompletedCheckpoints());
		assertEquals(counts.getNumberOfFailedCheckpoints(), copy.getCounts().getNumberOfFailedCheckpoints());
		assertEquals(counts.getNumberOfInProgressCheckpoints(), copy.getCounts().getNumberOfInProgressCheckpoints());
		assertEquals(counts.getNumberOfRestoredCheckpoints(), copy.getCounts().getNumberOfRestoredCheckpoints());
		assertEquals(counts.getTotalNumberOfCheckpoints(), copy.getCounts().getTotalNumberOfCheckpoints());

		assertEquals(summary.getStateSizeStats().getSum(), copy.getSummaryStats().getStateSizeStats().getSum());
		assertEquals(summary.getEndToEndDurationStats().getSum(), copy.getSummaryStats().getEndToEndDurationStats().getSum());
		assertEquals(summary.getAlignmentBufferedStats().getSum(), copy.getSummaryStats().getAlignmentBufferedStats().getSum());

		assertEquals(restored.getCheckpointId(), copy.getLatestRestoredCheckpoint().getCheckpointId());
	}

	private CompletedCheckpointStats createCompletedCheckpointsStats(
			long stateSize,
			long endToEndDuration,
			long alignmentBuffered) {

		CompletedCheckpointStats completed = mock(CompletedCheckpointStats.class);
		when(completed.getStateSize()).thenReturn(stateSize);
		when(completed.getEndToEndDuration()).thenReturn(endToEndDuration);
		when(completed.getAlignmentBuffered()).thenReturn(alignmentBuffered);

		return completed;
	}

}