22c0ff3cf5
This allows golang tests to be marked as root or !root using build tags. The matching tests are then run as expected using our test runner. This also disables test caching which is unfriendly to repeated test running and is an absurd golang default to add. Lastly this hooks up the testing verbose flag to tests that accept a debug variable. These tests aren't enabled on travis yet because of how it installs golang.
100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
// Mgmt
|
|
// Copyright (C) 2013-2018+ James Shubin and the project contributors
|
|
// Written by James Shubin <james@shubin.ca> and the project contributors
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
// +build !root
|
|
|
|
package facts
|
|
|
|
import (
|
|
"log"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/purpleidea/mgmt/lang/funcs"
|
|
"github.com/purpleidea/mgmt/pgraph"
|
|
)
|
|
|
|
const Debug = false // switch on for more interactive log messages when testing!
|
|
|
|
// logf switches messages to use realtime logging when debugging tests, and the
|
|
// quiet logging which is not shown until test failures, when debug mode is off.
|
|
func logf(t *testing.T, format string, args ...interface{}) {
|
|
if Debug {
|
|
log.Printf(format, args...)
|
|
} else {
|
|
t.Logf(format, args...)
|
|
}
|
|
}
|
|
|
|
func TestFuncGraph0(t *testing.T) {
|
|
logf(t, "Hello!")
|
|
g, _ := pgraph.NewGraph("empty") // empty graph
|
|
|
|
obj := &funcs.Engine{
|
|
Graph: g,
|
|
}
|
|
|
|
logf(t, "Init...")
|
|
if err := obj.Init(); err != nil {
|
|
t.Errorf("could not init: %+v", err)
|
|
return
|
|
}
|
|
|
|
logf(t, "Validate...")
|
|
if err := obj.Validate(); err != nil {
|
|
t.Errorf("could not validate: %+v", err)
|
|
return
|
|
}
|
|
|
|
logf(t, "Run...")
|
|
if err := obj.Run(); err != nil {
|
|
t.Errorf("could not run: %+v", err)
|
|
return
|
|
}
|
|
|
|
// wait for some activity
|
|
logf(t, "Stream...")
|
|
stream := obj.Stream()
|
|
logf(t, "Loop...")
|
|
br := time.After(time.Duration(5) * time.Second)
|
|
Loop:
|
|
for {
|
|
select {
|
|
case err, ok := <-stream:
|
|
if !ok {
|
|
logf(t, "Stream break...")
|
|
break Loop
|
|
}
|
|
if err != nil {
|
|
logf(t, "Error: %+v", err)
|
|
continue
|
|
}
|
|
|
|
case <-br:
|
|
logf(t, "Break...")
|
|
t.Errorf("empty graph should have closed stream")
|
|
break Loop
|
|
}
|
|
}
|
|
|
|
logf(t, "Closing...")
|
|
if err := obj.Close(); err != nil {
|
|
t.Errorf("could not close: %+v", err)
|
|
return
|
|
}
|
|
}
|