104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
|
// 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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|