X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/e5a29b6cccf6d59f8bcfca19ed1f1b7dce47e64a..b88fcdd0db5785b2cb69357098db1bd612c1590c:/htest/Test/Ganeti/Objects.hs

diff --git a/htest/Test/Ganeti/Objects.hs b/htest/Test/Ganeti/Objects.hs
index af82560..7baaf70 100644
--- a/htest/Test/Ganeti/Objects.hs
+++ b/htest/Test/Ganeti/Objects.hs
@@ -26,14 +26,34 @@ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 
 -}
 
-module Test.Ganeti.Objects (testObjects) where
+module Test.Ganeti.Objects
+  ( testObjects
+  , Objects.Hypervisor(..)
+  , Objects.Node(..)
+  ) where
 
+import Control.Applicative
 import qualified Data.Map as Map
+import qualified Data.Set as Set
 import Test.QuickCheck
 
 import Test.Ganeti.TestHelper
+import Test.Ganeti.TestCommon
 import qualified Ganeti.Objects as Objects
 
+instance Arbitrary Objects.Hypervisor where
+  arbitrary = elements [minBound..maxBound]
+
+instance Arbitrary Objects.PartialNDParams where
+  arbitrary = Objects.PartialNDParams <$> arbitrary <*> arbitrary
+
+instance Arbitrary Objects.Node where
+  arbitrary = Objects.Node <$> getFQDN <*> getFQDN <*> getFQDN
+              <*> arbitrary <*> arbitrary <*> arbitrary <*> getFQDN
+              <*> arbitrary <*> arbitrary <*> arbitrary <*> arbitrary
+              <*> arbitrary <*> arbitrary <*> getFQDN <*> arbitrary
+              <*> (Set.fromList <$> genTags)
+
 -- | Tests that fillDict behaves correctly
 prop_Objects_fillDict :: [(Int, Int)] -> [(Int, Int)] -> Property
 prop_Objects_fillDict defaults custom =