// Copyright (C) 2018 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.googlesource.gerrit.plugins.reviewers;

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

import com.google.common.collect.ImmutableSet;
import com.google.gerrit.acceptance.AbstractDaemonTest;
import com.google.gerrit.acceptance.NoHttpd;
import com.google.gerrit.acceptance.TestAccount;
import com.google.gerrit.reviewdb.client.Account;
import com.google.inject.Inject;
import java.util.Collections;
import java.util.Set;
import org.junit.Before;
import org.junit.Test;

@NoHttpd
public class ReviewersResolverIT extends AbstractDaemonTest {

  @Inject private ReviewersResolver resolver;
  private int change;

  @Before
  public void setUp() {
    change = 1;
  }

  @Test
  public void testUploaderSkippedAsReviewer() throws Exception {
    Set<Account.Id> reviewers =
        resolver.resolve(
            db,
            Collections.singleton(user.email),
            project,
            change,
            gApi.accounts().id(user.id.get()).get());
    assertThat(reviewers).isEmpty();
  }

  @Test
  public void testAccountResolve() throws Exception {
    Set<Account.Id> reviewers =
        resolver.resolve(
            db,
            ImmutableSet.of(user.email, admin.email),
            project,
            change,
            gApi.accounts().id(admin.id.get()).get());
    assertThat(reviewers).containsExactly(user.id);
  }

  @Test
  public void testAccountGroupResolve() throws Exception {
    String group1 = createGroup("group1");
    TestAccount foo = createAccount("foo", group1);
    TestAccount bar = createAccount("bar", group1);

    String group2 = createGroup("group2");
    TestAccount baz = createAccount("baz", group2);
    TestAccount qux = createAccount("qux", group2);

    TestAccount system = createAccount("system", "Administrators");

    Set<Account.Id> reviewers =
        resolver.resolve(
            db,
            ImmutableSet.of(system.email, group1, group2),
            project,
            change,
            gApi.accounts().id(admin.id.get()).get());
    assertThat(reviewers).containsExactly(system.id, foo.id, bar.id, baz.id, qux.id);
  }

  private TestAccount createAccount(String name, String group) throws Exception {
    name = name(name);
    return accounts.create(name, name + "@example.com", name + " full name", group);
  }
}
