github.com/wanddynosios/cli/v8@v8.7.9-0.20240221182337-1a92e3a7017f/cf/actors/pluginrepo/plugin_repo_test.go

package pluginrepo_test

import (
	"fmt"
	"net/http"
	"net/http/httptest"

	. "code.cloudfoundry.org/cli/cf/actors/pluginrepo"
	"code.cloudfoundry.org/cli/cf/models"
	. "code.cloudfoundry.org/cli/cf/util/testhelpers/matchers"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("PluginRepo", func() {
	var (
		repoActor            PluginRepo
		testServer1CallCount int
		testServer2CallCount int
		testServer1          *httptest.Server
		testServer2          *httptest.Server
	)

	BeforeEach(func() {
		repoActor = NewPluginRepo()
	})

	Context("request data from all repos", func() {
		BeforeEach(func() {
			testServer1CallCount = 0
			h1 := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				defer GinkgoRecover()
				testServer1CallCount++
				fmt.Fprintln(w, `{"plugins":[]}`)
				Expect(r.Header.Get("User-Agent")).NotTo(Equal("Go-http-client/1.1"))
			})
			testServer1 = httptest.NewServer(h1)

			testServer2CallCount = 0
			h2 := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				defer GinkgoRecover()
				testServer2CallCount++
				fmt.Fprintln(w, `{"plugins":[]}`)
				Expect(r.Header.Get("User-Agent")).NotTo(Equal("Go-http-client/1.1"))
			})
			testServer2 = httptest.NewServer(h2)

		})

		AfterEach(func() {
			testServer1.Close()
			testServer2.Close()
		})

		It("make query to all repos listed in config.json", func() {
			repoActor.GetPlugins([]models.PluginRepo{
				{
					Name: "repo1",
					URL:  testServer1.URL,
				},
				{
					Name: "repo2",
					URL:  testServer2.URL,
				},
			})

			Expect(testServer1CallCount).To(Equal(1))
			Expect(testServer2CallCount).To(Equal(1))
		})

		It("lists each of the repos in config.json", func() {
			list, _ := repoActor.GetPlugins([]models.PluginRepo{
				{
					Name: "repo1",
					URL:  testServer1.URL,
				},
				{
					Name: "repo2",
					URL:  testServer2.URL,
				},
			})

			Expect(list["repo1"]).NotTo(BeNil())
			Expect(list["repo2"]).NotTo(BeNil())
		})

	})

	Context("Getting data from repos", func() {
		Context("When data is valid", func() {
			BeforeEach(func() {
				h1 := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
					defer GinkgoRecover()
					fmt.Fprintln(w, `{"plugins":[
						{
							"name":"plugin1",
							"description":"none",
							"version":"1.3.4",
							"binaries":[
								{
									"platform":"osx",
									"url":"https://github.com/simonleung8/cli-plugin-echo/raw/master/bin/osx/echo",
									"checksum":"2a087d5cddcfb057fbda91e611c33f46"
								}
							]
						},
						{
							"name":"plugin2",
							"binaries":[
								{
									"platform":"windows",
									"url":"http://going.no.where",
									"checksum":"abcdefg"
								}
							]
						}]
					}`)
					Expect(r.Header.Get("User-Agent")).NotTo(Equal("Go-http-client/1.1"))
				})
				testServer1 = httptest.NewServer(h1)

			})

			AfterEach(func() {
				testServer1.Close()
			})

			It("lists the info for each plugin", func() {
				list, _ := repoActor.GetPlugins([]models.PluginRepo{
					{
						Name: "repo1",
						URL:  testServer1.URL,
					},
				})

				Expect(list["repo1"]).NotTo(BeNil())
				Expect(len(list["repo1"])).To(Equal(2))

				Expect(list["repo1"][0].Name).To(Equal("plugin1"))
				Expect(list["repo1"][0].Description).To(Equal("none"))
				Expect(list["repo1"][0].Version).To(Equal("1.3.4"))
				Expect(list["repo1"][0].Binaries[0].Platform).To(Equal("osx"))
				Expect(list["repo1"][1].Name).To(Equal("plugin2"))
				Expect(list["repo1"][1].Binaries[0].Platform).To(Equal("windows"))
			})

		})
	})

	Context("When data is invalid", func() {
		Context("json is invalid", func() {
			BeforeEach(func() {
				h1 := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
					defer GinkgoRecover()
					fmt.Fprintln(w, `"plugins":[]}`)
					Expect(r.Header.Get("User-Agent")).NotTo(Equal("Go-http-client/1.1"))
				})
				testServer1 = httptest.NewServer(h1)
			})

			AfterEach(func() {
				testServer1.Close()
			})

			It("informs user of invalid json", func() {
				_, err := repoActor.GetPlugins([]models.PluginRepo{
					{
						Name: "repo1",
						URL:  testServer1.URL,
					},
				})

				Expect(err).To(ContainSubstrings(
					[]string{"Invalid json data"},
				))

			})

		})

		Context("when data is valid json, but not valid plugin repo data", func() {
			BeforeEach(func() {
				h1 := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
					defer GinkgoRecover()
					fmt.Fprintln(w, `{"bad_plugin_tag":[]}`)
					Expect(r.Header.Get("User-Agent")).NotTo(Equal("Go-http-client/1.1"))
				})
				testServer1 = httptest.NewServer(h1)
			})

			AfterEach(func() {
				testServer1.Close()
			})

			It("informs user of invalid repo data", func() {
				_, err := repoActor.GetPlugins([]models.PluginRepo{
					{
						Name: "repo1",
						URL:  testServer1.URL,
					},
				})

				Expect(err).To(ContainSubstrings(
					[]string{"Invalid data", "plugin data does not exist"},
				))

			})

		})
	})
})