Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(154)

Side by Side Diff: tools/perf/profile_creators/cookie_profile_extender_unittest.py

Issue 914253005: Telemetry: Create new profile creator large_profile_creator. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Comments from nednguyen. Created 5 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 # Copyright 2015 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 import os
5 import sqlite3
6 import tempfile
7 import unittest
8
9 from profile_creators.cookie_profile_extender import CookieProfileExtender
10
11
12 # Testing private method.
13 # pylint: disable=protected-access
14 class CookieProfileExtenderTest(unittest.TestCase):
15 def _CreateCookieTable(self, path):
16 connection = sqlite3.connect(path)
17 cursor = connection.cursor()
18 cursor.execute("CREATE TABLE cookies (url text)")
19 connection.commit()
20 connection.close()
21
22 def _AddCookiesToTable(self, path, count):
23 connection = sqlite3.connect(path)
24 cursor = connection.cursor()
25 for i in range(count):
26 cursor.execute("INSERT INTO cookies VALUES ('%s')" % i)
27 connection.commit()
28 connection.close()
29
30 def testCookieCount(self):
31 handle, path = tempfile.mkstemp()
32 try:
33 os.close(handle)
34
35 self._CreateCookieTable(path)
36 self.assertEquals(CookieProfileExtender._CookieCountInDB(path), 0)
37
38 self._AddCookiesToTable(path, 100)
39 self.assertEquals(CookieProfileExtender._CookieCountInDB(path), 100)
40 finally:
41 os.remove(path)
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698