Skip to content
Closed
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 38 additions & 0 deletions crates/pyrometer/tests/no_killed_ctxs.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,14 @@ use std::env;
mod helpers;
use helpers::*;

#[test]
fn test_assign() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
let path_str = format!("{manifest_dir}/tests/test_data/assign.sol");
let sol = include_str!("./test_data/assign.sol");
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_bitwise() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
Expand All @@ -10,6 +18,14 @@ fn test_bitwise() {
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_binop() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
let path_str = format!("{manifest_dir}/tests/test_data/bin_op.sol");
let sol = include_str!("./test_data/bin_op.sol");
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_cast() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
Expand All @@ -18,6 +34,13 @@ fn test_cast() {
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_condop() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
let path_str = format!("{manifest_dir}/tests/test_data/cond_op.sol");
assert_no_parse_errors(path_str);
}

#[test]
fn test_dyn_types() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
Expand All @@ -42,6 +65,14 @@ fn test_function_calls() {
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_literals() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
let path_str = format!("{manifest_dir}/tests/test_data/literals.sol");
let sol = include_str!("./test_data/literals.sol");
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_logical() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
Expand Down Expand Up @@ -82,6 +113,13 @@ fn test_require() {
assert_no_ctx_killed(path_str, sol);
}

#[test]
fn test_require_with_killed() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
let path_str = format!("{manifest_dir}/tests/test_data/require_with_killed.sol");
assert_no_parse_errors(path_str);
}

#[test]
fn test_using() {
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
Expand Down
20 changes: 20 additions & 0 deletions crates/pyrometer/tests/test_data/assign.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
pragma solidity ^0.8.0;

contract Assign {

function doAssignment() public {
// Multi-value LHS (tuple)
(uint x, uint y) = (uint16(1), 2);

// Single value RHS
uint z = 3;

(x, y) = (z, z);

// uint[2] memory a = [uint(1), uint(2)];
// uint[2] memory b = [uint(3), uint(4)];


// (a, b) = (b, a);
}
}
10 changes: 10 additions & 0 deletions crates/pyrometer/tests/test_data/bin_op.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
pragma solidity ^0.8.0;

contract BinOp {

function testBinOp(int y) public {
int x = 1;
int z = 5 + x;
int a = x * y;
}
}
10 changes: 10 additions & 0 deletions crates/pyrometer/tests/test_data/cmp.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
pragma solidity ^0.8.0;

contract Cmp {

function testCmp(int x) public {
uint a = 5;
bool b = (5 < 6) || (5 == 6 && 0 < 1); // Correct the tuple comparison

}
}
53 changes: 53 additions & 0 deletions crates/pyrometer/tests/test_data/cond_op.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
pragma solidity ^0.8.0;

contract CondOp {

function if_both_possible(uint x) public returns (uint) {
if (x > 100) {
return 1;
} else {
return 2;
}
}

function if_first_possible() public returns (uint) {
uint x = 101;
if (x > 100) {
return 1;
} else {
return 2;
}
}

function if_second_possible() public returns (uint) {
uint x = 99;
if (x > 100) {
return 1;
} else {
return 2;
}
}

function if_neither_possible(uint x) public returns (uint) {
if (x > 100) {
require(x < 100); // not possible
return 1;
} else {
require(x > 100); // not possible
return 2;
}
return 0;
}


function ternaryParam(uint x) public returns (uint) {
uint y = x > 2 ? 1 : 2;
return y;
}

function ternaryLiteral() public returns (uint) {
uint y = 1 > 2 ? 1 : 2;
"pyro::variable::y::range::[2,2]";
return y;
}
}
45 changes: 45 additions & 0 deletions crates/pyrometer/tests/test_data/dyn_types.sol
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ contract DynTypes {
}

mapping(address => Strukt) public someMapping;
mapping(address => Strukt[]) public someMapping2;

function bytes_dyn(bytes calldata x) public {
bytes memory y = x;
Expand All @@ -15,6 +16,7 @@ contract DynTypes {
require(y.length == 9);
}


function array_dyn(uint256[] memory x) public {
x[0] = 5;
require(x.length < 10);
Expand Down Expand Up @@ -80,4 +82,47 @@ contract DynTypes {
address holder = holders[j];
}
}

struct DontUseMoreThanOnce {
uint256 a;
uint256 b;
}

function dynUserType() public {
DontUseMoreThanOnce[] memory dont = new DontUseMoreThanOnce[](1);
dont[0].a = 100;
dont[0].b = 100;
require(dont[0].a == 100);
}

function getReturnedUserType() public pure {
// Strukt[] memory strukt = returnUserType()[0];
Strukt memory strukt = returnUserType()[0];
require(strukt.a == 100);
}

function returnUserType() public pure returns (Strukt[] memory) {
Strukt[] memory strukt = new Strukt[](1);
strukt[0].a = 100;
strukt[0].b = 100;
return strukt;
}

function multiDimensionalArray() public returns (bool z) {
uint256[][] memory multiArray = new uint256[][](2);
uint256[] memory indices = new uint256[](2);

indices[0] = 0;
indices[1] = 1;

for (uint i = 0; i < multiArray.length; i++) {
multiArray[i] = new uint256[](2);
for (uint j = 0; j < multiArray[i].length; j++) {
multiArray[i][j] = 1;
}
}

z = true;
}

}
41 changes: 41 additions & 0 deletions crates/pyrometer/tests/test_data/env.sol
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,45 @@ contract Env {
function msg_data() public returns (bytes memory) {
return msg.data;
}

function testBlock() public payable {
/*
blockhash(uint blockNumber) returns (bytes32): hash of the given block - only works for 256 most recent blocks
blobhash(uint index) returns (bytes32): versioned hash of the index-th blob associated with the current transaction. A versioned hash consists of a single byte representing the version (currently 0x01), followed by the last 31 bytes of the SHA256 hash of the KZG commitment (EIP-4844).
block.basefee (uint): current block’s base fee (EIP-3198 and EIP-1559)
block.blobbasefee (uint): current block’s blob base fee (EIP-7516 and EIP-4844)
block.chainid (uint): current chain id
block.coinbase (address payable): current block miner’s address
block.difficulty (uint): current block difficulty (EVM < Paris). For other EVM versions it behaves as a deprecated alias for block.prevrandao that will be removed in the next breaking release
block.gaslimit (uint): current block gaslimit
block.number (uint): current block number
block.prevrandao (uint): random number provided by the beacon chain (EVM >= Paris) (see EIP-4399 )
block.timestamp (uint): current block timestamp in seconds since Unix epoch
gasleft() returns (uint256): remaining gas
msg.data (bytes): complete calldata
msg.sender (address): sender of the message (current call)
msg.sig (bytes4): first four bytes of the calldata (i.e. function identifier)
msg.value (uint): number of wei sent with the message
tx.gasprice (uint): gas price of the transaction
tx.origin (address): sender of the transaction (full call chain)
*/
bytes32 a = blockhash(1);
bytes32 b = blobhash(1);
uint c = block.basefee;
uint d = block.blobbasefee;
uint e = block.chainid;
address payable f = block.coinbase;
uint g = block.difficulty;
uint h = block.gaslimit;
uint i = block.number;
uint j = block.prevrandao;
uint k = block.timestamp;
uint l = gasleft();
bytes memory m = msg.data;
address n = msg.sender;
bytes4 o = msg.sig;
uint p = msg.value;
uint q = tx.gasprice;
address r = tx.origin;
}
}
52 changes: 52 additions & 0 deletions crates/pyrometer/tests/test_data/function_calls.sol
Original file line number Diff line number Diff line change
Expand Up @@ -83,3 +83,55 @@ contract ExternalFuncCalls {

// function foo(address by, address from, address to, uint256 id) internal {}
// }

contract S1 {
function a(uint x) internal pure virtual returns (uint) {
return 100;
}
}

contract S2 {
function a(uint x) internal pure virtual returns (uint) {
return 10;
}

function b(uint x) internal pure virtual returns (uint) {
return 10;
}
}

contract C is S1, S2 {
function supers(uint128 x) public pure returns (uint) {
uint local_a = a(1);
uint super_a = super.a(1);
require(local_a == 50);
require(super_a == 10);

uint local_super_b = b(x);
uint super_b = super.b(x);
require(local_super_b == super_b);
return 0;
}

function a(uint256 x) internal pure override(S1, S2) returns (uint) {
return 50;
}
}

contract D is S2, S1 {
function supers(uint128 x) public pure returns (uint) {
uint local_a = a(1);
uint super_a = super.a(1);
require(local_a == 50);
require(super_a == 100);

uint local_super_b = b(x);
uint super_b = super.b(x);
require(local_super_b == super_b);
return 0;
}

function a(uint256 x) internal pure override(S1, S2) returns (uint) {
return 50;
}
}
39 changes: 39 additions & 0 deletions crates/pyrometer/tests/test_data/literals.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
contract Literals {

function foo() public returns (string memory) {
uint a = 115792089237316195423570985008687907853269984665640564039457584007913129639935; // ok
// uint b = 115792089237316195423570985008687907853269984665640564039457584007913129639936; // too big
// uint c = 115792089237316195423570985008687907853269984665640564039457584007913129639935 ** 2; // too big
int d = -57896044618658097711785492504343953926634992332820282019728792003956564819968; // ok
// int e = -57896044618658097711785492504343953926634992332820282019728792003956564819969; // too big
uint f = 1.0 ** 2; // ok
// uint g = 1.5 ** 2; // not uint
uint h = 1.5 ** 0; // ok
"pyro::variable::h::range::[1,1]";
uint256 i = 123 ** 10; // 792594609605189126649
address w = address(0xdCad3a6d3569DF655070DEd06cb7A1b2Ccd1D3AF);
w = 0xdCad3a6d3569D_F655070DEd06cb7A_1b2Ccd1D3AF;

uint j = 0.000000002e18;
j = 25;
j = 1.5e0 ether;
"pyro::variable::j::range::[1500000000000000000,1500000000000000000]";
int k = -0;
k = 25.5e5;
k = -23.5e5 ether;
"pyro::variable::k::range::[-2350000000000000000000000,-2350000000000000000000000]";

k = -23.5e5 seconds;
k = -23.5e5 minutes;
k = -23.5e5 hours;
k = -23.5e5 days;
k = -23.5e5 weeks;
k = -0x54;
"pyro::variable::k::range::[-84,-84]";
string memory s = unicode"🔥🔫"; // TODO unicode string values is not correct yet

bytes memory r = hex"11111111111111111111111111111111111111111111111111111111111111111111111111111111";
r = hex"1111111111111111111111111111111111111111" hex"111111111111111111111111111111111111111111111111";
return s;
}
}
Loading