From fc713a1763b8eb2137768e1306116a7670578554 Mon Sep 17 00:00:00 2001 From: chlins Date: Tue, 25 Jun 2019 15:51:36 +0800 Subject: [PATCH] refactor: fix unit tests Signed-off-by: chlins --- src/replication/adapter/huawei/huawei_adapter.go | 5 ++++- src/replication/adapter/huawei/image_registry_test.go | 7 ++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/replication/adapter/huawei/huawei_adapter.go b/src/replication/adapter/huawei/huawei_adapter.go index 49a60cd73..fc655e6ea 100644 --- a/src/replication/adapter/huawei/huawei_adapter.go +++ b/src/replication/adapter/huawei/huawei_adapter.go @@ -218,7 +218,10 @@ func AdapterFactory(registry *model.Registry) (adp.Adapter, error) { } var ( - modifiers []modifier.Modifier + modifiers = []modifier.Modifier{ + &auth.UserAgentModifier{ + UserAgent: adp.UserAgentReplication, + }} authorizer modifier.Modifier ) if registry.Credential != nil { diff --git a/src/replication/adapter/huawei/image_registry_test.go b/src/replication/adapter/huawei/image_registry_test.go index 8eff4b420..94e38908a 100644 --- a/src/replication/adapter/huawei/image_registry_test.go +++ b/src/replication/adapter/huawei/image_registry_test.go @@ -1,6 +1,7 @@ package huawei import ( + "os" "strings" "testing" @@ -20,7 +21,11 @@ func init() { Insecure: false, Status: "", } - HWAdapter.registry = hwRegistry + adp, err := AdapterFactory(hwRegistry) + if err != nil { + os.Exit(1) + } + HWAdapter = *adp.(*adapter) } func TestAdapter_FetchImages(t *testing.T) {