def test_defaults_dump(self):
info = WarcinfoRecord()
string = record_to_string(info)
self.assertEqual("WARC/1.0", string[:8])
self.assertIn("WARC-Type: ", string)
self.assertIn("WARC-Record-ID: ", string)
self.assertIn("WARC-Date: ", string)
self.assertIn("Content-Type: application/warc-fields", string)
self.assertIn("Content-Length: ", string)
self.assertIn("WARC-Block-Digest: ", string)
def test_sample_data(self):
headers = [("k1", "v1")]
info = WarcinfoRecord(id="myid", date="mydate", filename="myfilename",
content=("mytype","mycontent"), headers=headers)
string = record_to_string(info)
self.assertEqual("WARC/1.0", string[:8])
self.assertIn("WARC-Type: warcinfo", string)
self.assertIn("WARC-Record-ID: myid", string)
self.assertIn("WARC-Date: mydate", string)
self.assertIn("WARC-Filename: myfilename", string)
self.assertIn("Content-Type: mytype", string)
self.assertIn("Content-Length: 9", string)
self.assertIn("WARC-Block-Digest: sha1:2MKCNB7EJF4I7VPZPJVUNHBIJ5S37ASD", string)
self.assertEqual("\r\n\r\nmycontent\r\n\r\n", string [-17:])
def test_defaults(self):
info = WarcinfoRecord()
self.assertIsNotNone(info.get_header("WARC-Type"))
self.assertIsNotNone(info.get_header("WARC-Record-ID"))
self.assertIsNotNone(info.get_header("WARC-Date"))
def test_type(self):
info = WarcinfoRecord()
self.assertEqual(info.get_header(WarcRecord.TYPE), "warcinfo")
def test_filename(self):
info = WarcinfoRecord(filename="test")
self.assertEqual(info.get_header("WARC-Filename"), "test")
string = record_to_string(info)
self.assertIn("WARC-Filename: test", string)
def date_test(self, defaults):
info = WarcinfoRecord(date="test", defaults=defaults)
self.assertEqual(info.get_header("WARC-Date"), "test")
string = record_to_string(info)
self.assertIn("WARC-Date: test", string)
def id_test(self, defaults):
info = WarcinfoRecord(id="test", defaults=defaults)
self.assertEqual(info.get_header(WarcRecord.ID), "test")
string = record_to_string(info)
self.assertIn("WARC-Record-ID: test", string)
