class WriteBufferTest(unittest.TestCase): def setUp(self): item_writer = GroupingBufferFilesTracker(JsonExportFormatter({}, meta()), 'gz') self.write_buffer = WriteBuffer(1000, 1000, item_writer) def tearDown(self): self.write_buffer.close() def test_get_metadata(self): # given: self.write_buffer.metadata['somekey'] = {'items': 10} # then self.assertEqual(self.write_buffer.get_metadata('somekey', 'items'), 10, 'Wrong metadata') self.assertIsNone(self.write_buffer.get_metadata('somekey', 'nokey'))