Files
tasks/common/headers/headers_test.go
Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689)
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.

## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
2025-05-01 18:50:21 -07:00

130 lines
3.9 KiB
Go

package headers
import (
"context"
"testing"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
"google.golang.org/grpc/metadata"
)
type (
HeadersSuite struct {
*require.Assertions
suite.Suite
}
)
func TestHeadersSuite(t *testing.T) {
suite.Run(t, &HeadersSuite{})
}
func (s *HeadersSuite) SetupTest() {
s.Assertions = require.New(s.T())
}
func (s *HeadersSuite) TestPropagate_CreateNewOutgoingContext() {
ctx := context.Background()
ctx = metadata.NewIncomingContext(ctx, metadata.New(map[string]string{
ClientVersionHeaderName: "22.08.78",
SupportedServerVersionsHeaderName: ">21.04.16",
ClientNameHeaderName: "28.08.14",
SupportedFeaturesHeaderName: "my-feature",
}))
ctx = Propagate(ctx)
md, ok := metadata.FromOutgoingContext(ctx)
s.True(ok)
s.Equal("22.08.78", md.Get(ClientVersionHeaderName)[0])
s.Equal(">21.04.16", md.Get(SupportedServerVersionsHeaderName)[0])
s.Equal("28.08.14", md.Get(ClientNameHeaderName)[0])
s.Equal("my-feature", md.Get(SupportedFeaturesHeaderName)[0])
}
func (s *HeadersSuite) TestPropagate_CreateNewOutgoingContext_SomeMissing() {
ctx := context.Background()
ctx = metadata.NewIncomingContext(ctx, metadata.New(map[string]string{
ClientVersionHeaderName: "22.08.78",
ClientNameHeaderName: "28.08.14",
}))
ctx = Propagate(ctx)
md, ok := metadata.FromOutgoingContext(ctx)
s.True(ok)
s.Equal("22.08.78", md.Get(ClientVersionHeaderName)[0])
s.Equal(0, len(md.Get(SupportedServerVersionsHeaderName)))
s.Equal("28.08.14", md.Get(ClientNameHeaderName)[0])
s.Equal(0, len(md.Get(SupportedFeaturesHeaderName)))
}
func (s *HeadersSuite) TestPropagate_UpdateExistingEmptyOutgoingContext() {
ctx := context.Background()
ctx = metadata.NewIncomingContext(ctx, metadata.New(map[string]string{
ClientVersionHeaderName: "22.08.78",
SupportedServerVersionsHeaderName: "<21.04.16",
ClientNameHeaderName: "28.08.14",
SupportedFeaturesHeaderName: "my-feature",
}))
ctx = metadata.NewOutgoingContext(ctx, metadata.MD{})
ctx = Propagate(ctx)
md, ok := metadata.FromOutgoingContext(ctx)
s.True(ok)
s.Equal("22.08.78", md.Get(ClientVersionHeaderName)[0])
s.Equal("<21.04.16", md.Get(SupportedServerVersionsHeaderName)[0])
s.Equal("28.08.14", md.Get(ClientNameHeaderName)[0])
s.Equal("my-feature", md.Get(SupportedFeaturesHeaderName)[0])
}
func (s *HeadersSuite) TestPropagate_UpdateExistingNonEmptyOutgoingContext() {
ctx := context.Background()
ctx = metadata.NewIncomingContext(ctx, metadata.New(map[string]string{
ClientVersionHeaderName: "07.08.78", // Must be ignored
SupportedServerVersionsHeaderName: "<07.04.16", // Must be ignored
SupportedFeaturesHeaderName: "my-feature", // Passed through
}))
ctx = metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{
ClientVersionHeaderName: "22.08.78",
SupportedServerVersionsHeaderName: "<21.04.16",
ClientNameHeaderName: "28.08.14",
}))
ctx = Propagate(ctx)
md, ok := metadata.FromOutgoingContext(ctx)
s.True(ok)
s.Equal("22.08.78", md.Get(ClientVersionHeaderName)[0])
s.Equal("<21.04.16", md.Get(SupportedServerVersionsHeaderName)[0])
s.Equal("28.08.14", md.Get(ClientNameHeaderName)[0])
s.Equal("my-feature", md.Get(SupportedFeaturesHeaderName)[0])
}
func (s *HeadersSuite) TestPropagate_EmptyIncomingContext() {
ctx := context.Background()
ctx = metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{
ClientVersionHeaderName: "22.08.78",
SupportedServerVersionsHeaderName: "<21.04.16",
ClientNameHeaderName: "28.08.14",
}))
ctx = Propagate(ctx)
md, ok := metadata.FromOutgoingContext(ctx)
s.True(ok)
s.Equal("22.08.78", md.Get(ClientVersionHeaderName)[0])
s.Equal("<21.04.16", md.Get(SupportedServerVersionsHeaderName)[0])
s.Equal("28.08.14", md.Get(ClientNameHeaderName)[0])
}