github.com/m3db/m3@v1.5.0/src/dbnode/integration/client_load_schema_test.go (about)

     1  // +build integration
     2  
     3  // Copyright (c) 2019 Uber Technologies, Inc.
     4  //
     5  // Permission is hereby granted, free of charge, to any person obtaining a copy
     6  // of this software and associated documentation files (the "Software"), to deal
     7  // in the Software without restriction, including without limitation the rights
     8  // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
     9  // copies of the Software, and to permit persons to whom the Software is
    10  // furnished to do so, subject to the following conditions:
    11  //
    12  // The above copyright notice and this permission notice shall be included in
    13  // all copies or substantial portions of the Software.
    14  //
    15  // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    16  // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    17  // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    18  // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    19  // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    20  // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
    21  // THE SOFTWARE.
    22  
    23  package integration
    24  
    25  import (
    26  	"io/ioutil"
    27  	"os"
    28  	"path/filepath"
    29  	"testing"
    30  
    31  	"github.com/m3db/m3/src/dbnode/client"
    32  	"github.com/m3db/m3/src/dbnode/namespace"
    33  	"github.com/m3db/m3/src/dbnode/topology"
    34  	"github.com/m3db/m3/src/x/ident"
    35  
    36  	"github.com/golang/mock/gomock"
    37  	"github.com/stretchr/testify/require"
    38  )
    39  
    40  const (
    41  	testProtoStr = `syntax = "proto3";
    42  
    43  package mainpkg;
    44  
    45  message TestMessage {
    46    double latitude = 1;
    47    double longitude = 2;
    48    int64 epoch = 3;
    49    bytes deliveryID = 4;
    50    map<string, string> attributes = 5;
    51  }
    52  `
    53  )
    54  
    55  func TestClientLoadSchemaFromFile(t *testing.T) {
    56  	ctrl := gomock.NewController(t)
    57  	defer ctrl.Finish()
    58  
    59  	tempDir, err := ioutil.TempDir("", "m3db-client-schema")
    60  	require.NoError(t, err)
    61  	defer os.RemoveAll(tempDir)
    62  
    63  	protoFile := filepath.Join(tempDir, "test.proto")
    64  	require.NoError(t, ioutil.WriteFile(protoFile, []byte(testProtoStr), 0644))
    65  
    66  	schemaReg := namespace.NewSchemaRegistry(true, nil)
    67  	protoMsg := "mainpkg.TestMessage"
    68  	require.NoError(t, namespace.LoadSchemaRegistryFromFile(schemaReg, ident.StringID("ns1"), "fromconfig", protoFile, protoMsg))
    69  	expectedDescr, err := schemaReg.GetLatestSchema(ident.StringID("ns1"))
    70  	require.NoError(t, err)
    71  	require.EqualValues(t, protoMsg, expectedDescr.Get().GetFullyQualifiedName())
    72  
    73  	cfg := &client.Configuration{
    74  		Proto: &client.ProtoConfiguration{
    75  			Enabled: true,
    76  			SchemaRegistry: map[string]client.NamespaceProtoSchema{
    77  				"ns1": {MessageName: protoMsg, SchemaFilePath: protoFile},
    78  			},
    79  		},
    80  	}
    81  
    82  	mockTopo := topology.NewMockInitializer(ctrl)
    83  	adminClient, err := cfg.NewAdminClient(client.ConfigurationParameters{TopologyInitializer: mockTopo})
    84  	require.NoError(t, err)
    85  
    86  	descr, err := adminClient.
    87  		Options().
    88  		SchemaRegistry().
    89  		GetLatestSchema(ident.StringID("ns1"))
    90  	require.NoError(t, err)
    91  	require.NotNil(t, descr)
    92  	require.EqualValues(t, expectedDescr.String(), descr.String())
    93  }