| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115 |
- // +build go1.7
- package expression
- import (
- "reflect"
- "strings"
- "testing"
- "github.com/aws/aws-sdk-go/aws"
- "github.com/aws/aws-sdk-go/service/dynamodb"
- )
- type exprErrorMode string
- const (
- noExpressionError exprErrorMode = ""
- // invalidEscChar error will occer if the escape char '$' is either followed
- // by an unsupported character or if the escape char is the last character
- invalidEscChar = "invalid escape"
- // outOfRange error will occur if there are more escaped chars than there are
- // actual values to be aliased.
- outOfRange = "out of range"
- // invalidBuilderOperand error will occur if an invalid operand is used
- // as input for Build()
- invalidExpressionBuildOperand = "BuildOperand error"
- // unsetBuilder error will occur if Build() is called on an unset Builder
- unsetBuilder = "unset parameter: Builder"
- // unsetConditionBuilder error will occur if an unset ConditionBuilder is
- // used in WithCondition()
- unsetConditionBuilder = "unset parameter: ConditionBuilder"
- )
- func TestBuild(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected Expression
- err exprErrorMode
- }{
- {
- name: "condition",
- input: NewBuilder().WithCondition(Name("foo").Equal(Value(5))),
- expected: Expression{
- expressionMap: map[expressionType]string{
- condition: "#0 = :0",
- },
- namesMap: map[string]*string{
- "#0": aws.String("foo"),
- },
- valuesMap: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- },
- },
- {
- name: "projection",
- input: NewBuilder().WithProjection(NamesList(Name("foo"), Name("bar"), Name("baz"))),
- expected: Expression{
- expressionMap: map[expressionType]string{
- projection: "#0, #1, #2",
- },
- namesMap: map[string]*string{
- "#0": aws.String("foo"),
- "#1": aws.String("bar"),
- "#2": aws.String("baz"),
- },
- },
- },
- {
- name: "keyCondition",
- input: NewBuilder().WithKeyCondition(Key("foo").Equal(Value(5))),
- expected: Expression{
- expressionMap: map[expressionType]string{
- keyCondition: "#0 = :0",
- },
- namesMap: map[string]*string{
- "#0": aws.String("foo"),
- },
- valuesMap: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- },
- },
- {
- name: "filter",
- input: NewBuilder().WithFilter(Name("foo").Equal(Value(5))),
- expected: Expression{
- expressionMap: map[expressionType]string{
- filter: "#0 = :0",
- },
- namesMap: map[string]*string{
- "#0": aws.String("foo"),
- },
- valuesMap: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- },
- },
- {
- name: "update",
- input: NewBuilder().WithUpdate(Set(Name("foo"), (Value(5)))),
- expected: Expression{
- expressionMap: map[expressionType]string{
- update: "SET #0 = :0\n",
- },
- namesMap: map[string]*string{
- "#0": aws.String("foo"),
- },
- valuesMap: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- },
- },
- {
- name: "compound",
- input: NewBuilder().
- WithCondition(Name("foo").Equal(Value(5))).
- WithFilter(Name("bar").LessThan(Value(6))).
- WithProjection(NamesList(Name("foo"), Name("bar"), Name("baz"))).
- WithKeyCondition(Key("foo").Equal(Value(5))).
- WithUpdate(Set(Name("foo"), Value(5))),
- expected: Expression{
- expressionMap: map[expressionType]string{
- condition: "#0 = :0",
- filter: "#1 < :1",
- projection: "#0, #1, #2",
- keyCondition: "#0 = :2",
- update: "SET #0 = :3\n",
- },
- namesMap: map[string]*string{
- "#0": aws.String("foo"),
- "#1": aws.String("bar"),
- "#2": aws.String("baz"),
- },
- valuesMap: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- ":1": {
- N: aws.String("6"),
- },
- ":2": {
- N: aws.String("5"),
- },
- ":3": {
- N: aws.String("5"),
- },
- },
- },
- },
- {
- name: "invalid Builder",
- input: NewBuilder().WithCondition(Name("").Equal(Value(5))),
- err: invalidExpressionBuildOperand,
- },
- {
- name: "unset Builder",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- actual, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- }
- })
- }
- }
- func TestCondition(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected *string
- err exprErrorMode
- }{
- {
- name: "condition",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- condition: Name("foo").Equal(Value(5)),
- },
- },
- expected: aws.String("#0 = :0"),
- },
- {
- name: "unset builder",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.Condition()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestFilter(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected *string
- err exprErrorMode
- }{
- {
- name: "filter",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- filter: Name("foo").Equal(Value(5)),
- },
- },
- expected: aws.String("#0 = :0"),
- },
- {
- name: "unset builder",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.Filter()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestProjection(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected *string
- err exprErrorMode
- }{
- {
- name: "projection",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- projection: NamesList(Name("foo"), Name("bar"), Name("baz")),
- },
- },
- expected: aws.String("#0, #1, #2"),
- },
- {
- name: "unset builder",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.Projection()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestKeyCondition(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected *string
- err exprErrorMode
- }{
- {
- name: "keyCondition",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- keyCondition: KeyConditionBuilder{
- operandList: []OperandBuilder{
- KeyBuilder{
- key: "foo",
- },
- ValueBuilder{
- value: 5,
- },
- },
- mode: equalKeyCond,
- },
- },
- },
- expected: aws.String("#0 = :0"),
- },
- {
- name: "empty builder",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.KeyCondition()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestUpdate(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected *string
- err exprErrorMode
- }{
- {
- name: "update",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- update: UpdateBuilder{
- operationList: map[operationMode][]operationBuilder{
- setOperation: {
- {
- name: NameBuilder{
- name: "foo",
- },
- value: ValueBuilder{
- value: 5,
- },
- mode: setOperation,
- },
- },
- },
- },
- },
- },
- expected: aws.String("SET #0 = :0\n"),
- },
- {
- name: "multiple sets",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- update: UpdateBuilder{
- operationList: map[operationMode][]operationBuilder{
- setOperation: {
- {
- name: NameBuilder{
- name: "foo",
- },
- value: ValueBuilder{
- value: 5,
- },
- mode: setOperation,
- },
- {
- name: NameBuilder{
- name: "bar",
- },
- value: ValueBuilder{
- value: 6,
- },
- mode: setOperation,
- },
- {
- name: NameBuilder{
- name: "baz",
- },
- value: ValueBuilder{
- value: 7,
- },
- mode: setOperation,
- },
- },
- },
- },
- },
- },
- expected: aws.String("SET #0 = :0, #1 = :1, #2 = :2\n"),
- },
- {
- name: "unset builder",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.Update()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestNames(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected map[string]*string
- err exprErrorMode
- }{
- {
- name: "projection",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- projection: NamesList(Name("foo"), Name("bar"), Name("baz")),
- },
- },
- expected: map[string]*string{
- "#0": aws.String("foo"),
- "#1": aws.String("bar"),
- "#2": aws.String("baz"),
- },
- },
- {
- name: "aggregate",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- condition: ConditionBuilder{
- operandList: []OperandBuilder{
- NameBuilder{
- name: "foo",
- },
- ValueBuilder{
- value: 5,
- },
- },
- mode: equalCond,
- },
- filter: ConditionBuilder{
- operandList: []OperandBuilder{
- NameBuilder{
- name: "bar",
- },
- ValueBuilder{
- value: 6,
- },
- },
- mode: lessThanCond,
- },
- projection: ProjectionBuilder{
- names: []NameBuilder{
- {
- name: "foo",
- },
- {
- name: "bar",
- },
- {
- name: "baz",
- },
- },
- },
- },
- },
- expected: map[string]*string{
- "#0": aws.String("foo"),
- "#1": aws.String("bar"),
- "#2": aws.String("baz"),
- },
- },
- {
- name: "unset",
- input: Builder{},
- err: unsetBuilder,
- },
- {
- name: "unset ConditionBuilder",
- input: NewBuilder().WithCondition(ConditionBuilder{}),
- err: unsetConditionBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.Names()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestValues(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expected map[string]*dynamodb.AttributeValue
- err exprErrorMode
- }{
- {
- name: "condition",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- condition: Name("foo").Equal(Value(5)),
- },
- },
- expected: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- },
- {
- name: "aggregate",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- condition: ConditionBuilder{
- operandList: []OperandBuilder{
- NameBuilder{
- name: "foo",
- },
- ValueBuilder{
- value: 5,
- },
- },
- mode: equalCond,
- },
- filter: ConditionBuilder{
- operandList: []OperandBuilder{
- NameBuilder{
- name: "bar",
- },
- ValueBuilder{
- value: 6,
- },
- },
- mode: lessThanCond,
- },
- projection: ProjectionBuilder{
- names: []NameBuilder{
- {
- name: "foo",
- },
- {
- name: "bar",
- },
- {
- name: "baz",
- },
- },
- },
- },
- },
- expected: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- ":1": {
- N: aws.String("6"),
- },
- },
- },
- {
- name: "unset",
- input: Builder{},
- err: unsetBuilder,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.Build()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- actual := expr.Values()
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestBuildChildTrees(t *testing.T) {
- cases := []struct {
- name string
- input Builder
- expectedaliasList aliasList
- expectedStringMap map[expressionType]string
- err exprErrorMode
- }{
- {
- name: "aggregate",
- input: Builder{
- expressionMap: map[expressionType]treeBuilder{
- condition: ConditionBuilder{
- operandList: []OperandBuilder{
- NameBuilder{
- name: "foo",
- },
- ValueBuilder{
- value: 5,
- },
- },
- mode: equalCond,
- },
- filter: ConditionBuilder{
- operandList: []OperandBuilder{
- NameBuilder{
- name: "bar",
- },
- ValueBuilder{
- value: 6,
- },
- },
- mode: lessThanCond,
- },
- projection: ProjectionBuilder{
- names: []NameBuilder{
- {
- name: "foo",
- },
- {
- name: "bar",
- },
- {
- name: "baz",
- },
- },
- },
- },
- },
- expectedaliasList: aliasList{
- namesList: []string{"foo", "bar", "baz"},
- valuesList: []dynamodb.AttributeValue{
- {
- N: aws.String("5"),
- },
- {
- N: aws.String("6"),
- },
- },
- },
- expectedStringMap: map[expressionType]string{
- condition: "#0 = :0",
- filter: "#1 < :1",
- projection: "#0, #1, #2",
- },
- },
- {
- name: "unset",
- input: Builder{},
- expectedaliasList: aliasList{},
- expectedStringMap: map[expressionType]string{},
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- actualAL, actualSM, err := c.input.buildChildTrees()
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- }
- if e, a := c.expectedaliasList, actualAL; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- if e, a := c.expectedStringMap, actualSM; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestBuildExpressionString(t *testing.T) {
- cases := []struct {
- name string
- input exprNode
- expectedNames map[string]*string
- expectedValues map[string]*dynamodb.AttributeValue
- expectedExpression string
- err exprErrorMode
- }{
- {
- name: "basic name",
- input: exprNode{
- names: []string{"foo"},
- fmtExpr: "$n",
- },
- expectedValues: map[string]*dynamodb.AttributeValue{},
- expectedNames: map[string]*string{
- "#0": aws.String("foo"),
- },
- expectedExpression: "#0",
- },
- {
- name: "basic value",
- input: exprNode{
- values: []dynamodb.AttributeValue{
- {
- N: aws.String("5"),
- },
- },
- fmtExpr: "$v",
- },
- expectedNames: map[string]*string{},
- expectedValues: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- expectedExpression: ":0",
- },
- {
- name: "nested path",
- input: exprNode{
- names: []string{"foo", "bar"},
- fmtExpr: "$n.$n",
- },
- expectedValues: map[string]*dynamodb.AttributeValue{},
- expectedNames: map[string]*string{
- "#0": aws.String("foo"),
- "#1": aws.String("bar"),
- },
- expectedExpression: "#0.#1",
- },
- {
- name: "nested path with index",
- input: exprNode{
- names: []string{"foo", "bar", "baz"},
- fmtExpr: "$n.$n[0].$n",
- },
- expectedValues: map[string]*dynamodb.AttributeValue{},
- expectedNames: map[string]*string{
- "#0": aws.String("foo"),
- "#1": aws.String("bar"),
- "#2": aws.String("baz"),
- },
- expectedExpression: "#0.#1[0].#2",
- },
- {
- name: "basic size",
- input: exprNode{
- names: []string{"foo"},
- fmtExpr: "size ($n)",
- },
- expectedValues: map[string]*dynamodb.AttributeValue{},
- expectedNames: map[string]*string{
- "#0": aws.String("foo"),
- },
- expectedExpression: "size (#0)",
- },
- {
- name: "duplicate path name",
- input: exprNode{
- names: []string{"foo", "foo"},
- fmtExpr: "$n.$n",
- },
- expectedValues: map[string]*dynamodb.AttributeValue{},
- expectedNames: map[string]*string{
- "#0": aws.String("foo"),
- },
- expectedExpression: "#0.#0",
- },
- {
- name: "equal expression",
- input: exprNode{
- children: []exprNode{
- {
- names: []string{"foo"},
- fmtExpr: "$n",
- },
- {
- values: []dynamodb.AttributeValue{
- {
- N: aws.String("5"),
- },
- },
- fmtExpr: "$v",
- },
- },
- fmtExpr: "$c = $c",
- },
- expectedNames: map[string]*string{
- "#0": aws.String("foo"),
- },
- expectedValues: map[string]*dynamodb.AttributeValue{
- ":0": {
- N: aws.String("5"),
- },
- },
- expectedExpression: "#0 = :0",
- },
- {
- name: "missing char after $",
- input: exprNode{
- names: []string{"foo", "foo"},
- fmtExpr: "$n.$",
- },
- err: invalidEscChar,
- },
- {
- name: "names out of range",
- input: exprNode{
- names: []string{"foo"},
- fmtExpr: "$n.$n",
- },
- err: outOfRange,
- },
- {
- name: "values out of range",
- input: exprNode{
- fmtExpr: "$v",
- },
- err: outOfRange,
- },
- {
- name: "children out of range",
- input: exprNode{
- fmtExpr: "$c",
- },
- err: outOfRange,
- },
- {
- name: "invalid escape char",
- input: exprNode{
- fmtExpr: "$!",
- },
- err: invalidEscChar,
- },
- {
- name: "unset exprNode",
- input: exprNode{},
- expectedExpression: "",
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- expr, err := c.input.buildExpressionString(&aliasList{})
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- if e, a := c.expectedExpression, expr; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- }
- })
- }
- }
- func TestReturnExpression(t *testing.T) {
- cases := []struct {
- name string
- input Expression
- expected *string
- }{
- {
- name: "projection exists",
- input: Expression{
- expressionMap: map[expressionType]string{
- projection: "#0, #1, #2",
- },
- },
- expected: aws.String("#0, #1, #2"),
- },
- {
- name: "projection not exists",
- input: Expression{
- expressionMap: map[expressionType]string{},
- },
- expected: nil,
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- actual := c.input.returnExpression(projection)
- if e, a := c.expected, actual; !reflect.DeepEqual(a, e) {
- t.Errorf("expect %v, got %v", e, a)
- }
- })
- }
- }
- func TestAliasValue(t *testing.T) {
- cases := []struct {
- name string
- input *aliasList
- expected string
- err exprErrorMode
- }{
- {
- name: "first item",
- input: &aliasList{},
- expected: ":0",
- },
- {
- name: "fifth item",
- input: &aliasList{
- valuesList: []dynamodb.AttributeValue{
- {},
- {},
- {},
- {},
- },
- },
- expected: ":4",
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- str, err := c.input.aliasValue(dynamodb.AttributeValue{})
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- if e, a := c.expected, str; e != a {
- t.Errorf("expect %v, got %v", e, a)
- }
- }
- })
- }
- }
- func TestAliasPath(t *testing.T) {
- cases := []struct {
- name string
- inputList *aliasList
- inputName string
- expected string
- err exprErrorMode
- }{
- {
- name: "new unique item",
- inputList: &aliasList{},
- inputName: "foo",
- expected: "#0",
- },
- {
- name: "duplicate item",
- inputList: &aliasList{
- namesList: []string{
- "foo",
- "bar",
- },
- },
- inputName: "foo",
- expected: "#0",
- },
- }
- for _, c := range cases {
- t.Run(c.name, func(t *testing.T) {
- str, err := c.inputList.aliasPath(c.inputName)
- if c.err != noExpressionError {
- if err == nil {
- t.Errorf("expect error %q, got no error", c.err)
- } else {
- if e, a := string(c.err), err.Error(); !strings.Contains(a, e) {
- t.Errorf("expect %q error message to be in %q", e, a)
- }
- }
- } else {
- if err != nil {
- t.Errorf("expect no error, got unexpected Error %q", err)
- }
- if e, a := c.expected, str; e != a {
- t.Errorf("expect %v, got %v", e, a)
- }
- }
- })
- }
- }
|