func (s *ListenerSuite) TestNewRunListenerOnExistingSocketRemovesItAndSucceeds(c *gc.C) {
s.NewRunListener(c)
listener, err := uniter.NewRunListener(&mockRunner{}, s.socketPath)
c.Assert(err, gc.IsNil)
c.Assert(listener, gc.NotNil)
listener.Close()
}
// Mirror the params to uniter.NewRunListener, but add cleanup to close it.
func (s *ListenerSuite) NewRunListener(c *gc.C) *uniter.RunListener {
s.socketPath = filepath.Join(c.MkDir(), "test.listener")
listener, err := uniter.NewRunListener(&mockRunner{c}, s.socketPath)
c.Assert(err, gc.IsNil)
c.Assert(listener, gc.NotNil)
s.AddCleanup(func(*gc.C) {
listener.Close()
})
return listener
}
func (s *RunTestSuite) runListenerForAgent(c *gc.C, agent string) {
s.PatchValue(&AgentDir, c.MkDir())
testAgentDir := filepath.Join(AgentDir, agent)
err := os.Mkdir(testAgentDir, 0755)
c.Assert(err, gc.IsNil)
socketPath := filepath.Join(testAgentDir, uniter.RunListenerFile)
listener, err := uniter.NewRunListener(&mockRunner{c}, socketPath)
c.Assert(err, gc.IsNil)
c.Assert(listener, gc.NotNil)
s.AddCleanup(func(*gc.C) {
listener.Close()
})
}
