~ubuntu-branches/ubuntu/lucid/landscape-client/lucid-updates-201411191716

« back to all changes in this revision

Viewing changes to landscape/manager/tests/test_customgraph.py

  • Committer: Bazaar Package Importer
  • Author(s): Free Ekanayaka
  • Date: 2009-12-16 10:50:05 UTC
  • mto: This revision was merged to the branch mainline in revision 25.
  • Revision ID: james.westby@ubuntu.com-20091216105005-9lgmwl5zklvvp6e2
ImportĀ upstreamĀ versionĀ 1.4.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
24
24
        self.manager.store = self.store
25
25
        self.broker_service.message_store.set_accepted_types(
26
26
            ["custom-graph"])
27
 
        self.data_path = self.make_dir()
 
27
        self.data_path = self.makeDir()
28
28
        self.manager.config.data_path = self.data_path
29
29
        os.makedirs(os.path.join(self.data_path, "custom-graph-scripts"))
30
30
        self.manager.config.script_users = "ALL"
92
92
        class pwnam(object):
93
93
            pw_uid = 1234
94
94
            pw_gid = 5678
95
 
            pw_dir = self.make_path()
 
95
            pw_dir = self.makeFile()
96
96
 
97
97
        self.expect(mock_getpwnam("bar")).result(pwnam)
98
98
        self.mocker.replay()
197
197
        return self.graph_manager.run().addCallback(check)
198
198
 
199
199
    def test_run_cast_result_error(self):
200
 
        filename = self.make_path("some_content")
 
200
        filename = self.makeFile("some_content")
201
201
        self.store.add_graph(123, filename, None)
202
202
        factory = StubProcessFactory()
203
203
        self.graph_manager.process_factory = factory
223
223
        return result.addCallback(check)
224
224
 
225
225
    def test_run_no_output_error(self):
226
 
        filename = self.make_path("some_content")
 
226
        filename = self.makeFile("some_content")
227
227
        self.store.add_graph(123, filename, None)
228
228
        factory = StubProcessFactory()
229
229
        self.graph_manager.process_factory = factory
248
248
        return result.addCallback(check)
249
249
 
250
250
    def test_run_no_output_error_with_other_result(self):
251
 
        filename1 = self.make_path("some_content")
 
251
        filename1 = self.makeFile("some_content")
252
252
        self.store.add_graph(123, filename1, None)
253
 
        filename2 = self.make_path("some_content")
 
253
        filename2 = self.makeFile("some_content")
254
254
        self.store.add_graph(124, filename2, None)
255
255
        factory = StubProcessFactory()
256
256
        self.graph_manager.process_factory = factory
278
278
        return result.addCallback(check)
279
279
 
280
280
    def test_multiple_errors(self):
281
 
        filename1 = self.make_path("some_content")
 
281
        filename1 = self.makeFile("some_content")
282
282
        self.store.add_graph(123, filename1, None)
283
 
        filename2 = self.make_path("some_content")
 
283
        filename2 = self.makeFile("some_content")
284
284
        self.store.add_graph(124, filename2, None)
285
285
        factory = StubProcessFactory()
286
286
        self.graph_manager.process_factory = factory
309
309
        return result.addCallback(check)
310
310
 
311
311
    def test_run_user(self):
312
 
        filename = self.make_path("some content")
 
312
        filename = self.makeFile("some content")
313
313
        self.store.add_graph(123, filename, "bar")
314
314
        factory = StubProcessFactory()
315
315
        self.graph_manager.process_factory = factory
318
318
        class pwnam(object):
319
319
            pw_uid = 1234
320
320
            pw_gid = 5678
321
 
            pw_dir = self.make_path()
 
321
            pw_dir = self.makeFile()
322
322
 
323
323
        self.expect(mock_getpwnam("bar")).result(pwnam)
324
324
        self.mocker.replay()
344
344
        username = info.pw_name
345
345
        self.manager.config.script_users = "foo"
346
346
 
347
 
        filename = self.make_path("some content")
348
 
        self.store.add_graph(123, filename, username)
 
347
        self.store.add_graph(123, "filename", username)
349
348
        factory = StubProcessFactory()
350
349
        self.graph_manager.process_factory = factory
351
350
        result = self.graph_manager.run()
374
373
 
375
374
        self.manager.config.script_users = "foo"
376
375
 
377
 
        filename = self.make_path("some content")
378
 
        self.store.add_graph(123, filename, "foo")
 
376
        self.store.add_graph(123, "filename", "foo")
379
377
        factory = StubProcessFactory()
380
378
        self.graph_manager.process_factory = factory
381
379
        result = self.graph_manager.run()
395
393
        return result.addCallback(check)
396
394
 
397
395
    def test_run_timeout(self):
398
 
        filename = self.make_path("some content")
 
396
        filename = self.makeFile("some content")
399
397
        self.store.add_graph(123, filename, None)
400
398
        factory = StubProcessFactory()
401
399
        self.graph_manager.process_factory = factory