GaoreGo/goes
7
0
Fork 0
Code Pull Requests Releases 2 Activity

Addition of getConnection() method and ES_HOST/ES_PORT env vars in example_test.go

Browse Source
This commit is contained in:
Amadeo Casas 2014-08-17 13:26:08 -07:00
parent 9b27ec1889
commit 9a6ac4f5c5
1 changed files with 30 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
example_test.go
View File

@ -8,13 +8,35 @@ import (
"fmt" "fmt"
"net/http" "net/http"
"net/url" "net/url"
"os"
"time" "time"
"github.com/belogik/goes" "github.com/belogik/goes"
) )
var (
ES_HOST = "localhost"
ES_PORT = "9200"
)
func getConnection() (conn *goes.Connection) {
h := os.Getenv("TEST_ELASTICSEARCH_HOST")
if h == "" {
h = ES_HOST
}
p := os.Getenv("TEST_ELASTICSEARCH_PORT")
if p == "" {
p = ES_PORT
}
conn = goes.NewConnection(h, p)
return
}
func ExampleConnection_CreateIndex() { func ExampleConnection_CreateIndex() {
conn := goes.NewConnection("localhost", "9200") conn := getConnection()
mapping := map[string]interface{}{ mapping := map[string]interface{}{
"settings": map[string]interface{}{ "settings": map[string]interface{}{
@ -43,7 +65,7 @@ func ExampleConnection_CreateIndex() {
} }
func ExampleConnection_DeleteIndex() { func ExampleConnection_DeleteIndex() {
conn := goes.NewConnection("localhost", "9200") conn := getConnection()
resp, err := conn.DeleteIndex("yourinde") resp, err := conn.DeleteIndex("yourinde")
if err != nil { if err != nil {
@ -54,7 +76,7 @@ func ExampleConnection_DeleteIndex() {
} }
func ExampleConnection_RefreshIndex() { func ExampleConnection_RefreshIndex() {
conn := goes.NewConnection("localhost", "9200") conn := getConnection()
resp, err := conn.RefreshIndex("yourindex") resp, err := conn.RefreshIndex("yourindex")
if err != nil { if err != nil {
@ -65,7 +87,7 @@ func ExampleConnection_RefreshIndex() {
} }
func ExampleConnection_Search() { func ExampleConnection_Search() {
conn := goes.NewConnection("localhost", "9200") conn := getConnection()
var query = map[string]interface{}{ var query = map[string]interface{}{
"query": map[string]interface{}{ "query": map[string]interface{}{
@ -102,7 +124,7 @@ func ExampleConnection_Search() {
} }
func ExampleConnection_Index() { func ExampleConnection_Index() {
conn := goes.NewConnection("localhost", "9200") conn := getConnection()
d := goes.Document{ d := goes.Document{
Index: "twitter", Index: "twitter",
@ -126,7 +148,7 @@ func ExampleConnection_Index() {
} }
func ExampleConnection_Delete() { func ExampleConnection_Delete() {
conn := goes.NewConnection("localhost", "9200") conn := getConnection()
//[create index, index document ...] //[create index, index document ...]
@ -154,7 +176,8 @@ func ExampleConnectionOverrideHttpClient() {
cl := &http.Client{ cl := &http.Client{
Transport: tr, Transport: tr,
} }
conn := goes.NewConnection("localhost", "9200").WithClient(cl) conn := getConnection()
conn.WithClient(cl)
fmt.Printf("%v\n", conn.Client) fmt.Printf("%v\n", conn.Client)
} }