def test_agent_with_package(self): # Note that this test assumes the import is installed framework = object() registry.initialize(framework) registry.trust_controller = False registry.trust_agent = True f = IPATrustPackageCheck(registry) f.config = config.Config() self.results = capture_results(f) assert len(self.results) == 1 result = self.results.results[0] assert result.result == constants.SUCCESS assert result.source == 'ipahealthcheck.ipa.trust' assert result.check == 'IPATrustPackageCheck'
def test_agent_without_package(self): # Note that this test assumes the import is installed framework = object() registry.initialize(framework) registry.trust_controller = False registry.trust_agent = True # Hose up the module so the import fails save = sys.modules['ipaserver.install'] sys.modules['ipaserver.install'] = 'foo' f = IPATrustPackageCheck(registry) f.config = config.Config() self.results = capture_results(f) assert len(self.results) == 1 result = self.results.results[0] assert result.result == constants.WARNING assert result.source == 'ipahealthcheck.ipa.trust' assert result.check == 'IPATrustPackageCheck' sys.modules['ipaserver.install'] = save