mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-12-18 17:08:29 +00:00
89 lines
3.1 KiB
Go
89 lines
3.1 KiB
Go
// Copyright (c) All respective contributors to the Peridot Project. All rights reserved.
|
|
// Copyright (c) 2021-2022 Rocky Enterprise Software Foundation, Inc. All rights reserved.
|
|
// Copyright (c) 2021-2022 Ctrl IQ, Inc. All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are met:
|
|
//
|
|
// 1. Redistributions of source code must retain the above copyright notice,
|
|
// this list of conditions and the following disclaimer.
|
|
//
|
|
// 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
// this list of conditions and the following disclaimer in the documentation
|
|
// and/or other materials provided with the distribution.
|
|
//
|
|
// 3. Neither the name of the copyright holder nor the names of its contributors
|
|
// may be used to endorse or promote products derived from this software without
|
|
// specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
// POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
package rherrata
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/go-chi/chi"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
)
|
|
|
|
type MockInstance struct {
|
|
API *API
|
|
// Mapped to advisory id/name
|
|
HTMLResponses map[string]string
|
|
Advisories *internalAdvisoriesResponse
|
|
TestServerErrata *httptest.Server
|
|
TestServerMock *httptest.Server
|
|
}
|
|
|
|
func NewMock() *MockInstance {
|
|
mockInstance := &MockInstance{
|
|
HTMLResponses: map[string]string{},
|
|
Advisories: &internalAdvisoriesResponse{
|
|
Response: &internalAdvisoriesInnerResponse{
|
|
Docs: []*CompactErrata{},
|
|
},
|
|
},
|
|
}
|
|
|
|
muxErrata := chi.NewMux()
|
|
muxErrata.Get("/{advisory}", func(w http.ResponseWriter, r *http.Request) {
|
|
advisory := chi.URLParam(r, "advisory")
|
|
|
|
if response := mockInstance.HTMLResponses[advisory]; response != "" {
|
|
w.Header().Set("Content-Type", "text/html")
|
|
_, _ = w.Write([]byte(response))
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}
|
|
})
|
|
|
|
muxAPI := chi.NewMux()
|
|
muxAPI.Get("/*", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
_ = json.NewEncoder(w).Encode(mockInstance.Advisories)
|
|
})
|
|
|
|
api := NewClient()
|
|
tsErrata := httptest.NewServer(muxErrata)
|
|
tsAPI := httptest.NewServer(muxAPI)
|
|
api.baseURLErrata = tsErrata.URL
|
|
api.baseURLAPI = tsAPI.URL
|
|
|
|
mockInstance.API = api
|
|
mockInstance.TestServerErrata = tsErrata
|
|
|
|
return mockInstance
|
|
}
|