| /* |
| * Copyright 2014-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.python; |
| |
| import static org.junit.Assert.assertEquals; |
| |
| import com.facebook.buck.model.BuildTarget; |
| import com.facebook.buck.model.BuildTargetFactory; |
| import com.facebook.buck.rules.BuildRuleResolver; |
| import com.facebook.buck.rules.SourcePath; |
| import com.facebook.buck.rules.SourcePathResolver; |
| import com.facebook.buck.rules.TestSourcePath; |
| import com.facebook.buck.rules.coercer.Either; |
| import com.google.common.base.Optional; |
| import com.google.common.collect.ImmutableMap; |
| import com.google.common.collect.ImmutableSortedSet; |
| |
| import org.junit.Test; |
| |
| import java.nio.file.Path; |
| |
| public class PythonUtilTest { |
| |
| @Test |
| public void toModuleMapWithExplicitMap() { |
| BuildTarget target = BuildTargetFactory.newInstance("//foo:bar"); |
| ImmutableMap<Path, SourcePath> srcs = PythonUtil.toModuleMap( |
| target, |
| new SourcePathResolver(new BuildRuleResolver()), |
| "srcs", |
| target.getBasePath(), Optional.of( |
| Either.<ImmutableSortedSet<SourcePath>, ImmutableMap<String, SourcePath>>ofRight( |
| ImmutableMap.<String, SourcePath>of( |
| "hello.py", new TestSourcePath("goodbye.py"))))); |
| assertEquals( |
| ImmutableMap.<Path, SourcePath>of( |
| target.getBasePath().resolve("hello.py"), |
| new TestSourcePath("goodbye.py")), |
| srcs); |
| } |
| |
| } |