← Back to branch summary

~ubuntu-branches/ubuntu/trusty/golang-go-systemd/trusty-proposed

« back to all changes in this revision

Viewing changes to activation/files_test.go

  • Committer: Package Import Robot
  • Author(s): Tianon Gravi
  • Date: 2014-05-08 09:48:52 UTC
  • Revision ID: package-import@ubuntu.com-20140508094852-0csgkii3h2wjlitj
Tags: upstream-1
Import upstream version 1

Show diffs side-by-side

added added

removed removed

Lines of Context:
activation/files_test.go
 
1
/*
 
2
Copyright 2013 CoreOS Inc.
 
3
 
 
4
Licensed under the Apache License, Version 2.0 (the "License");
 
5
you may not use this file except in compliance with the License.
 
6
You may obtain a copy of the License at
 
7
 
 
8
     http://www.apache.org/licenses/LICENSE-2.0
 
9
 
 
10
Unless required by applicable law or agreed to in writing, software
 
11
distributed under the License is distributed on an "AS IS" BASIS,
 
12
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 
13
See the License for the specific language governing permissions and
 
14
limitations under the License.
 
15
*/
 
16
 
 
17
package activation
 
18
 
 
19
import (
 
20
        "bytes"
 
21
        "io"
 
22
        "os"
 
23
        "os/exec"
 
24
        "testing"
 
25
)
 
26
 
 
27
// correctStringWritten fails the text if the correct string wasn't written
 
28
// to the other side of the pipe.
 
29
func correctStringWritten(t *testing.T, r *os.File, expected string) bool {
 
30
        bytes := make([]byte, len(expected))
 
31
        io.ReadAtLeast(r, bytes, len(expected))
 
32
 
 
33
        if string(bytes) != expected {
 
34
                t.Fatalf("Unexpected string %s", string(bytes))
 
35
        }
 
36
 
 
37
        return true
 
38
}
 
39
 
 
40
// TestActivation forks out a copy of activation.go example and reads back two
 
41
// strings from the pipes that are passed in.
 
42
func TestActivation(t *testing.T) {
 
43
        cmd := exec.Command("go", "run", "../examples/activation/activation.go")
 
44
 
 
45
        r1, w1, _ := os.Pipe()
 
46
        r2, w2, _ := os.Pipe()
 
47
        cmd.ExtraFiles = []*os.File{
 
48
                w1,
 
49
                w2,
 
50
        }
 
51
 
 
52
        cmd.Env = os.Environ()
 
53
        cmd.Env = append(cmd.Env, "LISTEN_FDS=2", "FIX_LISTEN_PID=1")
 
54
 
 
55
        err := cmd.Run()
 
56
        if err != nil {
 
57
                t.Fatalf(err.Error())
 
58
        }
 
59
 
 
60
        correctStringWritten(t, r1, "Hello world")
 
61
        correctStringWritten(t, r2, "Goodbye world")
 
62
}
 
63
 
 
64
func TestActivationNoFix(t *testing.T) {
 
65
        cmd := exec.Command("go", "run", "../examples/activation/activation.go")
 
66
        cmd.Env = os.Environ()
 
67
        cmd.Env = append(cmd.Env, "LISTEN_FDS=2")
 
68
 
 
69
        out, _ := cmd.CombinedOutput()
 
70
        if bytes.Contains(out, []byte("No files")) == false {
 
71
                t.Fatalf("Child didn't error out as expected")
 
72
        }
 
73
}
 
74
 
 
75
func TestActivationNoFiles(t *testing.T) {
 
76
        cmd := exec.Command("go", "run", "../examples/activation/activation.go")
 
77
        cmd.Env = os.Environ()
 
78
        cmd.Env = append(cmd.Env, "LISTEN_FDS=0", "FIX_LISTEN_PID=1")
 
79
 
 
80
        out, _ := cmd.CombinedOutput()
 
81
        if bytes.Contains(out, []byte("No files")) == false {
 
82
                t.Fatalf("Child didn't error out as expected")
 
83
        }
 
84
}