Index: appengine/findit/crash/loglinear/test/changelist_classifier_test.py |
diff --git a/appengine/findit/crash/loglinear/test/changelist_classifier_test.py b/appengine/findit/crash/loglinear/test/changelist_classifier_test.py |
index c80db81d3890d503e9f3e3523e06b9caa705459e..0d5eb9a778ed6c5a1f3d877e43423e6d06faeb0a 100644 |
--- a/appengine/findit/crash/loglinear/test/changelist_classifier_test.py |
+++ b/appengine/findit/crash/loglinear/test/changelist_classifier_test.py |
@@ -10,7 +10,7 @@ from common.dependency import DependencyRoll |
from common import chrome_dependency_fetcher |
from crash import changelist_classifier |
from crash.crash_report import CrashReport |
-import crash.loglinear.changelist_classifier as loglinear_changelist_classifier |
+from crash.loglinear.changelist_classifier import LogLinearChangelistClassifier |
from crash.loglinear.feature import ChangedFile |
from crash.loglinear.feature import FeatureValue |
from crash.suspect import AnalysisInfo |
@@ -117,23 +117,11 @@ class LogLinearChangelistClassifierTest(CrashTestSuite): |
'TopFrameIndex': 1., |
} |
- repository = GitilesRepository(self.GetMockHttpClient()) |
+ def MockGitilesRepositoryFactory(repo_url): # pragma: no cover |
+ return GitilesRepository(self.GetMockHttpClient(), repo_url) |
- # TODO(crbug.com/677224): should replace this with an actual factory. |
- def MutateTheRepo(dep_url): # pragma: no cover |
- """A factory function for returning ``Repository`` objects. |
- |
- The current definition captures the functionality of before |
- we factored out this factory method. That is, it's not really a |
- "factory" but rather mutates the main repo object in place. In |
- the future this should be changed to do the right thing instead. |
- """ |
- repository.repo_url = dep_url |
- return repository |
- |
- self.changelist_classifier = ( |
- loglinear_changelist_classifier.LogLinearChangelistClassifier( |
- repository, MutateTheRepo, weights)) |
+ self.changelist_classifier = LogLinearChangelistClassifier( |
+ MockGitilesRepositoryFactory, weights) |
def testAggregateChangedFilesAggreegates(self): |
"""Test that ``AggregateChangedFiles`` does aggregate reasons per file. |