Function TestBasicHandler_serveUpload

internal/route/lfs/basic_test.go:137–197

(t *testing.T)

}

func TestBasicHandler_serveUpload(t *testing.T) {
	s := &mockStorage{buf: &bytes.Buffer{}}
	basic := &basicHandler{
		defaultStorage: s.Storage(),
		storagers: map[lfsutil.Storage]lfsutil.Storager{
			s.Storage(): s,
		},
	}

	m := macaron.New()
	m.Use(macaron.Renderer())
	m.Use(func(c *macaron.Context) {
		c.Map(&database.Repository{Name: "repo"})
		c.Map(lfsutil.OID("ef797c8118f02dfb649607dd5d3f8c7623048c9c063d532cc95c5ed7a898a64f"))
	})
	m.Put("/", basic.serveUpload)

	tests := []struct {
		name          string
		mockStore     func() *MockStore
		expStatusCode int
		expBody       string
	}{
		{
			name: "object already exists",
			mockStore: func() *MockStore {
				mockStore := NewMockStore()
				mockStore.GetLFSObjectByOIDFunc.SetDefaultReturn(&database.LFSObject{}, nil)
				return mockStore
			},
			expStatusCode: http.StatusOK,
		},
		{
			name: "new object",
			mockStore: func() *MockStore {
				mockStore := NewMockStore()
				mockStore.GetLFSObjectByOIDFunc.SetDefaultReturn(nil, database.ErrLFSObjectNotExist{})
				return mockStore
			},
			expStatusCode: http.StatusOK,
		},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			basic.store = test.mockStore()

			r, err := http.NewRequest("PUT", "/", strings.NewReader("Hello world!"))
			require.NoError(t, err)

			rr := httptest.NewRecorder()
			m.ServeHTTP(rr, r)

			resp := rr.Result()
			assert.Equal(t, test.expStatusCode, resp.StatusCode)

			body, err := io.ReadAll(resp.Body)
			require.NoError(t, err)
			assert.Equal(t, test.expBody, string(body))