github.com/aigarnetwork/aigar@v0.0.0-20191115204914-d59a6eb70f8e/core/asm/compiler_test.go (about)

     1  //  Copyright 2018 The go-ethereum Authors
     2  //  Copyright 2019 The go-aigar Authors
     3  //  This file is part of the go-aigar library.
     4  //
     5  //  The go-aigar library is free software: you can redistribute it and/or modify
     6  //  it under the terms of the GNU Lesser General Public License as published by
     7  //  the Free Software Foundation, either version 3 of the License, or
     8  //  (at your option) any later version.
     9  //
    10  //  The go-aigar library is distributed in the hope that it will be useful,
    11  //  but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    13  //  GNU Lesser General Public License for more details.
    14  //
    15  //  You should have received a copy of the GNU Lesser General Public License
    16  //  along with the go-aigar library. If not, see <http://www.gnu.org/licenses/>.
    17  
    18  package asm
    19  
    20  import (
    21  	"testing"
    22  )
    23  
    24  func TestCompiler(t *testing.T) {
    25  	tests := []struct {
    26  		input, output string
    27  	}{
    28  		{
    29  			input: `
    30  	GAS
    31  	label:
    32  	PUSH @label
    33  `,
    34  			output: "5a5b6300000001",
    35  		},
    36  		{
    37  			input: `
    38  	PUSH @label
    39  	label:
    40  `,
    41  			output: "63000000055b",
    42  		},
    43  		{
    44  			input: `
    45  	PUSH @label
    46  	JUMP
    47  	label:
    48  `,
    49  			output: "6300000006565b",
    50  		},
    51  		{
    52  			input: `
    53  	JUMP @label
    54  	label:
    55  `,
    56  			output: "6300000006565b",
    57  		},
    58  	}
    59  	for _, test := range tests {
    60  		ch := Lex([]byte(test.input), false)
    61  		c := NewCompiler(false)
    62  		c.Feed(ch)
    63  		output, err := c.Compile()
    64  		if len(err) != 0 {
    65  			t.Errorf("compile error: %v\ninput: %s", err, test.input)
    66  			continue
    67  		}
    68  		if output != test.output {
    69  			t.Errorf("incorrect output\ninput: %sgot:  %s\nwant: %s\n", test.input, output, test.output)
    70  		}
    71  	}
    72  }