func (s *AgentTestSuite) SetUpSuite(t *C) {
var err error
s.tmpDir, err = ioutil.TempDir("/tmp", "agent-test")
t.Assert(err, IsNil)
if err := pct.Basedir.Init(s.tmpDir); err != nil {
t.Fatal(err)
}
s.configFile = filepath.Join(s.tmpDir, pct.CONFIG_DIR, "agent"+pct.CONFIG_FILE_SUFFIX)
// Log
// todo: use log.Manager instead
s.logChan = make(chan *proto.LogEntry, 10)
s.logger = pct.NewLogger(s.logChan, "agent-test")
// Agent
s.config = &agent.Config{
AgentUuid: "abc-123-def",
ApiKey: "789",
ApiHostname: agent.DEFAULT_API_HOSTNAME,
}
s.sendChan = make(chan *proto.Cmd, 5)
s.recvChan = make(chan *proto.Reply, 5)
s.sendDataChan = make(chan interface{}, 5)
s.recvDataChan = make(chan interface{}, 5)
s.client = mock.NewWebsocketClient(s.sendChan, s.recvChan, s.sendDataChan, s.recvDataChan)
s.client.ErrChan = make(chan error)
s.readyChan = make(chan bool, 2)
s.traceChan = make(chan string, 10)
s.doneChan = make(chan bool, 1)
}
func (s *RelayTestSuite) SetUpSuite(t *C) {
s.logFile = fmt.Sprintf("/tmp/log_test.go.%d", os.Getpid())
s.sendChan = make(chan interface{}, 5)
s.recvChan = make(chan interface{}, 5)
s.connectChan = make(chan bool)
s.client = mock.NewWebsocketClient(nil, nil, s.sendChan, s.recvChan)
s.logChan = make(chan *proto.LogEntry, log.BUFFER_SIZE*3)
s.relay = log.NewRelay(s.client, s.logChan, "", proto.LOG_INFO, false)
s.logger = pct.NewLogger(s.relay.LogChan(), "test")
go s.relay.Run() // calls client.Connect()
}
func (s *ManagerTestSuite) SetUpSuite(t *C) {
var err error
s.tmpDir, err = ioutil.TempDir("/tmp", "agent-test")
t.Assert(err, IsNil)
if err := pct.Basedir.Init(s.tmpDir); err != nil {
t.Fatal(err)
}
s.sendChan = make(chan interface{}, 5)
s.recvChan = make(chan interface{}, 5)
s.connectChan = make(chan bool)
s.client = mock.NewWebsocketClient(nil, nil, s.sendChan, s.recvChan)
s.logChan = make(chan *proto.LogEntry, log.BUFFER_SIZE*3)
s.logFile = s.tmpDir + "/log"
}