starlark 0.13.0

/*
 * Copyright 2019 The Starlark in Rust Authors.
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * 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
 *
 *     https://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.
 */

#![cfg(test)]

use starlark_syntax::golden_test_template::golden_test_template;
use starlark_syntax::syntax::AstModule;
use starlark_syntax::syntax::Dialect;

use crate::assert::test_functions;
use crate::environment::GlobalsBuilder;
use crate::environment::Module;
use crate::eval::runtime::profile::data::ProfileDataImpl;
use crate::eval::Evaluator;
use crate::eval::ProfileData;
use crate::eval::ProfileMode;

fn test_profile_golden_for_mode(mode: ProfileMode) {
    let module = Module::new();
    let mut eval = Evaluator::new(&module);
    eval.enable_profile(&mode).unwrap();
    eval.eval_module(
        AstModule::parse(
            "test.star",
            r#"
def inner(x: int):
    if noop():
        return 10
    else:
        for x in range(10):
            noop()

def test():
    r = []
    for x in noop([1, 2, 3, 4, 5]):
        inner(x)
        r += noop([1] * 3)
    return r

test()
test()
test()

R = test()
"#
            .to_owned(),
            &Dialect::AllOptionsInternal,
        )
        .unwrap(),
        &GlobalsBuilder::extended().with(test_functions).build(),
    )
    .unwrap();

    let mut profile_data = match mode {
        ProfileMode::HeapSummaryRetained | ProfileMode::HeapFlameRetained => {
            drop(eval);
            let module = module.freeze().unwrap();
            module.heap_profile().unwrap()
        }
        _ => eval.gen_profile().unwrap(),
    };

    if let ProfileDataImpl::HeapFlameRetained(profile)
    | ProfileDataImpl::HeapFlameAllocated(profile)
    | ProfileDataImpl::HeapSummaryRetained(profile)
    | ProfileDataImpl::HeapSummaryAllocated(profile) = &mut profile_data.profile
    {
        profile.normalize_for_golden_tests();
    }

    golden_test_template(
        &format!(
            "src/eval/runtime/profile/golden/{}.golden",
            mode.name().replace('-', "_")
        ),
        &profile_data.gen().unwrap(),
    );

    // Smoke test for profile merging.
    ProfileData::merge([&profile_data, &profile_data]).unwrap();
}

#[test]
fn test_profile_golden_heap_summary_allocated() {
    test_profile_golden_for_mode(ProfileMode::HeapSummaryAllocated);
}

#[test]
fn test_profile_golden_heap_summary_retained() {
    test_profile_golden_for_mode(ProfileMode::HeapSummaryRetained);
}

#[test]
fn test_profile_golden_heap_flame_allocated() {
    test_profile_golden_for_mode(ProfileMode::HeapFlameAllocated);
}

#[test]
fn test_profile_golden_heap_flame_retained() {
    test_profile_golden_for_mode(ProfileMode::HeapFlameRetained);
}

#[test]
fn test_profile_golden_statement() {
    test_profile_golden_for_mode(ProfileMode::Statement);
}

#[test]
fn test_profile_golden_coverage() {
    test_profile_golden_for_mode(ProfileMode::Coverage);
}

#[test]
fn test_profile_golden_bytecode() {
    test_profile_golden_for_mode(ProfileMode::Bytecode);
}

#[test]
fn test_profile_golden_bytecode_pairs() {
    test_profile_golden_for_mode(ProfileMode::BytecodePairs);
}

#[test]
fn test_profile_golden_time_flame() {
    test_profile_golden_for_mode(ProfileMode::TimeFlame);
}

#[test]
fn test_profile_golden_typecheck() {
    test_profile_golden_for_mode(ProfileMode::Typecheck);
}