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 |