1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// Copyright 2015 Canonical Ltd.
// Licensed under the LGPLv3, see LICENCE file for details.
package shell_test
import (
"github.com/juju/testing"
jc "github.com/juju/testing/checkers"
gc "gopkg.in/check.v1"
"github.com/juju/utils/shell"
)
var _ = gc.Suite(&powershellSuite{})
type powershellSuite struct {
testing.IsolationSuite
dirname string
filename string
renderer *shell.PowershellRenderer
}
func (s *powershellSuite) SetUpTest(c *gc.C) {
s.IsolationSuite.SetUpTest(c)
s.dirname = `C:\some\dir`
s.filename = s.dirname + `\file`
s.renderer = &shell.PowershellRenderer{}
}
func (s powershellSuite) TestExeSuffix(c *gc.C) {
suffix := s.renderer.ExeSuffix()
c.Check(suffix, gc.Equals, ".exe")
}
func (s powershellSuite) TestShQuote(c *gc.C) {
quoted := s.renderer.Quote("abc")
c.Check(quoted, gc.Equals, `'abc'`)
}
func (s powershellSuite) TestChmod(c *gc.C) {
commands := s.renderer.Chmod(s.filename, 0644)
c.Check(commands, gc.HasLen, 0)
}
func (s powershellSuite) TestWriteFile(c *gc.C) {
data := []byte("something\nhere\n")
commands := s.renderer.WriteFile(s.filename, data)
expected := `
Set-Content 'C:\some\dir\file' @"
something
here
"@`[1:]
c.Check(commands, jc.DeepEquals, []string{
expected,
})
}
func (s powershellSuite) TestMkdir(c *gc.C) {
commands := s.renderer.Mkdir(s.dirname)
c.Check(commands, jc.DeepEquals, []string{
`mkdir 'C:\some\dir'`,
})
}
func (s powershellSuite) TestMkdirAll(c *gc.C) {
commands := s.renderer.MkdirAll(s.dirname)
c.Check(commands, jc.DeepEquals, []string{
`mkdir 'C:\some\dir'`,
})
}
func (s powershellSuite) TestNewPSEncodedCommand(c *gc.C) {
script := `
Get-WmiObject win32_processor
`
expected := "powershell.exe -Sta -NonInteractive -ExecutionPolicy RemoteSigned -EncodedCommand CgAJAEcAZQB0AC0AVwBtAGkATwBiAGoAZQBjAHQAIAB3AGkAbgAzADIAXwBwAHIAbwBjAGUAcwBzAG8AcgAKAA=="
out, err := shell.NewPSEncodedCommand(script)
c.Assert(err, gc.IsNil)
c.Assert((len(out) > 0), gc.Equals, true)
c.Assert(out, jc.DeepEquals, expected)
}
|
