ridv/gorealis
1
0
Fork 0
Code Issues Pull requests Projects Releases 23 Packages Wiki Activity Actions

Changed from the old style of creating clients to the new clojure pattern.

Browse source
This commit is contained in:
Renan DelValle 2017-09-28 17:36:41 -07:00
parent bf354bcc0a
commit 7db2395df1
1 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

19
realis_e2e_test.go
View file

@ -12,11 +12,12 @@
* limitations under the License. * limitations under the License.
*/ */
package realis package realis_test
import ( import (
"fmt" "fmt"
"github.com/rdelval/gorealis/gen-go/apache/aurora" "github.com/rdelval/gorealis/gen-go/apache/aurora"
"github.com/rdelval/gorealis"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"io/ioutil" "io/ioutil"
"os" "os"
@ -24,15 +25,19 @@ import (
"time" "time"
) )
var r Realis var r realis.Realis
var thermosPayload []byte var thermosPayload []byte
func TestMain(m *testing.M) { func TestMain(m *testing.M) {
var err error var err error
// New configuration to connect to Vagrant image // New configuration to connect to Vagrant image
r, err = NewDefaultClientUsingUrl("http://192.168.33.7:8081","aurora", "secret") r, err = realis.NewRealisClient(realis.SchedulerUrl("http://192.168.33.7:8081"),
if err != nil { realis.BasicAuth("aurora", "secret"),
realis.ThriftJSON(),
realis.TimeoutMS(20000),
realis.BackOff(&realis.Backoff{Steps: 2, Duration: 10 * time.Second, Factor: 2.0, Jitter: 0.1}))
if err != nil {
fmt.Println("Please run vagrant box before running test suite") fmt.Println("Please run vagrant box before running test suite")
os.Exit(1) os.Exit(1)
} }
@ -48,10 +53,10 @@ func TestMain(m *testing.M) {
func TestRealisClient_CreateJob_Thermos(t *testing.T) { func TestRealisClient_CreateJob_Thermos(t *testing.T) {
job := NewJob(). job := realis.NewJob().
Environment("prod"). Environment("prod").
Role("vagrant"). Role("vagrant").
Name("create_job_test"). Name("create_thermos_job_test").
ExecutorName(aurora.AURORA_EXECUTOR_NAME). ExecutorName(aurora.AURORA_EXECUTOR_NAME).
ExecutorData(string(thermosPayload)). ExecutorData(string(thermosPayload)).
CPU(1). CPU(1).
@ -95,7 +100,7 @@ func TestRealisClient_ScheduleCronJob_Thermos(t *testing.T) {
os.Exit(1) os.Exit(1)
} }
job := NewJob(). job := realis.NewJob().
Environment("prod"). Environment("prod").
Role("vagrant"). Role("vagrant").
Name("cronsched_job_test"). Name("cronsched_job_test").