cloudinit/config/validate/validate_test.go

/*
   Copyright 2014 CoreOS, Inc.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/

package validate

import (
	"errors"
	"reflect"
	"testing"
)

func TestParseCloudConfig(t *testing.T) {
	tests := []struct {
		config string

		entries []Entry
	}{
		{},
		{
			config: "	",
			entries: []Entry{{entryError, "found character that cannot start any token", 1}},
		},
		{
			config:  "a:\na",
			entries: []Entry{{entryError, "could not find expected ':'", 2}},
		},
		{
			config:  "#hello\na:\na",
			entries: []Entry{{entryError, "could not find expected ':'", 3}},
		},
	}

	for _, tt := range tests {
		r := Report{}
		parseCloudConfig([]byte(tt.config), &r)

		if e := r.Entries(); !reflect.DeepEqual(tt.entries, e) {
			t.Errorf("bad report (%s): want %#v, got %#v", tt.config, tt.entries, e)
		}
	}
}

func TestValidateCloudConfig(t *testing.T) {
	tests := []struct {
		config string
		rules  []rule

		report Report
		err    error
	}{
		{
			rules: []rule{func(_ node, _ *Report) { panic("something happened") }},
			err:   errors.New("something happened"),
		},
		{
			config: "write_files:\n  - permissions: 0744",
			rules:  Rules,
		},
		{
			config: "write_files:\n  - permissions: '0744'",
			rules:  Rules,
		},
		{
			config: "write_files:\n  - permissions: 744",
			rules:  Rules,
		},
		{
			config: "write_files:\n  - permissions: '744'",
			rules:  Rules,
		},
		{
			config: "coreos:\n  update:\n    reboot-strategy: off",
			rules:  Rules,
		},
		{
			config: "coreos:\n  update:\n    reboot-strategy: false",
			rules:  Rules,
			report: Report{entries: []Entry{{entryError, "invalid value false", 3}}},
		},
	}

	for _, tt := range tests {
		r, err := validateCloudConfig([]byte(tt.config), tt.rules)
		if !reflect.DeepEqual(tt.err, err) {
			t.Errorf("bad error (%s): want %v, got %v", tt.config, tt.err, err)
		}
		if !reflect.DeepEqual(tt.report, r) {
			t.Errorf("bad report (%s): want %+v, got %+v", tt.config, tt.report, r)
		}
	}
}

func TestValidate(t *testing.T) {
	tests := []struct {
		config string

		report Report
	}{
		{},
		{
			config: "#!/bin/bash\necho hey",
		},
	}

	for i, tt := range tests {
		r, err := Validate([]byte(tt.config))
		if err != nil {
			t.Errorf("bad error (case #%d): want %v, got %v", i, nil, err)
		}
		if !reflect.DeepEqual(tt.report, r) {
			t.Errorf("bad report (case #%d): want %+v, got %+v", i, tt.report, r)
		}
	}
}

func BenchmarkValidate(b *testing.B) {
	config := `#cloud-config
hostname: test

coreos:
  etcd:
    name:      node001
    discovery: https://discovery.etcd.io/disco
    addr:      $public_ipv4:4001
    peer-addr: $private_ipv4:7001
  fleet:
    verbosity: 2
    metadata:  "hi"
  update:
    reboot-strategy: off
  units:
    - name:    hi.service
      command: start
      enable:  true
    - name:    bye.service
      command: stop

ssh_authorized_keys:
  - ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQC0g+ZTxC7weoIJLUafOgrm+h...
  - ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQC0g+ZTxC7weoIJLUafOgrm+h...

users:
  - name: me

write_files:
  - path: /etc/yes
    content: "Hi"

manage_etc_hosts: localhost`

	for i := 0; i < b.N; i++ {
		if _, err := Validate([]byte(config)); err != nil {
			panic(err)
		}
	}
}