/*
 * Copyright (C) 2025 The Android Open Source Project
 *
 * Licensed 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 com.android.server.healthconnect.backuprestore;

import static android.health.HealthFitnessStatsLog.HEALTH_CONNECT_DATA_BACKUP_INVOKED;
import static android.health.HealthFitnessStatsLog.HEALTH_CONNECT_DATA_BACKUP_INVOKED__BACKUP_TYPE__DATA_BACKUP_TYPE_INCREMENTAL;
import static android.health.HealthFitnessStatsLog.HEALTH_CONNECT_DATA_BACKUP_INVOKED__STATUS__DATA_BACKUP_STATUS_ERROR_PARTIAL_BACKUP;
import static android.health.HealthFitnessStatsLog.HEALTH_CONNECT_DATA_RESTORE_INVOKED;
import static android.health.HealthFitnessStatsLog.HEALTH_CONNECT_DATA_RESTORE_INVOKED__STATUS__DATA_RESTORE_STATUS_ERROR_NONE;
import static android.health.HealthFitnessStatsLog.HEALTH_CONNECT_RESTORE_ELIGIBILITY_CHECKED;

import static org.mockito.Mockito.eq;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import android.health.HealthFitnessStatsLog;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

public class BackupRestoreLoggerTest {

    @Rule public final MockitoRule mMockitoRule = MockitoJUnit.rule();

    @Mock private HealthFitnessStatsLog mHealthFitnessStatsLog;

    private BackupRestoreLogger mBackupRestoreLogger;

    @Before
    public void setUp() {
        mBackupRestoreLogger = new BackupRestoreLogger(mHealthFitnessStatsLog);
    }

    @Test
    public void test_logDataBackupStatus() {
        // variables created to comply with java formatter line length
        int statusPartialBackup =
                HEALTH_CONNECT_DATA_BACKUP_INVOKED__STATUS__DATA_BACKUP_STATUS_ERROR_PARTIAL_BACKUP;
        int backupTypeIncremental =
                HEALTH_CONNECT_DATA_BACKUP_INVOKED__BACKUP_TYPE__DATA_BACKUP_TYPE_INCREMENTAL;

        mBackupRestoreLogger.logDataBackupStatus(
                statusPartialBackup, backupTypeIncremental, /*totalRecordCount*/ 50);

        verify(mHealthFitnessStatsLog, times(1))
                .write(
                        eq(HEALTH_CONNECT_DATA_BACKUP_INVOKED),
                        eq(statusPartialBackup),
                        eq(backupTypeIncremental),
                        eq(50));
    }

    @Test
    public void test_logDataRestoreStatus() {
        // variable to comply with java formatter line length
        int statusDataRestore =
                HEALTH_CONNECT_DATA_RESTORE_INVOKED__STATUS__DATA_RESTORE_STATUS_ERROR_NONE;

        mBackupRestoreLogger.logDataRestoreStatus(
                statusDataRestore,
                /*timeToSucceedOrFailMillis*/ 100,
                /*totalRecordCount*/ 50,
                /*successfulRecordCount*/ 45);
        verify(mHealthFitnessStatsLog, times(1))
                .write(
                        eq(HEALTH_CONNECT_DATA_RESTORE_INVOKED),
                        eq(statusDataRestore),
                        eq(100),
                        eq(50),
                        eq(45));
    }

    @Test
    public void test_logRestoreEligibility() {
        mBackupRestoreLogger.logRestoreEligibility(true);
        verify(mHealthFitnessStatsLog, times(1))
                .write(eq(HEALTH_CONNECT_RESTORE_ELIGIBILITY_CHECKED), eq(true));
    }
}