| OLD | NEW |
| (Empty) |
| 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 | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import unittest | |
| 6 | |
| 7 from loading_trace_database import LoadingTraceDatabase | |
| 8 | |
| 9 | |
| 10 class LoadingTraceDatabaseUnittest(unittest.TestCase): | |
| 11 _JSON_DATABASE = { | |
| 12 "traces/trace1.json" : { "url" : "http://bar.html", }, | |
| 13 "traces/trace2.json" : { "url" : "http://bar.html", }, | |
| 14 "traces/trace3.json" : { "url" : "http://qux.html", }, | |
| 15 } | |
| 16 | |
| 17 def setUp(self): | |
| 18 self.database = LoadingTraceDatabase.FromJsonDict(self._JSON_DATABASE) | |
| 19 | |
| 20 def testGetTraceFilesForURL(self): | |
| 21 # Test a URL with no matching traces. | |
| 22 self.assertEqual( | |
| 23 self.database.GetTraceFilesForURL("http://foo.html"), | |
| 24 []) | |
| 25 | |
| 26 # Test a URL with matching traces. | |
| 27 self.assertEqual( | |
| 28 set(self.database.GetTraceFilesForURL("http://bar.html")), | |
| 29 set(["traces/trace1.json", "traces/trace2.json"])) | |
| 30 | |
| 31 def testSerialization(self): | |
| 32 self.assertEqual( | |
| 33 self._JSON_DATABASE, self.database.ToJsonDict()) | |
| 34 | |
| 35 def testSetTrace(self): | |
| 36 dummy_url = "http://dummy.com" | |
| 37 new_trace_file = "traces/new_trace.json" | |
| 38 self.assertEqual(self.database.GetTraceFilesForURL(dummy_url), []) | |
| 39 self.database.SetTrace(new_trace_file, {"url" : dummy_url}) | |
| 40 self.assertEqual(self.database.GetTraceFilesForURL(dummy_url), | |
| 41 [new_trace_file]) | |
| 42 | |
| 43 | |
| 44 if __name__ == '__main__': | |
| 45 unittest.main() | |
| OLD | NEW |