mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-11-04 07:49:35 +00:00 
			
		
		
		
	fix fake controller source watch behavior
This commit is contained in:
		@@ -158,6 +158,7 @@ func (f *FakeControllerSource) Watch(resourceVersion string) (watch.Interface, e
 | 
				
			|||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return nil, err
 | 
							return nil, err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						rc++ // Don't re-send them a change they already have.
 | 
				
			||||||
	if rc < len(f.changes) {
 | 
						if rc < len(f.changes) {
 | 
				
			||||||
		changes := []watch.Event{}
 | 
							changes := []watch.Event{}
 | 
				
			||||||
		for _, c := range f.changes[rc:] {
 | 
							for _, c := range f.changes[rc:] {
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										48
									
								
								pkg/controller/framework/fake_controller_source_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										48
									
								
								pkg/controller/framework/fake_controller_source_test.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,48 @@
 | 
				
			|||||||
 | 
					/*
 | 
				
			||||||
 | 
					Copyright 2015 Google Inc. All rights reserved.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Licensed under the Apache License, Version 2.0 (the "License");
 | 
				
			||||||
 | 
					you may not use this file except in compliance with the License.
 | 
				
			||||||
 | 
					You may obtain a copy of the License at
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    http://www.apache.org/licenses/LICENSE-2.0
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Unless required by applicable law or agreed to in writing, software
 | 
				
			||||||
 | 
					distributed under the License is distributed on an "AS IS" BASIS,
 | 
				
			||||||
 | 
					WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
				
			||||||
 | 
					See the License for the specific language governing permissions and
 | 
				
			||||||
 | 
					limitations under the License.
 | 
				
			||||||
 | 
					*/
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					package framework
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"testing"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func TestRCNumber(t *testing.T) {
 | 
				
			||||||
 | 
						pod := func(name string) *api.Pod {
 | 
				
			||||||
 | 
							return &api.Pod{
 | 
				
			||||||
 | 
								ObjectMeta: api.ObjectMeta{
 | 
				
			||||||
 | 
									Name: name,
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						source := NewFakeControllerSource()
 | 
				
			||||||
 | 
						source.Add(pod("foo"))
 | 
				
			||||||
 | 
						source.Modify(pod("foo"))
 | 
				
			||||||
 | 
						source.Modify(pod("foo"))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						w, err := source.Watch("1")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							t.Fatalf("Unexpected error: %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						defer w.Stop()
 | 
				
			||||||
 | 
						got := <-w.ResultChan()
 | 
				
			||||||
 | 
						if e, a := "2", got.Object.(*api.Pod).ObjectMeta.ResourceVersion; e != a {
 | 
				
			||||||
 | 
							t.Errorf("wanted %v, got %v", e, a)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
		Reference in New Issue
	
	Block a user