refactor(worker): export worker's LoadConfig

This commit is contained in:
bigeagle 2016-04-28 19:37:29 +08:00
parent 0dcd89da31
commit f8151e689f
No known key found for this signature in database
GPG Key ID: 9171A4571C27920A
2 changed files with 5 additions and 4 deletions

View File

@ -76,7 +76,8 @@ type mirrorConfig struct {
Stage1Profile string `toml:"stage1_profile"` Stage1Profile string `toml:"stage1_profile"`
} }
func loadConfig(cfgFile string) (*Config, error) { // LoadConfig loads configuration
func LoadConfig(cfgFile string) (*Config, error) {
if _, err := os.Stat(cfgFile); err != nil { if _, err := os.Stat(cfgFile); err != nil {
return nil, err return nil, err
} }

View File

@ -54,7 +54,7 @@ exclude_file = "/etc/tunasync.d/fedora-exclude.txt"
` `
Convey("When giving invalid file", t, func() { Convey("When giving invalid file", t, func() {
cfg, err := loadConfig("/path/to/invalid/file") cfg, err := LoadConfig("/path/to/invalid/file")
So(err, ShouldNotBeNil) So(err, ShouldNotBeNil)
So(cfg, ShouldBeNil) So(cfg, ShouldBeNil)
}) })
@ -68,7 +68,7 @@ exclude_file = "/etc/tunasync.d/fedora-exclude.txt"
So(err, ShouldEqual, nil) So(err, ShouldEqual, nil)
defer tmpfile.Close() defer tmpfile.Close()
cfg, err := loadConfig(tmpfile.Name()) cfg, err := LoadConfig(tmpfile.Name())
So(err, ShouldBeNil) So(err, ShouldBeNil)
So(cfg.Global.Name, ShouldEqual, "test_worker") So(cfg.Global.Name, ShouldEqual, "test_worker")
So(cfg.Global.Interval, ShouldEqual, 240) So(cfg.Global.Interval, ShouldEqual, 240)
@ -107,7 +107,7 @@ exclude_file = "/etc/tunasync.d/fedora-exclude.txt"
So(err, ShouldEqual, nil) So(err, ShouldEqual, nil)
defer tmpfile.Close() defer tmpfile.Close()
cfg, err := loadConfig(tmpfile.Name()) cfg, err := LoadConfig(tmpfile.Name())
So(err, ShouldBeNil) So(err, ShouldBeNil)
w := &Worker{ w := &Worker{