Revision 910d960d snf-cyclades-app/synnefo/db/models_factory.py

b/snf-cyclades-app/synnefo/db/models_factory.py
112 112
    operstate = "STARTED"
113 113

  
114 114

  
115
class VolumeFactory(factory.DjangoModelFactory):
116
    FACTORY_FOR = models.Volume
117
    userid = factory.Sequence(user_seq())
118
    size = factory.Sequence(lambda x: x, type=int)
119
    name = factory.Sequence(lambda x: "volume-name-"+x, type=str)
120
    machine = factory.SubFactory(VirtualMachineFactory,
121
                                 userid=factory.SelfAttribute('..userid'))
122
    disk_template = factory.LazyAttribute(lambda v:
123
                                          v.machine.flavor.disk_template
124
                                          if v.machine else "drbd")
125

  
126

  
115 127
class DeletedVirtualMachine(VirtualMachineFactory):
116 128
    deleted = True
117 129

  
......
286 298
    server_id = 1
287 299
    network_id = 1
288 300
    active = True
289

  
290

  
291
class VolumeFactory(factory.DjangoModelFactory):
292
    FACTORY_FOR = models.Volume
293
    userid = factory.Sequence(user_seq())
294
    size = factory.Sequence(lambda x: x, type=int)
295
    name = factory.Sequence(lambda x: "volume-name-"+x, type=str)

Also available in: Unified diff