diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 4 | ||||
| -rw-r--r-- | src/tests.rs | 5 |
2 files changed, 5 insertions, 4 deletions
diff --git a/src/main.rs b/src/main.rs index d484e5e..77f230a 100644 --- a/src/main.rs +++ b/src/main.rs @@ -51,8 +51,8 @@ impl Alias { } fn write_to_file<P: AsRef<Path>>(&self, file: P) -> Result<(), io::Error> { - let mut f = try!(OpenOptions::new().write(true).open(file)); - try!(f.write_all(self.to_string().as_bytes())); + let mut f = try!(OpenOptions::new().append(true).open(file)); + try!(f.write_all(format!("{}\n", self.to_string()).as_bytes())); Ok(()) } diff --git a/src/tests.rs b/src/tests.rs index 1034bc6..2a6e264 100644 --- a/src/tests.rs +++ b/src/tests.rs @@ -130,13 +130,14 @@ fn alias_write_to_file_must_write_given_alias_to_file() { let alias = update_alias_id_sample_alias(); let test_file = "./testdata/write_to_file"; - File::create(test_file).unwrap(); + fs::copy("./testdata/aliases", test_file).unwrap(); alias.write_to_file(test_file).unwrap(); let mut f = File::open(test_file).unwrap(); let mut file_contents = String::new(); f.read_to_string(&mut file_contents).unwrap(); + let file_contents: Vec<&str> = file_contents.split('\n').collect(); fs::remove_file(test_file).unwrap(); - assert_eq!(alias.to_string(), file_contents); + assert_eq!(alias.to_string(), file_contents[file_contents.len() - 2]); } |