This repository was archived by the owner on Apr 2, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 525
/
Copy pathrst_encoder_test.go
107 lines (92 loc) · 3.29 KB
/
rst_encoder_test.go
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/***** BEGIN LICENSE BLOCK *****
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.
#
# The Initial Developer of the Original Code is the Mozilla Foundation.
# Portions created by the Initial Developer are Copyright (C) 2014
# the Initial Developer. All Rights Reserved.
#
# Contributor(s):
# Rob Miller ([email protected])
#
# ***** END LICENSE BLOCK *****/
package plugins
import (
"time"
"github.com/mozilla-services/heka/message"
"github.com/mozilla-services/heka/pipeline"
"github.com/pborman/uuid"
gs "github.com/rafrombrc/gospec/src/gospec"
)
func RstEncoderSpec(c gs.Context) {
c.Specify("A RstEncoder", func() {
encoder := new(RstEncoder)
supply := make(chan *pipeline.PipelinePack, 1)
pack := pipeline.NewPipelinePack(supply)
payload := "This is the payload!"
pack.Message.SetPayload(payload)
loc, err := time.LoadLocation("America/Chicago")
c.Assume(err, gs.IsNil)
timestamp := time.Date(1971, 10, 7, 18, 47, 0, 0, loc)
pack.Message.SetTimestamp(timestamp.UnixNano())
pack.Message.SetType("test.type")
pack.Message.SetHostname("somehost.example.com")
pack.Message.SetPid(12345)
pack.Message.SetUuid(uuid.Parse("72de6a05-1b99-4a88-84c2-90797624c68f"))
pack.Message.SetLogger("loggyloglog")
pack.Message.SetEnvVersion("0.8")
field, err := message.NewField("intfield", 23, "count")
c.Assume(err, gs.IsNil)
field.AddValue(24)
field.AddValue(25)
pack.Message.AddField(field)
field, err = message.NewField("stringfield", "hold", "foreigner")
c.Assume(err, gs.IsNil)
field.AddValue("your")
field.AddValue("head")
field.AddValue("up")
pack.Message.AddField(field)
field, err = message.NewField("bool", true, "")
c.Assume(err, gs.IsNil)
field.AddValue(false)
pack.Message.AddField(field)
field, err = message.NewField("bool", false, "false")
c.Assume(err, gs.IsNil)
pack.Message.AddField(field)
field, err = message.NewField("float", 345.6789, "kelvin")
c.Assume(err, gs.IsNil)
field.AddValue(139847987987987.878732819)
pack.Message.AddField(field)
field, err = message.NewField("bytes", []byte("encode me"), "binary")
c.Assume(err, gs.IsNil)
field.AddValue([]byte("and me"))
field.AddValue([]byte("and me too"))
pack.Message.AddField(field)
expected := `
:Timestamp: 1971-10-07 23:47:00 +0000 UTC
:Type: test.type
:Hostname: somehost.example.com
:Pid: 12345
:Uuid: 72de6a05-1b99-4a88-84c2-90797624c68f
:Logger: loggyloglog
:Payload: This is the payload!
:EnvVersion: 0.8
:Severity: 7
:Fields:
| name:"intfield" type:integer value:[23,24,25] representation:"count"
| name:"stringfield" type:string value:["hold","your","head","up"] representation:"foreigner"
| name:"bool" type:bool value:[true,false]
| name:"bool" type:bool value:false representation:"false"
| name:"float" type:double value:[345.6789,1.3984798798798788e+14] representation:"kelvin"
| name:"bytes" type:bytes value:[ZW5jb2RlIG1l,YW5kIG1l,YW5kIG1lIHRvbw==] representation:"binary"
`
c.Specify("serializes a message correctly", func() {
err = encoder.Init(nil)
c.Assume(err, gs.IsNil)
output, err := encoder.Encode(pack)
c.Expect(err, gs.IsNil)
c.Expect(string(output), gs.Equals, expected)
})
})
}