Add unit tests in desired_state_test.go and envelope_test.go

[k-gostev]

Unit tests in desired_state_test.go and envelope_test.go should be added.