mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
zeekayandHanzo Dev
b726a55cab
github.com/argoproj/pkg/v2 was one of two live argoproj dependencies. It is 11 files and 176K total, and we used five packages from it: grpc/http, kubeclientmetrics, stats, sync, time. Vendored into util/vendored/ and the dependency dropped from go.mod, so the gRPC surface (2 references, in grpc-http/forwarders.go) is now ours to remove on our own schedule rather than upstream's. Kubernetes client-go pulls gRPC transitively regardless, so this is isolation — a realistic first step — not a claim that gRPC is gone. Internalized rather than forked to a shared repo because the only other consumer, hanzo/gitops, has zero live deployments and zero dependents. A second repo would be a second thing to version for one caller. NOTICE carries the Apache-2.0 attribution. go build ./... exit 0. Co-authored-by: Hanzo Dev <[email protected]>
104 lines
2.0 KiB
Go
104 lines
2.0 KiB
Go
package http
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type testStruct struct {
|
|
Metadata *testStruct `json:"metadata,omitempty"`
|
|
Spec *testStruct `json:"spec,omitempty"`
|
|
Source *testStruct `json:"source,omitempty"`
|
|
Status *testStruct `json:"status,omitempty"`
|
|
|
|
Name string `json:"name,omitempty"`
|
|
Path string `json:"path,omitempty"`
|
|
Message string `json:"message,omitempty"`
|
|
}
|
|
|
|
var testVal = testStruct{
|
|
Metadata: &testStruct{Name: "test"},
|
|
Spec: &testStruct{
|
|
Source: &testStruct{
|
|
Path: "test_path",
|
|
},
|
|
},
|
|
Status: &testStruct{
|
|
Message: "Failed",
|
|
},
|
|
}
|
|
|
|
func TestMarshalerIncludeFields(t *testing.T) {
|
|
m := messageMarshaler{fields: map[string]interface{}{
|
|
"metadata.name": true,
|
|
"spec.source": true,
|
|
}, exclude: false}
|
|
|
|
out, err := m.Marshal(testVal)
|
|
|
|
require.NoError(t, err)
|
|
assert.JSONEq(t, `{"metadata":{"name":"test"},"spec":{"source":{"path":"test_path"}}}`, string(out))
|
|
}
|
|
|
|
func TestMarshalerExcludeFields(t *testing.T) {
|
|
m := messageMarshaler{fields: map[string]interface{}{
|
|
"metadata.name": true,
|
|
}, exclude: true}
|
|
|
|
out, err := m.Marshal(testVal)
|
|
|
|
require.NoError(t, err)
|
|
assert.JSONEq(t, `{"metadata":{},"spec":{"source":{"path":"test_path"}},"status":{"message":"Failed"}}`, string(out))
|
|
}
|
|
|
|
func TestMarshalerSSE(t *testing.T) {
|
|
m := messageMarshaler{isSSE: true}
|
|
|
|
out, err := m.Marshal(testVal)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, `data: {"metadata":{"name":"test"},"spec":{"source":{"path":"test_path"}},"status":{"message":"Failed"}}
|
|
|
|
`, string(out))
|
|
}
|
|
|
|
var flushed bool
|
|
|
|
type flusher struct {
|
|
w *bufio.Writer
|
|
}
|
|
|
|
func (f *flusher) Flush() {
|
|
err := f.w.Flush()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
flushed = true
|
|
}
|
|
|
|
func TestFlushSuccess(t *testing.T) {
|
|
flushed = false
|
|
|
|
var buf bytes.Buffer
|
|
_, _ = fmt.Fprintf(&buf, "Size: %d MB.", 85)
|
|
f := flusher{w: bufio.NewWriter(&buf)}
|
|
flush(&f)
|
|
|
|
assert.True(t, flushed)
|
|
}
|
|
|
|
func TestFlushFailed(t *testing.T) {
|
|
flushed = false
|
|
|
|
f := flusher{}
|
|
flush(&f)
|
|
|
|
assert.False(t, flushed)
|
|
}
|