github.com/willyham/dosa@v2.3.1-0.20171024181418-1e446d37ee71+incompatible/entity_parser_index_test.go (about) 1 // Copyright (c) 2017 Uber Technologies, Inc. 2 // 3 // Permission is hereby granted, free of charge, to any person obtaining a copy 4 // of this software and associated documentation files (the "Software"), to deal 5 // in the Software without restriction, including without limitation the rights 6 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 // copies of the Software, and to permit persons to whom the Software is 8 // furnished to do so, subject to the following conditions: 9 // 10 // The above copyright notice and this permission notice shall be included in 11 // all copies or substantial portions of the Software. 12 // 13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 // THE SOFTWARE. 20 21 package dosa 22 23 import ( 24 "testing" 25 "time" 26 27 "github.com/stretchr/testify/assert" 28 ) 29 30 type SingleIndexNoParen struct { 31 Entity `dosa:"primaryKey=PrimaryKey"` 32 SearchByData Index `dosa:"key=Data"` 33 PrimaryKey int64 34 Data string 35 } 36 37 func TestSingleIndexNoParen(t *testing.T) { 38 dosaTable, err := TableFromInstance(&SingleIndexNoParen{}) 39 assert.Nil(t, err) 40 assert.Equal(t, map[string]*IndexDefinition{ 41 "searchbydata": { 42 Key: &PrimaryKey{PartitionKeys: []string{"data"}}, 43 }, 44 }, dosaTable.Indexes) 45 } 46 47 type MultipleIndexes struct { 48 Entity `dosa:"primaryKey=PrimaryKey"` 49 Index `dosa:"key=Data, name=SearchByData"` 50 SearchByDate Index `dosa:"key=Date"` 51 PrimaryKey int64 52 Data string 53 Date time.Time 54 } 55 56 func TestMultipleIndexes(t *testing.T) { 57 dosaTable, err := TableFromInstance(&MultipleIndexes{}) 58 assert.Nil(t, err) 59 assert.Equal(t, map[string]*IndexDefinition{ 60 "searchbydata": { 61 Key: &PrimaryKey{PartitionKeys: []string{"data"}}, 62 }, 63 "searchbydate": { 64 Key: &PrimaryKey{PartitionKeys: []string{"date"}}, 65 }, 66 }, dosaTable.Indexes) 67 } 68 69 type ComplexIndexes struct { 70 Entity `dosa:"primaryKey=PrimaryKey"` 71 SearchByData Index `dosa:"key=(Data, Date, PrimaryKey DESC) name=index_data"` 72 SearchByDate Index `dosa:"key=((Date, PrimaryKey), Data) name=index_date"` 73 PrimaryKey int64 74 Data string 75 Date time.Time 76 } 77 78 func TestComplexIndexes(t *testing.T) { 79 dosaTable, err := TableFromInstance(&ComplexIndexes{}) 80 assert.Nil(t, err) 81 assert.Equal(t, map[string]*IndexDefinition{ 82 "index_data": { 83 Key: &PrimaryKey{ 84 PartitionKeys: []string{"data"}, 85 ClusteringKeys: []*ClusteringKey{ 86 { 87 Name: "date", 88 Descending: false, 89 }, 90 { 91 Name: "primarykey", 92 Descending: true, 93 }, 94 }, 95 }, 96 }, 97 "index_date": { 98 Key: &PrimaryKey{ 99 PartitionKeys: []string{"date", "primarykey"}, 100 ClusteringKeys: []*ClusteringKey{ 101 { 102 Name: "data", 103 Descending: false, 104 }, 105 }, 106 }, 107 }, 108 }, dosaTable.Indexes) 109 }