def test_headers_get_prefixes(): """ Insert docstring here """ expected_prefixes = ['Title', 'Timestamp', 'Last Accessed', 'Times Accessed', 'Tags', 'Metadata'] # Instantiate mock objects title_header = Mock(['get_prefix'], name='title_header') timestamp_header = Mock(['get_prefix'], name='timestamp_header') last_accessed_header = Mock(['get_prefix'], name='last_accessed_header') times_accessed_header = Mock(['get_prefix'], name='times_accessed_header') tags_header = Mock(['get_prefix'], name='tags_header') metadata_header = Mock(['get_prefix'], name='metadata_header') # Set what is returned by the method get_prefix title_header.get_prefix.return_value = 'Title' timestamp_header.get_prefix.return_value = 'Timestamp' last_accessed_header.get_prefix.return_value = 'Last Accessed' times_accessed_header.get_prefix.return_value = 'Times Accessed' tags_header.get_prefix.return_value = 'Tags' metadata_header.get_prefix.return_value = 'Metadata' # Instantiate object under tests (headers) headers = Headers(title_header, timestamp_header, last_accessed_header, times_accessed_header, tags_header, metadata_header) assert(headers.get_prefixes() == expected_prefixes)
class TrunkNotesParser: """ Insert docstring here """ def __init__(self): """ Insert docstring here """ self.headers = Headers(TitleHeader(), TimestampHeader(), LastAccessedHeader(), TimesAccessedHeader(), TagsHeader(), MetadataHeader()) self.body = '' def parse_content(self, input_data): """ Insert docstring here """ metadata = join(input_data.splitlines()[0:6], '\n') self.body = join(input_data.splitlines(True)[6:], '') self.headers.read(metadata) def parse_file(self, file_path): """ Insert docstring here """ try: note_file = open(file_path) input_data = note_file.read() note_file.close() except IOError as errno: # No such directory Error if errno.args[0] == 2: # will be replace once code is ready for input checkecking pass ## is a directory ERROR if errno.args[0] == 21: # will be replace once code is ready for input checkecking pass else: return self.parse_content(input_data) def dump_note(self, note_path): """ Insert docstring here """ output_data = self.headers.write() + self.body # create the file with open(note_path.__str__(), 'w') as note_file: # write to the file note_file.write(output_data)
def __init__(self): """ Insert docstring here """ self.headers = Headers(TitleHeader(), TimestampHeader(), LastAccessedHeader(), TimesAccessedHeader(), TagsHeader(), MetadataHeader()) self.body = ''
def test_headers_write(): """ Insert docstring here """ # The string that headers.write should return expected_output_data = """\ Title: HelloWorld Timestamp: 2010-11-18 18:10:33 -0500 Last Accessed: 2010-11-19 20:02:28 -0500 Times Accessed: 24 Tags: Metadata: """ # Instantiate mock objects title_header = Mock(['write'], name='title_header') timestamp_header = Mock(['write'], name='timestamp_header') last_accessed_header = Mock(['write'], name='last_accessed_header') times_accessed_header = Mock(['write'], name='times_accessed_header') tags_header = Mock(['write'], name='tags_header') metadata_header = Mock(['write'], name='metadata_header') # Set what is returned by the method write title_header.write.return_value = 'Title: HelloWorld\n' timestamp_header.write.return_value = \ 'Timestamp: 2010-11-18 18:10:33 -0500\n' last_accessed_header.write.return_value = \ 'Last Accessed: 2010-11-19 20:02:28 -0500\n' times_accessed_header.write.return_value = 'Times Accessed: 24\n' tags_header.write.return_value = 'Tags:\n' metadata_header.write.return_value = 'Metadata:\n' # Instantiate object under tests (headers) headers = Headers(title_header, timestamp_header, last_accessed_header, times_accessed_header, tags_header, metadata_header) # Test that headers.write() returns the correct string assert(headers.write() == expected_output_data)
def test_headers_read(): """ Insert docstring here """ sample_input_data = """\ Title: HelloWorld Timestamp: 2010-11-18 18:10:33 -0500 Last Accessed: 2010-11-19 20:02:28 -0500 Times Accessed: 24 Tags: Metadata: """ # Instantiate mock objects title_header = Mock(['read'], name='title_header') timestamp_header = Mock(['read'], name='timestamp_header') last_accessed_header = Mock(['read'], name='last_accessed_header') times_accessed_header = Mock(['read'], name='times_accessed_header') tags_header = Mock(['read'], name='tags_header') metadata_header = Mock(['read'], name='metadata_header') # Instantiate object under tests (headers) headers = Headers(title_header, timestamp_header, last_accessed_header, times_accessed_header, tags_header, metadata_header) # Call method under test headers.read(sample_input_data) # Test that the headers' read was called once and # That the parameters they received are correct title_header.read.assert_called_once_with('Title: HelloWorld') timestamp_header.read.assert_called_once_with( 'Timestamp: 2010-11-18 18:10:33 -0500') last_accessed_header.read.assert_called_once_with( 'Last Accessed: 2010-11-19 20:02:28 -0500') times_accessed_header.read.assert_called_once_with('Times Accessed: 24') tags_header.read.assert_called_once_with('Tags:') metadata_header.read.assert_called_once_with('Metadata:')