Merge pull request #98 from ntnn/multi-get-provider

✨ Add multi.Provider.GetProvider

Author: k8s-ci-robot

providers/multi/provider.go

@@ -177,6 +177,14 @@ func (p *Provider) splitClusterName(clusterName string) (string, string) {
 	return parts[0], parts[1]
 }
 
+// GetProvider returns the provider for the given prefix.
+func (p *Provider) GetProvider(prefix string) (multicluster.Provider, bool) {
+	p.lock.RLock()
+	defer p.lock.RUnlock()
+	provider, ok := p.providers[prefix]
+	return provider, ok
+}
+
 // AddProvider adds a new provider with the given prefix.
 //
 // The startFunc is called to start the provider - starting the provider

providers/multi/provider_test.go

@@ -96,8 +96,15 @@ var _ = Describe("Provider Multi", Ordered, func() {
 		})
 
 		By("Adding the first namespace provider before starting the manager", func() {
+			_, ok := provider.GetProvider("cloud1")
+			Expect(ok).To(BeFalse())
+
 			err := provider.AddProvider("cloud1", cloud1provider)
 			Expect(err).NotTo(HaveOccurred())
+
+			returnedProvider, ok := provider.GetProvider("cloud1")
+			Expect(ok).To(BeTrue())
+			Expect(returnedProvider).To(Equal(cloud1provider))
 		})
 
 		By("Setting up the controller feeding the animals", func() {