// SPDX-License-Identifier: MIT
pragma solidity 0.8.15;

import { RLPWriter } from "src/libraries/rlp/RLPWriter.sol";
import { Test } from "forge-std/Test.sol";

contract RLPWriter_writeString_Test is Test {
    function test_writeString_empty_succeeds() external {
        assertEq(RLPWriter.writeString(""), hex"80");
    }

    function test_writeString_bytestring00_succeeds() external {
        assertEq(RLPWriter.writeString("\u0000"), hex"00");
    }

    function test_writeString_bytestring01_succeeds() external {
        assertEq(RLPWriter.writeString("\u0001"), hex"01");
    }

    function test_writeString_bytestring7f_succeeds() external {
        assertEq(RLPWriter.writeString("\u007F"), hex"7f");
    }

    function test_writeString_shortstring_succeeds() external {
        assertEq(RLPWriter.writeString("dog"), hex"83646f67");
    }

    function test_writeString_shortstring2_succeeds() external {
        assertEq(
            RLPWriter.writeString("Lorem ipsum dolor sit amet, consectetur adipisicing eli"),
            hex"b74c6f72656d20697073756d20646f6c6f722073697420616d65742c20636f6e7365637465747572206164697069736963696e6720656c69"
        );
    }

    function test_writeString_longstring_succeeds() external {
        assertEq(
            RLPWriter.writeString("Lorem ipsum dolor sit amet, consectetur adipisicing elit"),
            hex"b8384c6f72656d20697073756d20646f6c6f722073697420616d65742c20636f6e7365637465747572206164697069736963696e6720656c6974"
        );
    }

    function test_writeString_longstring2_succeeds() external {
        assertEq(
            RLPWriter.writeString(
                "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur mauris magna, suscipit sed vehicula non, iaculis faucibus tortor. Proin suscipit ultricies malesuada. Duis tortor elit, dictum quis tristique eu, ultrices at risus. Morbi a est imperdiet mi ullamcorper aliquet suscipit nec lorem. Aenean quis leo mollis, vulputate elit varius, consequat enim. Nulla ultrices turpis justo, et posuere urna consectetur nec. Proin non convallis metus. Donec tempor ipsum in mauris congue sollicitudin. Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia Curae; Suspendisse convallis sem vel massa faucibus, eget lacinia lacus tempor. Nulla quis ultricies purus. Proin auctor rhoncus nibh condimentum mollis. Aliquam consequat enim at metus luctus, a eleifend purus egestas. Curabitur at nibh metus. Nam bibendum, neque at auctor tristique, lorem libero aliquet arcu, non interdum tellus lectus sit amet eros. Cras rhoncus, metus ac ornare cursus, dolor justo ultrices metus, at ullamcorper volutpat"
            ),
            hex"b904004c6f72656d20697073756d20646f6c6f722073697420616d65742c20636f6e73656374657475722061646970697363696e6720656c69742e20437572616269747572206d6175726973206d61676e612c20737573636970697420736564207665686963756c61206e6f6e2c20696163756c697320666175636962757320746f72746f722e2050726f696e20737573636970697420756c74726963696573206d616c6573756164612e204475697320746f72746f7220656c69742c2064696374756d2071756973207472697374697175652065752c20756c7472696365732061742072697375732e204d6f72626920612065737420696d70657264696574206d6920756c6c616d636f7270657220616c6971756574207375736369706974206e6563206c6f72656d2e2041656e65616e2071756973206c656f206d6f6c6c69732c2076756c70757461746520656c6974207661726975732c20636f6e73657175617420656e696d2e204e756c6c6120756c74726963657320747572706973206a7573746f2c20657420706f73756572652075726e6120636f6e7365637465747572206e65632e2050726f696e206e6f6e20636f6e76616c6c6973206d657475732e20446f6e65632074656d706f7220697073756d20696e206d617572697320636f6e67756520736f6c6c696369747564696e2e20566573746962756c756d20616e746520697073756d207072696d697320696e206661756369627573206f726369206c756374757320657420756c74726963657320706f737565726520637562696c69612043757261653b2053757370656e646973736520636f6e76616c6c69732073656d2076656c206d617373612066617563696275732c2065676574206c6163696e6961206c616375732074656d706f722e204e756c6c61207175697320756c747269636965732070757275732e2050726f696e20617563746f722072686f6e637573206e69626820636f6e64696d656e74756d206d6f6c6c69732e20416c697175616d20636f6e73657175617420656e696d206174206d65747573206c75637475732c206120656c656966656e6420707572757320656765737461732e20437572616269747572206174206e696268206d657475732e204e616d20626962656e64756d2c206e6571756520617420617563746f72207472697374697175652c206c6f72656d206c696265726f20616c697175657420617263752c206e6f6e20696e74657264756d2074656c6c7573206c65637475732073697420616d65742065726f732e20437261732072686f6e6375732c206d65747573206163206f726e617265206375727375732c20646f6c6f72206a7573746f20756c747269636573206d657475732c20617420756c6c616d636f7270657220766f6c7574706174"
        );
    }
}

