// Smoldot

// Copyright (C) 2019-2022  Parity Technologies (UK) Ltd.

// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License

// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#![cfg(test)]

#[test]

fn basic_works() {

    // Note that this example file doesn't have the zstd header.

    super::zstd_decode(

        &include_bytes!("./example-runtime.wasm.zstd")[..],

        10 * 1024 * 1024,

    )

    .unwrap();

}

#[test]

fn limit_reached() {

    // Note that this example file doesn't have the zstd header.

    assert!(
matches!0
(

        super::zstd_decode(

            &include_bytes!("./example-runtime.wasm.zstd")[..],

            16 * 1024

        ),

        Err(super::Error::TooLarge)

    ));

}

#[test]

fn invalid_data() {

    assert!(
matches!0
(

        super::zstd_decode(
&(0..2048).map(1
|_| 0xff
).collect::<Vec<_>>(), 1024 * 10241
),

        Err(super::Error::InvalidZstd)

    ));

}

#[test]

fn basic_works_with_header() {

    super::zstd_decode_if_necessary(

        &include_bytes!("./polkadot-runtime-v9160.wasm.zstd")[..],

        10 * 1024 * 1024,

    )

    .unwrap();

}