# HG changeset patch
# User Henry S. Thompson <ht@inf.ed.ac.uk>
# Date 1689352754 -3600
# Node ID 9837840f332829fedc8828b94ac731e66a9eaf35
# Parent  11cbaee8bbc81e1dbafe724cd184d203be2d71dc
more tests

diff -r 11cbaee8bbc8 -r 9837840f3328 bin/test_warc.py
--- a/bin/test_warc.py	Fri Jul 14 17:38:54 2023 +0100
+++ b/bin/test_warc.py	Fri Jul 14 17:39:14 2023 +0100
@@ -5,6 +5,8 @@
 if (debug:=(sys.argv[1]=='-d')):
   sys.argv.pop(1)
 
+tt=int(sys.argv.pop(1))
+
 def showme(wtype,buf,part):
   # This should exactly reproduce a complete warc file if called
   #  as per version 1 below
@@ -19,8 +21,11 @@
     OUT.write(b"----end %d-----\n"%part)
   return OUT
 
-# 1
-warc.warc(sys.argv[1],showme,[b'response','warcinfo','request','metadata'],parts=int(sys.argv[2]),debug=debug)
-
-# 2
-#warc.warc(sys.argv[1],showme,[b'warcinfo'],parts=int(sys.argv[2]),debug=debug)
+if tt==1:
+  warc.warc(sys.argv[1],showme,[b'response','warcinfo','request','metadata'],parts=int(sys.argv[2]),debug=debug)
+elif tt==2:
+  warc.warc(sys.argv[1],showme,[b'warcinfo'],parts=int(sys.argv[2]),debug=debug)
+elif tt==3:
+  warc.warc(sys.argv[1],showme,[b'warcinfo'],whole=True,debug=debug)
+elif tt==4:
+  warc.warc(sys.argv[1],showme,[b'response','warcinfo','request','metadata'],whole=True,debug=debug)