contract RLPWriter_writeUint_Test is Test {
    function test_writeUint_zero_succeeds() external {
        assertEq(RLPWriter.writeUint(0x0), hex"80");
    }

    function test_writeUint_smallint_succeeds() external {
        assertEq(RLPWriter.writeUint(1), hex"01");
    }

    function test_writeUint_smallint2_succeeds() external {
        assertEq(RLPWriter.writeUint(16), hex"10");
    }

    function test_writeUint_smallint3_succeeds() external {
        assertEq(RLPWriter.writeUint(79), hex"4f");
    }

    function test_writeUint_smallint4_succeeds() external {
        assertEq(RLPWriter.writeUint(127), hex"7f");
    }

    function test_writeUint_mediumint_succeeds() external {
        assertEq(RLPWriter.writeUint(128), hex"8180");
    }

    function test_writeUint_mediumint2_succeeds() external {
        assertEq(RLPWriter.writeUint(1000), hex"8203e8");
    }

    function test_writeUint_mediumint3_succeeds() external {
        assertEq(RLPWriter.writeUint(100000), hex"830186a0");
    }
}

contract RLPWriter_writeList_Test is Test {
    function test_writeList_empty_succeeds() external {
        assertEq(RLPWriter.writeList(new bytes[](0)), hex"c0");
    }

    function test_writeList_stringList_succeeds() external {
        bytes[] memory list = new bytes[](3);
        list[0] = RLPWriter.writeString("dog");
        list[1] = RLPWriter.writeString("god");
        list[2] = RLPWriter.writeString("cat");

        assertEq(RLPWriter.writeList(list), hex"cc83646f6783676f6483636174");
    }

    function test_writeList_multiList_succeeds() external {
        bytes[] memory list = new bytes[](3);
        bytes[] memory list2 = new bytes[](1);
        list2[0] = RLPWriter.writeUint(4);

        list[0] = RLPWriter.writeString("zw");
        list[1] = RLPWriter.writeList(list2);
        list[2] = RLPWriter.writeUint(1);

        assertEq(RLPWriter.writeList(list), hex"c6827a77c10401");
    }

    function test_writeList_shortListMax1_succeeds() external {
        bytes[] memory list = new bytes[](11);
        list[0] = RLPWriter.writeString("asdf");
        list[1] = RLPWriter.writeString("qwer");
        list[2] = RLPWriter.writeString("zxcv");
        list[3] = RLPWriter.writeString("asdf");
        list[4] = RLPWriter.writeString("qwer");
        list[5] = RLPWriter.writeString("zxcv");
        list[6] = RLPWriter.writeString("asdf");
        list[7] = RLPWriter.writeString("qwer");
        list[8] = RLPWriter.writeString("zxcv");
        list[9] = RLPWriter.writeString("asdf");
        list[10] = RLPWriter.writeString("qwer");

        assertEq(
            RLPWriter.writeList(list),
            hex"f784617364668471776572847a78637684617364668471776572847a78637684617364668471776572847a78637684617364668471776572"
        );
    }

    function test_writeList_longlist1_succeeds() external {
        bytes[] memory list = new bytes[](4);
        bytes[] memory list2 = new bytes[](3);

        list2[0] = RLPWriter.writeString("asdf");
        list2[1] = RLPWriter.writeString("qwer");
        list2[2] = RLPWriter.writeString("zxcv");

        list[0] = RLPWriter.writeList(list2);
        list[1] = RLPWriter.writeList(list2);
        list[2] = RLPWriter.writeList(list2);
        list[3] = RLPWriter.writeList(list2);

        assertEq(
            RLPWriter.writeList(list),
            hex"f840cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376"
        );
    }

    function test_writeList_longlist2_succeeds() external {
        bytes[] memory list = new bytes[](32);
        bytes[] memory list2 = new bytes[](3);

        list2[0] = RLPWriter.writeString("asdf");
        list2[1] = RLPWriter.writeString("qwer");
        list2[2] = RLPWriter.writeString("zxcv");

        for (uint256 i = 0; i < 32; i++) {
            list[i] = RLPWriter.writeList(list2);
        }

        assertEq(
            RLPWriter.writeList(list),
            hex"f90200cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376cf84617364668471776572847a786376"
        );
    }

    function test_writeList_listoflists_succeeds() external {
        // [ [ [], [] ], [] ]
        bytes[] memory list = new bytes[](2);
        bytes[] memory list2 = new bytes[](2);

        list2[0] = RLPWriter.writeList(new bytes[](0));
        list2[1] = RLPWriter.writeList(new bytes[](0));

        list[0] = RLPWriter.writeList(list2);
        list[1] = RLPWriter.writeList(new bytes[](0));

        assertEq(RLPWriter.writeList(list), hex"c4c2c0c0c0");
    }

    function test_writeList_listoflists2_succeeds() external {
        // [ [], [[]], [ [], [[]] ] ]
        bytes[] memory list = new bytes[](3);
        list[0] = RLPWriter.writeList(new bytes[](0));

        bytes[] memory list2 = new bytes[](1);
        list2[0] = RLPWriter.writeList(new bytes[](0));

        list[1] = RLPWriter.writeList(list2);

        bytes[] memory list3 = new bytes[](2);
        list3[0] = RLPWriter.writeList(new bytes[](0));
        list3[1] = RLPWriter.writeList(list2);

        list[2] = RLPWriter.writeList(list3);

        assertEq(RLPWriter.writeList(list), hex"c7c0c1c0c3c0c1c0");
    }

    function test_writeList_dictTest1_succeeds() external {
        bytes[] memory list = new bytes[](4);

        bytes[] memory list1 = new bytes[](2);
        list1[0] = RLPWriter.writeString("key1");
        list1[1] = RLPWriter.writeString("val1");

        bytes[] memory list2 = new bytes[](2);
        list2[0] = RLPWriter.writeString("key2");
        list2[1] = RLPWriter.writeString("val2");

        bytes[] memory list3 = new bytes[](2);
        list3[0] = RLPWriter.writeString("key3");
        list3[1] = RLPWriter.writeString("val3");

        bytes[] memory list4 = new bytes[](2);
        list4[0] = RLPWriter.writeString("key4");
        list4[1] = RLPWriter.writeString("val4");

        list[0] = RLPWriter.writeList(list1);
        list[1] = RLPWriter.writeList(list2);
        list[2] = RLPWriter.writeList(list3);
        list[3] = RLPWriter.writeList(list4);

        assertEq(
            RLPWriter.writeList(list),
            hex"ecca846b6579318476616c31ca846b6579328476616c32ca846b6579338476616c33ca846b6579348476616c34"
        );
    }
}
