src/test/java/com/google/devtools/build/lib/skyframe/serialization/StatementCodecTest.java

// Copyright 2022 The Bazel Authors. All rights reserved.
//
// 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.devtools.build.lib.skyframe.serialization;

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

import com.google.common.collect.ImmutableList;
import com.google.devtools.build.lib.skyframe.serialization.testutils.SerializationTester;
import net.starlark.java.syntax.ParserInput;
import net.starlark.java.syntax.StarlarkFile;
import net.starlark.java.syntax.Statement;
import net.starlark.java.syntax.SyntaxError;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

/** Tests for {@link Statement_AutoCodec}. */
@RunWith(JUnit4.class)
public class StatementCodecTest {

  @Test
  public void testCodec() throws Exception {
    ImmutableList<Statement> stmts =
        parse(
            "x = y",
            "x += y",
            "5",
            "break",
            "for x in y: print(x)",
            "for x in y: pass",
            "def f(x): print(x)",
            // Additional ad hoc assertions for function signature below.
            "def f(a, b='B', *c, d=D, **e): print(x)",
            "def f(a, b='B', *, d=D, **e): print(x)",
            "if x < y: print(x)",
            "elif y < x: print(y)",
            "else: print(z)",
            "load(':foo.bzl', 'A', b='B')",
            "return",
            "return 5",
            "def fib(n):",
            "  if n < 0:",
            "    print('Illegal argument:', n, sep='\t')",
            "  elif n < 2:",
            "    return 1",
            "  else:",
            "    return fib(n - 2) + fib(n - 1)",
            "def cumulative_sum(arr):",
            "  if len(arr) <= 1:",
            "    pass",
            "  else:",
            "    for i in range(1, len(arr)):",
            "      arr[i] += arr[i-1]",
            "      print('Running total is ' + arr[i])",
            "      print('Remaining elements: ' + arr[i+1:])");
    // SerializationTester is dangerously overloaded: the argument must
    // be an ImmutableList or else the Object... overload is selected.
    new SerializationTester(stmts)
        .setVerificationFunction(
            (original, deserialized) ->
                assertThat(((Statement) deserialized).prettyPrint())
                    .isEqualTo(((Statement) original).prettyPrint()))
        .runTests();
  }

  private static ImmutableList<Statement> parse(String... lines) throws SyntaxError.Exception {
    ParserInput input = ParserInput.fromLines(lines);
    return StarlarkFile.parse(input).getStatements();
  }
}