// 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.google.gerrit.server.git;

import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.extensions.proto.ProtoTruth.assertThat;
import static com.google.gerrit.server.cache.testing.SerializedClassSubject.assertThatSerializedClass;

import com.google.common.collect.ImmutableMap;
import com.google.gerrit.reviewdb.client.Project;
import com.google.gerrit.server.cache.proto.Cache.TagSetHolderProto;
import com.google.gerrit.testing.GerritBaseTests;
import org.junit.Test;

public class TagSetHolderTest extends GerritBaseTests {
  @Test
  public void serializerWithTagSet() throws Exception {
    TagSetHolder holder = new TagSetHolder(new Project.NameKey("project"));
    holder.setTagSet(new TagSet(holder.getProjectName()));

    byte[] serialized = TagSetHolder.Serializer.INSTANCE.serialize(holder);
    assertThat(TagSetHolderProto.parseFrom(serialized))
        .ignoringRepeatedFieldOrder()
        .isEqualTo(
            TagSetHolderProto.newBuilder()
                .setProjectName("project")
                .setTags(holder.getTagSet().toProto())
                .build());

    TagSetHolder deserialized = TagSetHolder.Serializer.INSTANCE.deserialize(serialized);
    assertThat(deserialized.getProjectName()).isEqualTo(holder.getProjectName());
    TagSetTest.assertEqual(holder.getTagSet(), deserialized.getTagSet());
  }

  @Test
  public void serializerWithoutTagSet() throws Exception {
    TagSetHolder holder = new TagSetHolder(new Project.NameKey("project"));

    byte[] serialized = TagSetHolder.Serializer.INSTANCE.serialize(holder);
    assertThat(TagSetHolderProto.parseFrom(serialized))
        .ignoringRepeatedFieldOrder()
        .isEqualTo(TagSetHolderProto.newBuilder().setProjectName("project").build());

    TagSetHolder deserialized = TagSetHolder.Serializer.INSTANCE.deserialize(serialized);
    assertThat(deserialized.getProjectName()).isEqualTo(holder.getProjectName());
    TagSetTest.assertEqual(holder.getTagSet(), deserialized.getTagSet());
  }

  @Test
  public void fields() {
    assertThatSerializedClass(TagSetHolder.class)
        .hasFields(
            ImmutableMap.of(
                "buildLock", Object.class,
                "projectName", Project.NameKey.class,
                "tags", TagSet.class));
  }
}
