Add module to handle configuration files
Sayan Nandan
4 years ago
parent
308e57c846
commit
0690be4d4e
@ -0,0 +1,3 @@
|
||||
# This is a 'bad' configuration file since it contains an invalid port
|
||||
[server]
|
||||
port = 20033002
|
||||
@ -0,0 +1,4 @@
|
||||
# This is a 'good' configuration file since it contains a valid port
|
||||
# and appropriate keys
|
||||
[server]
|
||||
port = 2003 # Set the server's port to 2003
|
||||
@ -0,0 +1,95 @@
|
||||
/*
|
||||
* Created on Tue Sep 01 2020
|
||||
*
|
||||
* This file is a part of TerrabaseDB
|
||||
* Copyright (c) 2020, Sayan Nandan <ohsayan at outlook dot com>
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*
|
||||
*/
|
||||
|
||||
//! This module provides tools to handle configuration files and settings
|
||||
|
||||
use libtdb::TResult;
|
||||
use serde::Deserialize;
|
||||
use toml;
|
||||
|
||||
#[derive(Deserialize, Debug, PartialEq)]
|
||||
struct Config {
|
||||
server: ServerConfig,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Debug, PartialEq)]
|
||||
struct ServerConfig {
|
||||
port: u16,
|
||||
}
|
||||
|
||||
impl Config {
|
||||
pub fn new(file: String) -> TResult<Self> {
|
||||
let res: Config = toml::from_str(&file)?;
|
||||
Ok(res)
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[cfg(test)]
|
||||
fn test_config_toml_okayport() {
|
||||
let file = r#"
|
||||
[server]
|
||||
port = 2003
|
||||
"#
|
||||
.to_owned();
|
||||
let cfg = Config::new(file).unwrap();
|
||||
assert_eq!(
|
||||
cfg,
|
||||
Config {
|
||||
server: ServerConfig { port: 2003 }
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[cfg(test)]
|
||||
fn test_config_toml_badport() {
|
||||
let file = r#"
|
||||
[server]
|
||||
port = 20033002
|
||||
"#
|
||||
.to_owned();
|
||||
let cfg = Config::new(file);
|
||||
assert!(cfg.is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[cfg(test)]
|
||||
fn test_config_file_ok() {
|
||||
let fileloc = "../examples/config-files/tdb.toml";
|
||||
let file = std::fs::read_to_string(fileloc).unwrap();
|
||||
let cfg: Config = Config::new(file).unwrap();
|
||||
assert_eq!(
|
||||
cfg,
|
||||
Config {
|
||||
server: ServerConfig { port: 2003 }
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
#[cfg(test)]
|
||||
fn test_config_file_err() {
|
||||
let fileloc = "../examples/config-files/badcfg.toml";
|
||||
let file = std::fs::read_to_string(fileloc).unwrap();
|
||||
let cfg = Config::new(file);
|
||||
assert!(cfg.is_err());
|
||||
}
|
||||
Loading…
Reference in New Issue