Revision 13d62481 test/hs/Test/Ganeti/Objects.hs

b/test/hs/Test/Ganeti/Objects.hs
34 34 
  , genDisk
35 35 
  , genDiskWithChildren
36 36 
  , genEmptyCluster
37 
  , genInst
37 38 
  , genInstWithNets
38 39 
  , genValidNetwork
39 40 
  , genBitStringMaxLen
......
127 128 
      -- nics
128 129 
      <*> arbitrary
129 130 
      -- disks
130 
      <*> arbitrary
131 
      <*> vectorOf 5 genDisk
131 132 
      -- disk template
132 133 
      <*> arbitrary
133 134 
      -- network port
......
146 147 
genInstWithNets :: [String] -> Gen Instance
147 148 
genInstWithNets nets = do
148 149 
  plain_inst <- arbitrary
150 
  enhanceInstWithNets plain_inst nets
151 

  		





  
  152
  
    
-- | Generates an instance that is connected to some networks


  		





  
  153
  
    
genInst :: Gen Instance


  		





  
  154
  
    
genInst = genInstWithNets []


  		





  
  155
  
    

  		





  
  156
  
    
-- | Enhances a given instance with network information, by connecting it to the


  		





  
  157
  
    
-- given networks and possibly some other networks


  		





  
  158
  
    
enhanceInstWithNets :: Instance -> [String] -> Gen Instance


  		





  
  159
  
    
enhanceInstWithNets inst nets = do


  		





  149
  160
  
    
  mac <- arbitrary


  		





  150
  161
  
    
  ip <- arbitrary


  		





  151
  162
  
    
  nicparams <- arbitrary


  		





  ......




  157
  168
  
    
  let genNic net = PartialNic mac ip nicparams net name uuid


  		





  158
  169
  
    
      partial_nics = map (genNic . Just)


  		





  159
  170
  
    
                         (List.nub (nets ++ more_nets))


  		





  160
  
  
    
      new_inst = plain_inst { instNics = partial_nics }


  		





  
  171
  
    
      new_inst = inst { instNics = partial_nics }


  		





  161
  172
  
    
  return new_inst


  		





  162
  173
  
    

  		





  163
  174
  
    
genDiskWithChildren :: Int -> Gen Disk


  		












Also available in:
  Unified diff