public_git
/
postgres-operator
mirror of https://github.com/zalando/postgres-operator.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix patroni test with mock.

This commit is contained in:
Jan Mußler 2021-02-23 13:20:51 +01:00
parent 3e0a5d03a9
commit 510985ffd8
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/util/patroni/patroni.go
View File

@ -144,7 +144,7 @@ type MemberData struct {
State string `json:"state"` State string `json:"state"`
Role string `json:"role"` Role string `json:"role"`
ServerVersion int `json:"server_version"` ServerVersion int `json:"server_version"`
PendingRestart string `json:"pending_restart"` PendingRestart bool `json:"pending_restart"`
ClusterUnlocked bool `json:"cluster_unlocked"` ClusterUnlocked bool `json:"cluster_unlocked"`
Patroni MemberDataPatroni `json:"patroni"` Patroni MemberDataPatroni `json:"patroni"`
} }

11
pkg/util/patroni/patroni_test.go
View File

@ -93,13 +93,18 @@ func TestPatroniAPI(t *testing.T) {
} }
mockClient := mocks.NewMockHTTPClient(ctrl) mockClient := mocks.NewMockHTTPClient(ctrl)
mockClient.EXPECT().Get(gomock.Any()).Return(&response) mockClient.EXPECT().Get(gomock.Any()).Return(&response, nil)
p := New(nil, mockClient) p := New(nil, mockClient)
_, err := p.GetMemberData(nil) pod := v1.Pod{
Status: v1.PodStatus{
PodIP: "192.168.100.1",
},
}
_, err := p.GetMemberData(&pod)
if err != nil { if err != nil {
t.Errorf("Could not read Patroni data") t.Errorf("Could not read Patroni data: %v", err)
} }
} }