blob: a52f9561c2ed21ebc0a2e60a57ecb7284328182a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# Copyright 2015 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
import tempfile
import unittest
try:
import sqlite3 # Not present on ChromeOS.
except ImportError:
pass
from telemetry import decorators
from profile_creators.cookie_profile_extender import CookieProfileExtender
# Testing private method.
# pylint: disable=protected-access
class CookieProfileExtenderTest(unittest.TestCase):
def _CreateCookieTable(self, path):
connection = sqlite3.connect(path)
cursor = connection.cursor()
cursor.execute("CREATE TABLE cookies (url text)")
connection.commit()
connection.close()
def _AddCookiesToTable(self, path, count):
connection = sqlite3.connect(path)
cursor = connection.cursor()
for i in range(count):
cursor.execute("INSERT INTO cookies VALUES ('%s')" % i)
connection.commit()
connection.close()
@decorators.Disabled('chromeos') # crbug.com/483212
def testCookieCount(self):
# Neither tempfile.TemporaryFile() nor tempfile.NamedTemporaryFile() work
# well here. The former doesn't work at all, since it doesn't gaurantee a
# file-system visible path. The latter doesn't work well, since the
# returned file cannot be opened a second time on Windows. The returned
# file would have to be closed, and the method would need to be called with
# Delete=False, which makes its functionality no simpler than
# tempfile.mkstemp().
handle, path = tempfile.mkstemp()
try:
os.close(handle)
self._CreateCookieTable(path)
self.assertEquals(CookieProfileExtender._CookieCountInDB(path), 0)
self._AddCookiesToTable(path, 100)
self.assertEquals(CookieProfileExtender._CookieCountInDB(path), 100)
finally:
os.remove(path)
|
