blob: 7c5b2b4709bdbe1249e7a36839febf25f0a3ab67 [file] [log] [blame]
#!/usr/bin/python2
import unittest
import common
from autotest_lib.client.common_lib import profiler_manager
# simple job stub for using in tests
class stub_job(object):
tmpdir = "/home/autotest/tmp"
autodir = "/home/autotest"
# simple profiler stub for using in tests
class stub_profiler(object):
started = 0
def __init__(self, name):
self.name = name
@classmethod
def start(cls, test):
cls.started += 1
@classmethod
def stop(cls, test):
cls.started -= 1
# replace profiler_manager.load_profiler with a simple stub
class stub_manager(profiler_manager.profiler_manager):
def load_profiler(self, profiler, args, dargs):
return stub_profiler(profiler)
class TestProfilerManager(unittest.TestCase):
def test_starts_with_no_profilers(self):
p = stub_manager(stub_job)
self.assertEqual(set(), p.current_profilers())
def test_single_add(self):
p = stub_manager(stub_job)
p.add("prof1")
self.assertEqual(set(["prof1"]), p.current_profilers())
def test_duplicate_adds(self):
p = stub_manager(stub_job)
p.add("prof1")
p.add("prof1")
self.assertEqual(set(["prof1"]), p.current_profilers())
def test_multiple_adds(self):
p = stub_manager(stub_job)
p.add("prof1")
p.add("prof2")
self.assertEqual(set(["prof1", "prof2"]), p.current_profilers())
def test_add_and_delete(self):
p = stub_manager(stub_job)
p.add("prof1")
p.add("prof2")
p.delete("prof1")
self.assertEqual(set(["prof2"]), p.current_profilers())
def test_present_with_no_profilers(self):
p = stub_manager(stub_job)
self.assertEqual(False, p.present())
def test_present_after_add(self):
p = stub_manager(stub_job)
p.add("prof1")
self.assertEqual(True, p.present())
def test_present_after_add_and_remove(self):
p = stub_manager(stub_job)
p.add("prof1")
p.delete("prof1")
self.assertEqual(False, p.present())
def test_started(self):
p = stub_manager(stub_job)
p.add("prof1")
p.add("prof2")
started = stub_profiler.started
self.assertEqual(False, p.active())
p.start(object())
self.assertEqual(started + 2, stub_profiler.started)
self.assertEqual(True, p.active())
def test_stop(self):
p = stub_manager(stub_job)
p.add("prof1")
p.add("prof2")
started = stub_profiler.started
self.assertEqual(False, p.active())
test = object()
p.start(test)
p.stop(test)
self.assertEqual(started, stub_profiler.started)
self.assertEqual(False, p.active())
if __name__ == "__main__":
unittest.main()