go.fuchsia.dev/infra@v0.0.0-20240507153436-9b593402251b/cmd/bigquery/query_test.go

// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package main

import (
	"bytes"
	"encoding/json"
	"os"
	"path/filepath"
	"testing"

	"cloud.google.com/go/bigquery"
	"github.com/google/go-cmp/cmp"
)

func TestWriteOutput(t *testing.T) {
	t.Run("writes to json output path", func(t *testing.T) {
		rows := []map[string]bigquery.Value{
			{
				"number": float64(1), // All numbers are coerced to floats by round-tripping through JSON.
				"bool":   true,
				"string": "foo",
			},
		}

		outputPath := filepath.Join(t.TempDir(), "output.json")
		if err := writeOutput(rows, outputPath, nil); err != nil {
			t.Fatal(err)
		}

		b, err := os.ReadFile(outputPath)
		if err != nil {
			t.Fatal(err)
		}

		var outputRows []map[string]bigquery.Value
		if err := json.Unmarshal(b, &outputRows); err != nil {
			t.Fatal(err)
		}

		if diff := cmp.Diff(rows, outputRows); diff != "" {
			t.Fatalf("got wrong output rows (-want +got)\n%s", diff)
		}
	})

	t.Run("writes to stdout if json output path is unset", func(t *testing.T) {
		rows := []map[string]bigquery.Value{
			{"a": "foo", "b": "bar"},
			{"c": "foo", "d": "bar"},
		}

		stdout := bytes.NewBuffer([]byte{})
		if err := writeOutput(rows, "", stdout); err != nil {
			t.Fatal(err)
		}

		var outputRows []map[string]bigquery.Value
		if err := json.Unmarshal(stdout.Bytes(), &outputRows); err != nil {
			t.Fatal(err)
		}

		if diff := cmp.Diff(rows, outputRows); diff != "" {
			t.Fatalf("got wrong output rows (-want +got)\n%s", diff)
		}
	})
}