/**
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0.
 */

#include <aws/common/hash_table.h>
#include <aws/common/private/hash_table_impl.h>
#include <proof_helpers/make_common_data_structures.h>
#include <proof_helpers/utils.h>
// Currently takes 4m40s

void aws_hash_table_put_harness() {
    struct aws_hash_table map;
    ensure_allocated_hash_table(&map, MAX_TABLE_SIZE);
    __CPROVER_assume(aws_hash_table_is_valid(&map));
    map.p_impl->destroy_key_fn = nondet_bool() ? NULL : hash_proof_destroy_noop;
    map.p_impl->destroy_value_fn = nondet_bool() ? NULL : hash_proof_destroy_noop;
    map.p_impl->equals_fn = uninterpreted_equals_assert_inputs_nonnull;
    map.p_impl->hash_fn = uninterpreted_hasher;
    map.p_impl->alloc = aws_default_allocator();

    size_t empty_slot_idx;
    __CPROVER_assume(aws_hash_table_has_an_empty_slot(&map, &empty_slot_idx));

    void *key;
    void *value;
    bool track_was_created;
    int was_created;
    struct hash_table_state old_state = *map.p_impl;

    int rval = aws_hash_table_put(&map, key, value, track_was_created ? &was_created : NULL);
    if (rval == AWS_OP_SUCCESS) {
        if (track_was_created) {
            assert(map.p_impl->entry_count == old_state.entry_count + was_created);
        } else {
            assert(
                map.p_impl->entry_count == old_state.entry_count ||
                map.p_impl->entry_count == old_state.entry_count + 1);
        }
    } else {
        assert(map.p_impl->entry_count == old_state.entry_count);
    }
    assert(aws_hash_table_is_valid(&map));
}
