← Back to branch summary

~lutostag/ubuntu/trusty/maas/1.5.4+keystone

« back to all changes in this revision

Viewing changes to src/maasserver/testing/factory.py

  • Committer: Package Import Robot
  • Author(s): Andres Rodriguez
  • Date: 2013-03-04 11:49:44 UTC
  • mto: This revision was merged to the branch mainline in revision 25.
  • Revision ID: package-import@ubuntu.com-20130304114944-azcvu9anlf8mizpa
Tags: upstream-1.3+bzr1452+dfsg
Import upstream version 1.3+bzr1452+dfsg

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/testing/factory.py
43
43
    )
44
44
from maasserver.utils import map_enum
45
45
import maastesting.factory
46
 
from metadataserver.models import NodeCommissionResult
 
46
from metadataserver.fields import Bin
 
47
from metadataserver.models import (
 
48
    CommissioningScript,
 
49
    NodeCommissionResult,
 
50
    )
47
51
from netaddr import IPAddress
48
52
 
49
53
# We have a limited number of public keys:
226
230
        interface.save()
227
231
        return interface
228
232
 
229
 
    def make_node_commission_result(self, node=None, name=None, data=None):
 
233
    def make_node_commission_result(self, node=None, name=None,
 
234
                                    script_result=None, data=None):
230
235
        if node is None:
231
236
            node = self.make_node()
232
237
        if name is None:
233
238
            name = "ncrname-" + self.getRandomString(92)
234
239
        if data is None:
235
240
            data = "ncrdata-" + self.getRandomString(1000)
236
 
        ncr = NodeCommissionResult(node=node, name=name, data=data)
 
241
        if script_result is None:
 
242
            script_result = random.randint(0, 10)
 
243
        ncr = NodeCommissionResult(
 
244
            node=node, name=name, script_result=script_result, data=data)
237
245
        ncr.save()
238
246
        return ncr
239
247
 
276
284
        key.save()
277
285
        return key
278
286
 
279
 
    def make_tag(self, name=None, definition=None, comment='', created=None,
280
 
                 updated=None):
 
287
    def make_tag(self, name=None, definition=None, comment='',
 
288
                 kernel_opts=None, created=None, updated=None):
281
289
        if name is None:
282
290
            name = self.make_name('tag')
283
291
        if definition is None:
284
292
            # Is there a 'node' in this xml?
285
293
            definition = '//node'
286
 
        tag = Tag(name=name, definition=definition, comment=comment)
 
294
        tag = Tag(name=name, definition=definition, comment=comment,
 
295
            kernel_opts=kernel_opts)
287
296
        self._save_node_unchecked(tag)
288
297
        # Update the 'updated'/'created' fields with a call to 'update'
289
298
        # preventing a call to save() from overriding the values.
364
373
            release=release,
365
374
            purpose=purpose)
366
375
 
 
376
    def make_commissioning_script(self, name=None, content=None):
 
377
        if name is None:
 
378
            name = self.make_name('script')
 
379
        if content is None:
 
380
            content = b'content:' + self.getRandomString().encode('ascii')
 
381
        return CommissioningScript.objects.create(
 
382
            name=name, content=Bin(content))
 
383
 
367
384
 
368
385
# Create factory singleton.
369
386
factory = Factory()