forked from DanielKrawisz/bmd
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix bug that prevented default config file from being loaded.
Tests for LoadConfig().
- Loading branch information
1 parent
bb52081
commit 13934c2
Showing
4 changed files
with
211 additions
and
27 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,174 @@ | ||
| // Copyright 2016 Daniel Krawisz. | ||
| // Use of this source code is governed by an ISC | ||
| // license that can be found in the LICENSE file. | ||
|
|
||
| package main | ||
|
|
||
| import ( | ||
| "testing" | ||
| "io/ioutil" | ||
| "os" | ||
| "strconv" | ||
| "fmt" | ||
| ) | ||
|
|
||
| // If config files exist while we are doing | ||
| var oldDefaultConfigFile []byte = nil | ||
| var oldConfigFile []byte = nil | ||
| var oldConfigFilename *string = nil | ||
|
|
||
| func setup(defaultConfigContents, configFileContents, configFilename *string) error { | ||
| var err error | ||
|
|
||
| // Check if a default config file exists. If so, save it and remove it. | ||
| if _, err = os.Stat(defaultConfigFile); !os.IsNotExist(err) { | ||
| oldDefaultConfigFile, err = ioutil.ReadFile(defaultConfigFile) | ||
|
|
||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| err = os.Remove(defaultConfigFile) | ||
| if err != nil { | ||
| oldDefaultConfigFile = nil | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| // Check if defaultConfigContents is set. If so, make a config file. | ||
| if defaultConfigContents != nil { | ||
| err = ioutil.WriteFile(defaultConfigFile, []byte(*defaultConfigContents), 0644) | ||
| if err != nil { | ||
| cleanup() | ||
| return nil | ||
| } | ||
| } | ||
|
|
||
| // Check if configFilePath is set and is not equal to the default | ||
| // path. | ||
| if configFilename == nil || *configFilename == defaultConfigFile { | ||
| return nil | ||
| } else { | ||
| oldConfigFilename = configFilename | ||
| } | ||
|
|
||
| // If the file exists, save it. | ||
| if _, err = os.Stat(*configFilename); !os.IsNotExist(err) { | ||
| oldConfigFile, err = ioutil.ReadFile(*configFilename) | ||
|
|
||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| err = os.Remove(*configFilename) | ||
| if err != nil { | ||
| oldConfigFile = nil | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| if configFileContents != nil { | ||
| err = ioutil.WriteFile(*configFilename, []byte(*configFileContents), 0644) | ||
| if err != nil { | ||
| cleanup() | ||
| return nil | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func cleanup() { | ||
| if oldConfigFile == nil { | ||
| if _, err := os.Stat(defaultConfigFile); !os.IsNotExist(err) { | ||
| os.Remove(defaultConfigFile) | ||
| } | ||
| } else { | ||
| ioutil.WriteFile(defaultConfigFile, oldDefaultConfigFile, 0644) | ||
| } | ||
|
|
||
| if oldConfigFilename != nil { | ||
| if oldConfigFile == nil { | ||
| os.Remove(*oldConfigFilename) | ||
| } else { | ||
| ioutil.WriteFile(*oldConfigFilename, oldDefaultConfigFile, 0644) | ||
| } | ||
| } | ||
|
|
||
| oldConfigFile = nil; | ||
| oldConfigFilename = nil; | ||
| oldDefaultConfigFile = nil; | ||
| } | ||
|
|
||
| func testConfig(t *testing.T, testId int, expected uint64, cmdLine *uint64, defaultConfig *uint64, config *uint64, configFile *string) { | ||
| var defaultConfigContents *string | ||
| var configFileContents *string | ||
| var commandLine []string = make([]string, 0) | ||
|
|
||
| defer cleanup() | ||
|
|
||
| // first construct the command-line arguments. | ||
| if cmdLine != nil { | ||
| commandLine = append(commandLine, fmt.Sprintf("--maxpeers=%s", strconv.FormatUint(*cmdLine, 10))) | ||
| } | ||
| if configFile != nil { | ||
| commandLine = append(commandLine, fmt.Sprintf("--configfile=%s", *configFile)) | ||
| } | ||
|
|
||
| // Make the default config file. | ||
| if defaultConfig != nil { | ||
| var dcc string = fmt.Sprintf("maxpeers=%s", strconv.FormatUint(*defaultConfig, 10)) | ||
| defaultConfigContents = &dcc | ||
| } | ||
|
|
||
| // Make the extra config file. | ||
| if config != nil { | ||
| var cc string = fmt.Sprintf("maxpeers=%s", strconv.FormatUint(*config, 10)) | ||
| configFileContents = &cc | ||
| } | ||
|
|
||
| // Set up the test. | ||
| err := setup(defaultConfigContents, configFileContents, configFile) | ||
| if err != nil { | ||
| t.Fail() | ||
| } | ||
|
|
||
| cfg, _, err := LoadConfig("test", commandLine) | ||
|
|
||
| if cfg == nil { | ||
| t.Errorf("Error, test id %d: nil config returned! %s", testId, err.Error()) | ||
| return | ||
| } | ||
|
|
||
| if cfg.MaxPeers != int(expected) { | ||
| t.Errorf("Error, test id %d: expected %d got %d.", testId, expected, cfg.MaxPeers) | ||
| } | ||
|
|
||
| } | ||
|
|
||
| func TestLoadConfig(t *testing.T) { | ||
|
|
||
| // Test that an option is correctly set by default when | ||
| // no such option is specified in the default config file | ||
| // or on the command line. | ||
| testConfig(t, 1, defaultMaxPeers, nil, nil, nil, nil) | ||
|
|
||
| // Test that an option is correctly set when specified | ||
| // on the command line. | ||
| var q uint64 = 97 | ||
| testConfig(t, 2, q, &q, nil, nil, nil) | ||
|
|
||
| // Test that an option is correctly set when specified | ||
| // in the default config file without a command line | ||
| // option set. | ||
| var cfg string = "altbmd.conf" | ||
| testConfig(t, 3, q, nil, &q, nil, nil) | ||
| testConfig(t, 4, q, nil, nil, &q, &cfg) | ||
|
|
||
| // Test that an option is correctly set when specified | ||
| // on the command line and that it overwrites the | ||
| // option in the config file. | ||
| var z uint64 = 39 | ||
| testConfig(t, 5, q, &q, &z, nil, nil) | ||
| testConfig(t, 6, q, &q, nil, &z, &cfg) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters