@@ -121,8 +121,7 @@ def test_output_members(self):
121121 with click .Context (click .Command ('list' )) as ctx :
122122 ctx .obj = {'__config' : {}, '__mpp' : get_mpp ({})}
123123 scheduled_at = datetime .now (tzutc ) + timedelta (seconds = 600 )
124- target_site = 'baz'
125- cluster = get_cluster_initialized_with_leader (Failover (1 , 'foo' , 'bar' , scheduled_at , target_site ))
124+ cluster = get_cluster_initialized_with_leader (Failover (1 , 'foo' , 'bar' , scheduled_at ))
126125 del cluster .members [1 ].data ['conn_url' ]
127126 for fmt in ('pretty' , 'json' , 'yaml' , 'topology' ):
128127 self .assertIsNone (output_members (cluster , name = 'abc' , fmt = fmt ))
@@ -556,10 +555,8 @@ def test_flush_switchover(self):
556555 assert 'No pending scheduled switchover' in result .output
557556
558557 scheduled_at = datetime .now (tzutc ) + timedelta (seconds = 600 )
559- target_site = 'baz'
560558 with patch ('patroni.dcs.AbstractDCS.get_cluster' ,
561- Mock (return_value = get_cluster_initialized_with_leader (Failover (1 , 'a' , 'b' , scheduled_at ,
562- target_site )))):
559+ Mock (return_value = get_cluster_initialized_with_leader (Failover (1 , 'a' , 'b' , scheduled_at )))):
563560 result = self .runner .invoke (ctl , ['-k' , 'flush' , 'dummy' , 'switchover' ])
564561 assert result .output .startswith ('Success: ' )
565562
0 commit comments