2014-07-25 06:03:41 +00:00
|
|
|
package connector
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"testing"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func TestCopyStatement(t *testing.T) {
|
2014-12-07 21:51:07 +00:00
|
|
|
e := RedshiftBasicEmtitter{
|
2014-07-25 06:03:41 +00:00
|
|
|
Delimiter: ",",
|
|
|
|
|
S3Bucket: "test_bucket",
|
|
|
|
|
TableName: "test_table",
|
|
|
|
|
}
|
|
|
|
|
f := e.copyStatement("/test.txt")
|
|
|
|
|
|
|
|
|
|
copyStatement := "COPY test_table FROM 's3://test_bucket/test.txt' CREDENTIALS 'aws_access_key_id=;aws_secret_access_key=' DELIMITER ',';"
|
|
|
|
|
|
|
|
|
|
if f != copyStatement {
|
|
|
|
|
t.Errorf("copyStatement() = %s want %s", f, copyStatement)
|
|
|
|
|
}
|
|
|
|
|
}
|