| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import unittest | 5 import unittest |
| 6 | 6 |
| 7 from loading_trace_database import LoadingTraceDatabase | 7 from loading_trace_database import LoadingTraceDatabase |
| 8 | 8 |
| 9 | 9 |
| 10 class LoadingTraceDatabaseUnittest(unittest.TestCase): | 10 class LoadingTraceDatabaseUnittest(unittest.TestCase): |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 | 25 |
| 26 # Test a URL with matching traces. | 26 # Test a URL with matching traces. |
| 27 self.assertEqual( | 27 self.assertEqual( |
| 28 set(self.database.GetTraceFilesForURL("http://bar.html")), | 28 set(self.database.GetTraceFilesForURL("http://bar.html")), |
| 29 set(["traces/trace1.json", "traces/trace2.json"])) | 29 set(["traces/trace1.json", "traces/trace2.json"])) |
| 30 | 30 |
| 31 def testSerialization(self): | 31 def testSerialization(self): |
| 32 self.assertEqual( | 32 self.assertEqual( |
| 33 self._JSON_DATABASE, self.database.ToJsonDict()) | 33 self._JSON_DATABASE, self.database.ToJsonDict()) |
| 34 | 34 |
| 35 def testAddTrace(self): | 35 def testSetTrace(self): |
| 36 dummy_url = "http://dummy.com" | 36 dummy_url = "http://dummy.com" |
| 37 new_trace_file = "traces/new_trace.json" | 37 new_trace_file = "traces/new_trace.json" |
| 38 self.assertEqual(self.database.GetTraceFilesForURL(dummy_url), []) | 38 self.assertEqual(self.database.GetTraceFilesForURL(dummy_url), []) |
| 39 self.database.AddTrace(new_trace_file, {"url" : dummy_url}) | 39 self.database.SetTrace(new_trace_file, {"url" : dummy_url}) |
| 40 self.assertEqual(self.database.GetTraceFilesForURL(dummy_url), | 40 self.assertEqual(self.database.GetTraceFilesForURL(dummy_url), |
| 41 [new_trace_file]) | 41 [new_trace_file]) |
| 42 | 42 |
| 43 | 43 |
| 44 if __name__ == '__main__': | 44 if __name__ == '__main__': |
| 45 unittest.main() | 45 unittest.main() |
| OLD | NEW |