Example #1
0
func newConveyor(t testing.TB) *Conveyor {
	db := sqlx.MustConnect("postgres", databaseURL)
	if err := core.Reset(db); err != nil {
		t.Fatal(err)
	}

	c := core.New(db)
	c.BuildQueue = core.NewBuildQueue(100)
	c.Logger = logs.Discard

	ch := make(chan core.BuildContext)
	c.BuildQueue.Subscribe(ch)

	w := worker.New(c, worker.Options{
		Builder: builder.BuilderFunc(func(ctx context.Context, w io.Writer, options builder.BuildOptions) (string, error) {
			io.WriteString(w, "Pulling base image\n")
			return "remind101/acme-inc:1234", nil
		}),
		BuildRequests: ch,
	})

	go w.Start()

	return &Conveyor{
		Conveyor: c,
		worker:   w,
	}
}
Example #2
0
func TestServer_Push_Fork(t *testing.T) {
	var called bool
	b := func(ctx context.Context, w io.Writer, opts builder.BuildOptions) (string, error) {
		called = true
		return "", nil
	}
	s := NewServer(New(builder.BuilderFunc(b)))

	resp := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/", strings.NewReader(`{
  "ref": "refs/heads/master",
  "head_commit": {
    "id": "abcd"
  },
  "repository": {
    "full_name": "remind101/acme-inc",
    "fork": true
  }
}`))
	req.Header.Set("X-GitHub-Event", "push")

	s.ServeHTTP(resp, req)

	if resp.Code != http.StatusOK {
		t.Fatal("Expected 200 OK")
	}

	if called {
		t.Fatal("Expected builder to have not been called")
	}
}
Example #3
0
func TestBuilder_Build(t *testing.T) {
	c := new(mockStatsdClient)
	b := &Builder{
		Builder: builder.BuilderFunc(func(ctx context.Context, w io.Writer, options builder.BuildOptions) (string, error) {
			return "remind101/acme-inc:1234", nil
		}),
		statsd: c,
	}

	c.On("TimeInMilliseconds",
		"conveyor.build.time",
		float64(1000),
		[]string{"repo:remind101/acme-inc"},
		float64(1),
	).Return(nil)
	c.On("Event", &statsd.Event{
		Title: "Conveyor built remind101/acme-inc:1234",
		Text:  "Built remind101/acme-inc:1234 from remind101/acme-inc@master",
		Tags: []string{
			"repo:remind101/acme-inc",
			"branch:master",
			"sha:1234",
			"image:remind101/acme-inc:1234",
		},
	}).Return(nil)

	_, err := b.Build(context.Background(), ioutil.Discard, builder.BuildOptions{
		Repository: "remind101/acme-inc",
		Branch:     "master",
		Sha:        "1234",
	})
	assert.NoError(t, err)
}
Example #4
0
func TestServer_Ping(t *testing.T) {
	b := func(ctx context.Context, w io.Writer, opts builder.BuildOptions) (string, error) {
		return "", nil
	}
	s := NewServer(New(builder.BuilderFunc(b)))

	resp := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/", nil)
	req.Header.Set("X-GitHub-Event", "ping")

	s.ServeHTTP(resp, req)

	if resp.Code != http.StatusOK {
		t.Fatal("Expected 200 OK")
	}
}
Example #5
0
func TestBuilder(t *testing.T) {
	c, err := statsd.New("localhost:8125")
	assert.NoError(t, err)

	b := datadog.WithStats(
		builder.BuilderFunc(func(ctx context.Context, w io.Writer, options builder.BuildOptions) (string, error) {
			return "remind101/acme-inc:1234", nil
		}),
		c,
	)

	_, err = b.Build(context.Background(), ioutil.Discard, builder.BuildOptions{
		Repository: "remind101/acme-inc",
		Branch:     "master",
		Sha:        "1234",
	})
	assert.NoError(t, err)
}
Example #6
0
func TestServer_Push(t *testing.T) {
	called := make(chan struct{})
	b := func(ctx context.Context, w io.Writer, opts builder.BuildOptions) (string, error) {
		close(called)
		expected := builder.BuildOptions{
			Repository: "remind101/acme-inc",
			Branch:     "master",
			Sha:        "abcd",
		}
		if got, want := opts, expected; got != want {
			t.Fatalf("BuildOptions => %v; want %v", got, want)
		}
		return "", nil
	}
	s := NewServer(New(builder.BuilderFunc(b)))

	resp := httptest.NewRecorder()
	req, _ := http.NewRequest("POST", "/", strings.NewReader(`{
  "ref": "refs/heads/master",
  "head_commit": {
    "id": "abcd"
  },
  "repository": {
    "full_name": "remind101/acme-inc"
  }
}`))
	req.Header.Set("X-GitHub-Event", "push")

	s.ServeHTTP(resp, req)

	if resp.Code != http.StatusOK {
		t.Fatal("Expected 200 OK")
	}

	select {
	case <-called:
	case <-time.After(time.Second):
		t.Fatal("Expected builder to have been called")
	}
}
Example #7
0
func TestBuilder_Build_Err(t *testing.T) {
	errBoom := errors.New("container returned non-zero exit")

	c := new(mockStatsdClient)
	b := &Builder{
		Builder: builder.BuilderFunc(func(ctx context.Context, w io.Writer, options builder.BuildOptions) (string, error) {
			return "", errBoom
		}),
		statsd: c,
	}

	c.On("Count",
		"conveyor.build.error",
		int64(1),
		[]string{"repo:remind101/acme-inc"},
		float64(1),
	).Return(nil)

	_, err := b.Build(context.Background(), ioutil.Discard, builder.BuildOptions{
		Repository: "remind101/acme-inc",
	})
	assert.Equal(t, err, errBoom)
}