// Copyright (C) 2017 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.google.gerrit.server.schema;

import static com.google.common.truth.Truth.assertThat;

import com.google.gerrit.common.TimeUtil;
import com.google.gerrit.reviewdb.client.Account;
import com.google.gerrit.reviewdb.client.AccountGroup;
import com.google.gerrit.reviewdb.client.AccountGroupById;
import com.google.gerrit.reviewdb.client.AccountGroupByIdAud;
import com.google.gerrit.reviewdb.client.AccountGroupMember;
import com.google.gerrit.reviewdb.client.AccountGroupMemberAudit;
import com.google.gerrit.server.schema.GroupBundle.Source;
import com.google.gerrit.testing.GerritBaseTests;
import com.google.gerrit.testing.TestTimeUtil;
import java.sql.Timestamp;
import java.util.TimeZone;
import java.util.concurrent.TimeUnit;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class GroupBundleTest extends GerritBaseTests {
  // This class just contains sanity checks that GroupBundle#compare correctly compares all parts of
  // the bundle. Most other test coverage should come via the slightly more realistic
  // GroupRebuilderTest.

  private static final String TIMEZONE_ID = "US/Eastern";

  private String systemTimeZoneProperty;
  private TimeZone systemTimeZone;
  private Timestamp ts;

  @Before
  public void setUp() {
    systemTimeZoneProperty = System.setProperty("user.timezone", TIMEZONE_ID);
    systemTimeZone = TimeZone.getDefault();
    TimeZone.setDefault(TimeZone.getTimeZone(TIMEZONE_ID));
    TestTimeUtil.resetWithClockStep(1, TimeUnit.SECONDS);
    ts = TimeUtil.nowTs();
  }

  @After
  public void tearDown() {
    TestTimeUtil.useSystemTime();
    System.setProperty("user.timezone", systemTimeZoneProperty);
    TimeZone.setDefault(systemTimeZone);
  }

  @Test
  public void compareNonEqual() throws Exception {
    GroupBundle reviewDbBundle = newBundle().source(Source.REVIEW_DB).build();
    AccountGroup g2 = new AccountGroup(reviewDbBundle.group());
    g2.setDescription("Hello!");
    GroupBundle noteDbBundle = GroupBundle.builder().source(Source.NOTE_DB).group(g2).build();
    assertThat(GroupBundle.compareWithAudits(reviewDbBundle, noteDbBundle))
        .containsExactly(
            "AccountGroups differ\n"
                + ("ReviewDb: AccountGroup{name=group, groupId=1, description=null,"
                    + " visibleToAll=false, groupUUID=group-1, ownerGroupUUID=group-1,"
                    + " createdOn=2009-09-30 17:00:00.0}\n")
                + ("NoteDb  : AccountGroup{name=group, groupId=1, description=Hello!,"
                    + " visibleToAll=false, groupUUID=group-1, ownerGroupUUID=group-1,"
                    + " createdOn=2009-09-30 17:00:00.0}"),
            "AccountGroupMembers differ\n"
                + "ReviewDb: [AccountGroupMember{key=1000,1}]\n"
                + "NoteDb  : []",
            "AccountGroupMemberAudits differ\n"
                + ("ReviewDb: [AccountGroupMemberAudit{key=Key{groupId=1, accountId=1000,"
                    + " addedOn=2009-09-30 17:00:00.0}, addedBy=2000, removedBy=null,"
                    + " removedOn=null}]\n")
                + "NoteDb  : []",
            "AccountGroupByIds differ\n"
                + "ReviewDb: [AccountGroupById{key=1,subgroup}]\n"
                + "NoteDb  : []",
            "AccountGroupByIdAudits differ\n"
                + ("ReviewDb: [AccountGroupByIdAud{key=Key{groupId=1, includeUUID=subgroup,"
                    + " addedOn=2009-09-30 17:00:00.0}, addedBy=3000, removedBy=null,"
                    + " removedOn=null}]\n")
                + "NoteDb  : []");
  }

  @Test
  public void compareIgnoreAudits() throws Exception {
    GroupBundle reviewDbBundle = newBundle().source(Source.REVIEW_DB).build();
    AccountGroup group = new AccountGroup(reviewDbBundle.group());

    AccountGroupMember member =
        new AccountGroupMember(new AccountGroupMember.Key(new Account.Id(1), group.getId()));
    AccountGroupMemberAudit memberAudit =
        new AccountGroupMemberAudit(member, new Account.Id(2), ts);
    AccountGroupById byId =
        new AccountGroupById(
            new AccountGroupById.Key(group.getId(), new AccountGroup.UUID("subgroup-2")));
    AccountGroupByIdAud byIdAudit = new AccountGroupByIdAud(byId, new Account.Id(3), ts);

    GroupBundle noteDbBundle =
        newBundle().source(Source.NOTE_DB).memberAudit(memberAudit).byIdAudit(byIdAudit).build();

    assertThat(GroupBundle.compareWithAudits(reviewDbBundle, noteDbBundle)).isNotEmpty();
    assertThat(GroupBundle.compareWithoutAudits(reviewDbBundle, noteDbBundle)).isEmpty();
  }

  @Test
  public void compareEqual() throws Exception {
    GroupBundle reviewDbBundle = newBundle().source(Source.REVIEW_DB).build();
    GroupBundle noteDbBundle = newBundle().source(Source.NOTE_DB).build();
    assertThat(GroupBundle.compareWithAudits(reviewDbBundle, noteDbBundle)).isEmpty();
  }

  private GroupBundle.Builder newBundle() {
    AccountGroup group =
        new AccountGroup(
            new AccountGroup.NameKey("group"),
            new AccountGroup.Id(1),
            new AccountGroup.UUID("group-1"),
            ts);
    AccountGroupMember member =
        new AccountGroupMember(new AccountGroupMember.Key(new Account.Id(1000), group.getId()));
    AccountGroupMemberAudit memberAudit =
        new AccountGroupMemberAudit(member, new Account.Id(2000), ts);
    AccountGroupById byId =
        new AccountGroupById(
            new AccountGroupById.Key(group.getId(), new AccountGroup.UUID("subgroup")));
    AccountGroupByIdAud byIdAudit = new AccountGroupByIdAud(byId, new Account.Id(3000), ts);
    return GroupBundle.builder()
        .group(group)
        .members(member)
        .memberAudit(memberAudit)
        .byId(byId)
        .byIdAudit(byIdAudit);
  }
}