javascript - format_print_original_test.go

     1  package javascript_test
     2  
     3  import (
     4  	"bytes"
     5  	"fmt"
     6  	"reflect"
     7  	"testing"
     8  
     9  	"vimagination.zapto.org/javascript"
    10  	"vimagination.zapto.org/javascript/walk"
    11  	"vimagination.zapto.org/parser"
    12  )
    13  
    14  func TestPrintingOriginal(t *testing.T) {
    15  	for n, test := range [...]struct {
    16  		Input, Output string
    17  	}{
    18  		{ // 1
    19  			"a = 1",
    20  			"a=1",
    21  		},
    22  		{ // 2
    23  			"a = 1;",
    24  			"a=1",
    25  		},
    26  		{ // 3
    27  			"a = 1;b=2",
    28  			"a=1;b=2",
    29  		},
    30  		{ // 4
    31  			"import a from './b'",
    32  			"import a from'./b'",
    33  		},
    34  		{ // 5
    35  			"import {a as b} from './b'",
    36  			"import{a as b}from'./b'",
    37  		},
    38  		{ // 6
    39  			"export {a as b}",
    40  			"export{a as b}",
    41  		},
    42  		{ // 7
    43  			"for (a in b) {}",
    44  			"for(a in b){}",
    45  		},
    46  		{ // 8
    47  			"for (const [a] of [b]) {}",
    48  			"for(const[a]of[b]){}",
    49  		},
    50  		{ // 9
    51  			"#a in b",
    52  			"#a in b",
    53  		},
    54  		{ // 10
    55  			"{a=1}",
    56  			"{a=1}",
    57  		},
    58  		{ // 11
    59  			"throw 1",
    60  			"throw 1",
    61  		},
    62  		{ // 12
    63  			"class A{a;b;c;d(){}e}",
    64  			"class A{a;b;c;d(){}e}",
    65  		},
    66  		{ // 13
    67  			"({a})",
    68  			"({a})",
    69  		},
    70  		{ // 14
    71  			"({a,b})",
    72  			"({a,b})",
    73  		},
    74  		{ // 15
    75  			"({a:d,b:c})",
    76  			"({a:d,b:c})",
    77  		},
    78  		{ // 16
    79  			"({a,b:c})",
    80  			"({a,b:c})",
    81  		},
    82  		{ // 17
    83  			"for(const{a,b}in c){}",
    84  			"for(const{a,b}in c){}",
    85  		},
    86  	} {
    87  		tk := parser.NewStringTokeniser(test.Input)
    88  
    89  		m, err := javascript.ParseModule(&tk)
    90  		if err != nil {
    91  			t.Errorf("test %d.1: unexpected error: %s", n+1, err)
    92  
    93  			continue
    94  		}
    95  
    96  		var h walk.Handler
    97  
    98  		h = walk.HandlerFunc(func(t javascript.Type) error {
    99  			v := reflect.ValueOf(t)
   100  
   101  			if v.Type().Kind() != reflect.Pointer || v.Type().Elem().Kind() != reflect.Struct {
   102  				return nil
   103  			}
   104  
   105  			if f, ok := v.Type().Elem().FieldByName("Tokens"); ok {
   106  				v.Elem().FieldByIndex(f.Index).SetZero()
   107  			}
   108  
   109  			return walk.Walk(t, h)
   110  		})
   111  
   112  		if err = h.Handle(m); err != nil {
   113  			t.Errorf("test %d.2: unexpected error: %s", n+1, err)
   114  
   115  			continue
   116  		}
   117  
   118  		var buf bytes.Buffer
   119  
   120  		fmt.Fprintf(&buf, "%#s", m)
   121  
   122  		if str := buf.String(); str != test.Output {
   123  			t.Errorf("test %d.3: expecting %q, got %q", n+1, test.Output, str)
   124  		}
   125  	}
   126  }
   127