From e1bf27bb84efb4273c47ab8f5828c9f0db2dec96 Mon Sep 17 00:00:00 2001
From: Agata Murawska <agatamurawska@google.com>
Date: Tue, 15 Nov 2011 16:48:23 +0100
Subject: [PATCH] htools: Nicer arbitrary instantiation

Signed-off-by: Agata Murawska <agatamurawska@google.com>
Reviewed-by: Iustin Pop <iustin@google.com>
---
 htools/Ganeti/HTools/QC.hs | 15 ++-------------
 1 file changed, 2 insertions(+), 13 deletions(-)

diff --git a/htools/Ganeti/HTools/QC.hs b/htools/Ganeti/HTools/QC.hs
index a03830e6f..e1ddc511f 100644
--- a/htools/Ganeti/HTools/QC.hs
+++ b/htools/Ganeti/HTools/QC.hs
@@ -188,14 +188,7 @@ getFQDN = do
   return (felem ++ "." ++ intercalate "." frest')
 
 instance Arbitrary Types.InstanceStatus where
-    arbitrary = elements [ Types.AdminDown
-                         , Types.AdminOffline
-                         , Types.ErrorDown
-                         , Types.ErrorUp
-                         , Types.NodeDown
-                         , Types.NodeOffline
-                         , Types.Running
-                         , Types.WrongNode]
+    arbitrary = elements [minBound..maxBound]
 
 -- let's generate a random instance
 instance Arbitrary Instance.Instance where
@@ -246,11 +239,7 @@ instance Arbitrary Node.Node where
 
 -- replace disks
 instance Arbitrary OpCodes.ReplaceDisksMode where
-  arbitrary = elements [ OpCodes.ReplaceOnPrimary
-                       , OpCodes.ReplaceOnSecondary
-                       , OpCodes.ReplaceNewSecondary
-                       , OpCodes.ReplaceAuto
-                       ]
+  arbitrary = elements [minBound..maxBound]
 
 instance Arbitrary OpCodes.OpCode where
   arbitrary = do
-- 
GitLab