@@ -16,31 +16,35 @@ func TestScript(t *testing.T) {
16
16
data := make (map [string ]interface {})
17
17
data ["ami" ] = isAMIAvailable ("aws" , "us-east-1" )
18
18
19
- script , _ := renderScript (data )
19
+ script , err := renderScript (data )
20
+ assert .Nil (t , err )
20
21
assert .NotContains (t , script , "sudo ubuntu-drivers autoinstall" )
21
22
})
22
23
23
24
t .Run ("AWS unknown region should add the NVIDA drivers" , func (t * testing.T ) {
24
25
data := make (map [string ]interface {})
25
26
data ["ami" ] = isAMIAvailable ("aws" , "us-east-99" )
26
27
27
- script , _ := renderScript (data )
28
+ script , err := renderScript (data )
29
+ assert .Nil (t , err )
28
30
assert .Contains (t , script , "sudo ubuntu-drivers autoinstall" )
29
31
})
30
32
31
33
t .Run ("Azure known region should add the NVIDA drivers" , func (t * testing.T ) {
32
34
data := make (map [string ]interface {})
33
35
data ["ami" ] = isAMIAvailable ("azure" , "westus" )
34
36
35
- script , _ := renderScript (data )
37
+ script , err := renderScript (data )
38
+ assert .Nil (t , err )
36
39
assert .Contains (t , script , "sudo ubuntu-drivers autoinstall" )
37
40
})
38
41
39
42
t .Run ("Azure unknown region should add the NVIDA drivers" , func (t * testing.T ) {
40
43
data := make (map [string ]interface {})
41
44
data ["ami" ] = isAMIAvailable ("azure" , "us-east-99" )
42
45
43
- script , _ := renderScript (data )
46
+ script , err := renderScript (data )
47
+ assert .Nil (t , err )
44
48
assert .Contains (t , script , "sudo ubuntu-drivers autoinstall" )
45
49
})
46
50
@@ -49,7 +53,8 @@ func TestScript(t *testing.T) {
49
53
startupScript , _ := base64 .StdEncoding .DecodeString ("ZWNobyAiaGVsbG8gd29ybGQiCmVjaG8gImJ5ZSB3b3JsZCI=" )
50
54
data ["runner_startup_script" ] = string (startupScript )
51
55
52
- script , _ := renderScript (data )
56
+ script , err := renderScript (data )
57
+ assert .Nil (t , err )
53
58
assert .Contains (t , script , "echo \" hello world\" \n echo \" bye world\" " )
54
59
})
55
60
}
0 commit comments