| // Copyright 2015 Google Inc. All Rights Reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package bigquery |
| |
| import ( |
| "reflect" |
| "testing" |
| "time" |
| |
| "golang.org/x/net/context" |
| bq "google.golang.org/api/bigquery/v2" |
| ) |
| |
| type createTableRecorder struct { |
| conf *createTableConf |
| service |
| } |
| |
| func (rec *createTableRecorder) createTable(ctx context.Context, conf *createTableConf) error { |
| rec.conf = conf |
| return nil |
| } |
| |
| func TestCreateTableOptions(t *testing.T) { |
| s := &createTableRecorder{} |
| c := &Client{ |
| projectID: "p", |
| service: s, |
| } |
| ds := c.Dataset("d") |
| table := ds.Table("t") |
| exp := time.Now() |
| q := "query" |
| if err := table.Create(context.Background(), TableExpiration(exp), ViewQuery(q), UseStandardSQL()); err != nil { |
| t.Fatalf("err calling Table.Create: %v", err) |
| } |
| want := createTableConf{ |
| projectID: "p", |
| datasetID: "d", |
| tableID: "t", |
| expiration: exp, |
| viewQuery: q, |
| useStandardSQL: true, |
| } |
| if !reflect.DeepEqual(*s.conf, want) { |
| t.Errorf("createTableConf: got:\n%v\nwant:\n%v", *s.conf, want) |
| } |
| |
| sc := Schema{fieldSchema("desc", "name", "STRING", false, true)} |
| if err := table.Create(context.Background(), TableExpiration(exp), sc); err != nil { |
| t.Fatalf("err calling Table.Create: %v", err) |
| } |
| want = createTableConf{ |
| projectID: "p", |
| datasetID: "d", |
| tableID: "t", |
| expiration: exp, |
| // No need for an elaborate schema, that is tested in schema_test.go. |
| schema: &bq.TableSchema{ |
| Fields: []*bq.TableFieldSchema{ |
| bqTableFieldSchema("desc", "name", "STRING", "REQUIRED"), |
| }, |
| }, |
| } |
| if !reflect.DeepEqual(*s.conf, want) { |
| t.Errorf("createTableConf: got:\n%v\nwant:\n%v", *s.conf, want) |
| } |
| |
| partitionCases := []struct { |
| timePartitioning TimePartitioning |
| expectedExpiration time.Duration |
| }{ |
| {TimePartitioning{}, time.Duration(0)}, |
| {TimePartitioning{time.Second}, time.Second}, |
| } |
| |
| for _, c := range partitionCases { |
| if err := table.Create(context.Background(), c.timePartitioning); err != nil { |
| t.Fatalf("err calling Table.Create: %v", err) |
| } |
| want = createTableConf{ |
| projectID: "p", |
| datasetID: "d", |
| tableID: "t", |
| timePartitioning: &TimePartitioning{c.expectedExpiration}, |
| } |
| if !reflect.DeepEqual(*s.conf, want) { |
| t.Errorf("createTableConf: got:\n%v\nwant:\n%v", *s.conf, want) |
| } |
| } |
| } |