def testPurgeDeprecatedCompaniesWithNoneDeprecated( self, deleteCompaniesMock, queryCachedCompanySymbolsMock, getAllMetricSecuritiesMock): getAllMetricSecuritiesMock.return_value = [("IBM", "exg"), ("T", "exg")] queryCachedCompanySymbolsMock.return_value = set(["IBM", "T"]) # Execute agent._purgeDeprecatedCompanies() # Validate self.assertEqual(deleteCompaniesMock.call_count, 0)
def testPurgeDeprecatedCompanies( self, # pylint: disable=R0201 deleteCompaniesMock, queryCachedCompanySymbolsMock, getAllMetricSecuritiesMock): getAllMetricSecuritiesMock.return_value = [("IBM", "exg"), ("T", "exg")] queryCachedCompanySymbolsMock.return_value = set( ["IBM", "T", "FOO", "BAR"]) # Execute agent._purgeDeprecatedCompanies() # Validate deleteCompaniesMock.assert_called_once_with( tickerSymbols=set(["FOO", "BAR"]), engineServer=agent._TAURUS_HTM_SERVER, engineApiKey=agent._TAURUS_API_KEY, warnAboutDestructiveAction=False)