@@ -10,32 +10,49 @@ import (
1010 "github.com/stretchr/testify/require"
1111)
1212
13- func Test_NewSignal (t * testing.T ) {
14- s , err := NewSignal ("traces" )
15- require .NoError (t , err )
16- assert .Equal (t , Signal {name : "traces" }, s )
13+ func TestSignal_String (t * testing.T ) {
14+ assert .Equal (t , "traces" , SignalTraces .String ())
15+ assert .Equal (t , "metrics" , SignalMetrics .String ())
16+ assert .Equal (t , "logs" , SignalLogs .String ())
17+ assert .Equal (t , "profiles" , SignalProfiles .String ())
1718}
1819
19- func Test_NewSignal_Invalid (t * testing.T ) {
20- _ , err := NewSignal ("" )
21- require .Error (t , err )
22- _ , err = NewSignal ("TRACES" )
23- require .Error (t , err )
24- }
20+ func TestSignal_MarshalText (t * testing.T ) {
21+ b , err := SignalTraces .MarshalText ()
22+ require .NoError (t , err )
23+ assert .Equal (t , []byte ("traces" ), b )
2524
26- func Test_MustNewSignal (t * testing.T ) {
27- s := MustNewSignal ("traces" )
28- assert .Equal (t , Signal {name : "traces" }, s )
29- }
25+ b , err = SignalMetrics .MarshalText ()
26+ require .NoError (t , err )
27+ assert .Equal (t , []byte ("metrics" ), b )
3028
31- func Test_Signal_String (t * testing.T ) {
32- s := MustNewSignal ("traces" )
33- assert .Equal (t , "traces" , s .String ())
34- }
29+ b , err = SignalLogs .MarshalText ()
30+ require .NoError (t , err )
31+ assert .Equal (t , []byte ("logs" ), b )
3532
36- func Test_Signal_MarshalText (t * testing.T ) {
37- s := MustNewSignal ("traces" )
38- b , err := s .MarshalText ()
33+ b , err = SignalProfiles .MarshalText ()
3934 require .NoError (t , err )
40- assert .Equal (t , []byte ("traces" ), b )
35+ assert .Equal (t , []byte ("profiles" ), b )
36+
37+ var s Signal
38+ b , err = s .MarshalText ()
39+ require .NoError (t , err )
40+ assert .Equal (t , []byte ("" ), b )
41+ }
42+
43+ func TestSignal_UnmarshalText (t * testing.T ) {
44+ var s Signal
45+ require .NoError (t , s .UnmarshalText ([]byte ("traces" )))
46+ assert .Equal (t , SignalTraces , s )
47+
48+ require .NoError (t , s .UnmarshalText ([]byte ("metrics" )))
49+ assert .Equal (t , SignalMetrics , s )
50+
51+ require .NoError (t , s .UnmarshalText ([]byte ("logs" )))
52+ assert .Equal (t , SignalLogs , s )
53+
54+ require .NoError (t , s .UnmarshalText ([]byte ("profiles" )))
55+ assert .Equal (t , SignalProfiles , s )
56+
57+ require .Error (t , s .UnmarshalText ([]byte ("unknown" )))
4158}
0 commit comments