Home | History | Annotate | Download | only in blueprint 
      1 // Copyright 2014 Google Inc. All rights reserved.
      2 //
      3 // Licensed under the Apache License, Version 2.0 (the "License");
      4 // you may not use this file except in compliance with the License.
      5 // You may obtain a copy of the License at
      6 //
      7 //     http://www.apache.org/licenses/LICENSE-2.0
      8 //
      9 // Unless required by applicable law or agreed to in writing, software
     10 // distributed under the License is distributed on an "AS IS" BASIS,
     11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 // See the License for the specific language governing permissions and
     13 // limitations under the License.
     14 
     15 package blueprint
     16 
     17 import (
     18 	"bytes"
     19 	"testing"
     20 )
     21 
     22 func ck(err error) {
     23 	if err != nil {
     24 		panic(err)
     25 	}
     26 }
     27 
     28 var ninjaWriterTestCases = []struct {
     29 	input  func(w *ninjaWriter)
     30 	output string
     31 }{
     32 	{
     33 		input: func(w *ninjaWriter) {
     34 			ck(w.Comment("foo"))
     35 		},
     36 		output: "# foo\n",
     37 	},
     38 	{
     39 		input: func(w *ninjaWriter) {
     40 			ck(w.Pool("foo"))
     41 		},
     42 		output: "pool foo\n",
     43 	},
     44 	{
     45 		input: func(w *ninjaWriter) {
     46 			ck(w.Rule("foo"))
     47 		},
     48 		output: "rule foo\n",
     49 	},
     50 	{
     51 		input: func(w *ninjaWriter) {
     52 			ck(w.Build("foo comment", "foo", []string{"o1", "o2"}, []string{"io1", "io2"},
     53 				[]string{"e1", "e2"}, []string{"i1", "i2"}, []string{"oo1", "oo2"}))
     54 		},
     55 		output: "# foo comment\nbuild o1 o2 | io1 io2: foo e1 e2 | i1 i2 || oo1 oo2\n",
     56 	},
     57 	{
     58 		input: func(w *ninjaWriter) {
     59 			ck(w.Default("foo"))
     60 		},
     61 		output: "default foo\n",
     62 	},
     63 	{
     64 		input: func(w *ninjaWriter) {
     65 			ck(w.Assign("foo", "bar"))
     66 		},
     67 		output: "foo = bar\n",
     68 	},
     69 	{
     70 		input: func(w *ninjaWriter) {
     71 			ck(w.ScopedAssign("foo", "bar"))
     72 		},
     73 		output: "    foo = bar\n",
     74 	},
     75 	{
     76 		input: func(w *ninjaWriter) {
     77 			ck(w.BlankLine())
     78 		},
     79 		output: "\n",
     80 	},
     81 	{
     82 		input: func(w *ninjaWriter) {
     83 			ck(w.Pool("p"))
     84 			ck(w.ScopedAssign("depth", "3"))
     85 			ck(w.BlankLine())
     86 			ck(w.Comment("here comes a rule"))
     87 			ck(w.Rule("r"))
     88 			ck(w.ScopedAssign("command", "echo out: $out in: $in _arg: $_arg"))
     89 			ck(w.ScopedAssign("pool", "p"))
     90 			ck(w.BlankLine())
     91 			ck(w.Build("r comment", "r", []string{"foo.o"}, nil, []string{"foo.in"}, nil, nil))
     92 			ck(w.ScopedAssign("_arg", "arg value"))
     93 		},
     94 		output: `pool p
     95     depth = 3
     96 
     97 # here comes a rule
     98 rule r
     99     command = echo out: $out in: $in _arg: $_arg
    100     pool = p
    101 
    102 # r comment
    103 build foo.o: r foo.in
    104     _arg = arg value
    105 `,
    106 	},
    107 }
    108 
    109 func TestNinjaWriter(t *testing.T) {
    110 	for i, testCase := range ninjaWriterTestCases {
    111 		buf := bytes.NewBuffer(nil)
    112 		w := newNinjaWriter(buf)
    113 		testCase.input(w)
    114 		if buf.String() != testCase.output {
    115 			t.Errorf("incorrect output for test case %d", i)
    116 			t.Errorf("  expected: %q", testCase.output)
    117 			t.Errorf("       got: %q", buf.String())
    118 		}
    119 	}
    120 }
    121