/*
 * Copyright 2016-present Facebook, Inc.
 *
 * 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.facebook.buck.intellij.ideabuck.autodeps;

import static com.facebook.buck.intellij.ideabuck.test.TestUtil.buckFile;
import static org.junit.Assert.assertEquals;

import org.junit.Test;

public class MaybeAddVisibilityToTargetTest {

  @Test
  public void doesNothingWhenTargetIsIncluded() {
    String buckInput =
        buckFile(
            "# Comment",
            "rule(",
            "\tname = 'foo',",
            "\tvisibility = [",
            "\t\t'/this',",
            "\t]",
            ")");
    String expected = buckInput;
    String actual = BuckDeps.maybeAddVisibilityToTarget(buckInput, "/this", "foo");
    assertEquals(expected, actual);
  }

  @Test
  public void addsVisibilityWhenTargetIsAbsent() {
    String buckInput =
        buckFile(
            "# Comment",
            "rule(",
            "\tname = 'foo',",
            "\tvisibility = [",
            "\t\t'/this',",
            "\t]",
            ")");
    String expected =
        buckFile(
            "# Comment",
            "rule(",
            "\tname = 'foo',",
            "\tvisibility = [",
            "\t\t'/other:thing',",
            "\t\t'/this',",
            "\t]",
            ")");
    String actual = BuckDeps.maybeAddVisibilityToTarget(buckInput, "/other:thing", "foo");
    assertEquals(expected, actual);
  }
}