~ Use testing flag to hide internals.

We need some internal functions in tests and benchmarks, so it's better
to just use CPP for this.

2 files changed, 19 insertions, 4 deletions

diff --git a/bittorrent.cabal b/bittorrent.cabal
index e3d98e84..138f5d3b 100644
--- a/bittorrent.cabal
+++ b/bittorrent.cabal
@@ -24,6 +24,9 @@ source-repository head
   type:                git
   location:            git://github.com/pxqr/bittorrent.git
 
+flag testing
+  description:         Expose debug stuff in export declarations.
+  default:             False
 
 library
   exposed-modules:     Data.Torrent
@@ -79,9 +82,12 @@ library
 
   extensions:          PatternGuards
   hs-source-dirs:      src
+  if flag(testing)
+    cpp-options:       -DTESTING
   ghc-options:         -Wall
 
 
+
 test-suite info-hash
   type:                exitcode-stdio-1.0
   main-is:             info-hash.hs
@@ -96,6 +102,9 @@ test-suite info-hash
 
   ghc-options:         -Wall
 
+  if !flag(testing)
+     buildable:        False
+
 
 
 test-suite properties
@@ -119,6 +128,9 @@ test-suite properties
 
   ghc-options:         -Wall -fno-warn-orphans
 
+  if !flag(testing)
+     buildable:        False
+
 
 
 benchmark benchmarks
@@ -137,3 +149,6 @@ benchmark benchmarks
                      , bittorrent
 
   ghc-options:         -O2 -Wall -fno-warn-orphans
+
+  if !flag(testing)
+     buildable:        False
\ No newline at end of file
diff --git a/src/Data/Bitfield.hs b/src/Data/Bitfield.hs
index a1033ff2..ee0570e7 100644
--- a/src/Data/Bitfield.hs
+++ b/src/Data/Bitfield.hs
@@ -34,10 +34,10 @@ module Data.Bitfield
        , getBitfield, putBitfield
        , bitfieldByteCount
 
--- #ifdef TESTING
-       , -- * Debug
-         mkBitfield
--- #endif
+#if  defined (TESTING)
+         -- * Debug
+       , mkBitfield
+#endif
        ) where
 
 import Control.Monad