VirtualBox

source: vbox/trunk/src/VBox/ValidationKit/testmanager/core/testboxstatus.py@ 66998

Last change on this file since 66998 was 62484, checked in by vboxsync, 8 years ago

(C) 2016
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 11.8 KB
Line 
1# -*- coding: utf-8 -*-
2# $Id: testboxstatus.py 62484 2016-07-22 18:35:33Z vboxsync $
3
4"""
5Test Manager - TestBoxStatus.
6"""
7
8__copyright__ = \
9"""
10Copyright (C) 2012-2016 Oracle Corporation
11
12This file is part of VirtualBox Open Source Edition (OSE), as
13available from http://www.virtualbox.org. This file is free software;
14you can redistribute it and/or modify it under the terms of the GNU
15General Public License (GPL) as published by the Free Software
16Foundation, in version 2 as it comes in the "COPYING" file of the
17VirtualBox OSE distribution. VirtualBox OSE is distributed in the
18hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
19
20The contents of this file may alternatively be used under the terms
21of the Common Development and Distribution License Version 1.0
22(CDDL) only, as it comes in the "COPYING.CDDL" file of the
23VirtualBox OSE distribution, in which case the provisions of the
24CDDL are applicable instead of those of the GPL.
25
26You may elect to license modified versions of this file under the
27terms and conditions of either the GPL or the CDDL or both.
28"""
29__version__ = "$Revision: 62484 $"
30
31
32# Standard python imports.
33import unittest;
34
35# Validation Kit imports.
36from testmanager.core.base import ModelDataBase, ModelDataBaseTestCase, ModelLogicBase, TMTooManyRows, TMRowNotFound;
37from testmanager.core.testbox import TestBoxData;
38
39
40class TestBoxStatusData(ModelDataBase):
41 """
42 TestBoxStatus Data.
43 """
44
45 ## @name TestBoxState_T
46 # @{
47 ksTestBoxState_Idle = 'idle';
48 ksTestBoxState_Testing = 'testing';
49 ksTestBoxState_GangGathering = 'gang-gathering';
50 ksTestBoxState_GangGatheringTimedOut = 'gang-gathering-timedout';
51 ksTestBoxState_GangTesting = 'gang-testing';
52 ksTestBoxState_GangCleanup = 'gang-cleanup';
53 ksTestBoxState_Rebooting = 'rebooting';
54 ksTestBoxState_Upgrading = 'upgrading';
55 ksTestBoxState_UpgradingAndRebooting = 'upgrading-and-rebooting';
56 ksTestBoxState_DoingSpecialCmd = 'doing-special-cmd';
57 ## @}
58
59 ksParam_idTestBox = 'TestBoxStatus_idTestBox';
60 ksParam_idGenTestBox = 'TestBoxStatus_idGenTestBox'
61 ksParam_tsUpdated = 'TestBoxStatus_tsUpdated';
62 ksParam_enmState = 'TestBoxStatus_enmState';
63 ksParam_idTestSet = 'TestBoxStatus_idTestSet';
64 ksParam_iWorkItem = 'TestBoxStatus_iWorkItem';
65
66 kasAllowNullAttributes = ['idTestSet', ];
67 kasValidValues_enmState = \
68 [
69 ksTestBoxState_Idle, ksTestBoxState_Testing, ksTestBoxState_GangGathering,
70 ksTestBoxState_GangGatheringTimedOut, ksTestBoxState_GangTesting, ksTestBoxState_GangCleanup,
71 ksTestBoxState_Rebooting, ksTestBoxState_Upgrading, ksTestBoxState_UpgradingAndRebooting,
72 ksTestBoxState_DoingSpecialCmd,
73 ];
74
75 kcDbColumns = 6;
76
77 def __init__(self):
78 ModelDataBase.__init__(self);
79
80 #
81 # Initialize with defaults.
82 # See the database for explanations of each of these fields.
83 #
84 self.idTestBox = None;
85 self.idGenTestBox = None;
86 self.tsUpdated = None;
87 self.enmState = self.ksTestBoxState_Idle;
88 self.idTestSet = None;
89 self.iWorkItem = None;
90
91 def initFromDbRow(self, aoRow):
92 """
93 Internal worker for initFromDbWithId and initFromDbWithGenId as well as
94 TestBoxStatusLogic.
95 """
96
97 if aoRow is None:
98 raise TMRowNotFound('TestBoxStatus not found.');
99
100 self.idTestBox = aoRow[0];
101 self.idGenTestBox = aoRow[1];
102 self.tsUpdated = aoRow[2];
103 self.enmState = aoRow[3];
104 self.idTestSet = aoRow[4];
105 self.iWorkItem = aoRow[5];
106 return self;
107
108 def initFromDbWithId(self, oDb, idTestBox):
109 """
110 Initialize the object from the database.
111 """
112 oDb.execute('SELECT *\n'
113 'FROM TestBoxStatuses\n'
114 'WHERE idTestBox = %s\n'
115 , (idTestBox, ) );
116 return self.initFromDbRow(oDb.fetchOne());
117
118 def initFromDbWithGenId(self, oDb, idGenTestBox):
119 """
120 Initialize the object from the database.
121 """
122 oDb.execute('SELECT *\n'
123 'FROM TestBoxStatuses\n'
124 'WHERE idGenTestBox = %s\n'
125 , (idGenTestBox, ) );
126 return self.initFromDbRow(oDb.fetchOne());
127
128
129class TestBoxStatusLogic(ModelLogicBase):
130 """
131 TestBoxStatus logic.
132 """
133
134 ## The number of seconds between each time to call touchStatus() when
135 # returning CMD_IDLE.
136 kcSecIdleTouchStatus = 120;
137
138
139 def __init__(self, oDb):
140 ModelLogicBase.__init__(self, oDb);
141
142
143 def tryFetchStatus(self, idTestBox):
144 """
145 Attempts to fetch the status of the given testbox.
146
147 Returns a TestBoxStatusData object on success.
148 Returns None if no status was found.
149 Raises exception on other errors.
150 """
151 self._oDb.execute('SELECT *\n'
152 'FROM TestBoxStatuses\n'
153 'WHERE idTestBox = %s\n',
154 (idTestBox,));
155 if self._oDb.getRowCount() == 0:
156 return None;
157 oStatus = TestBoxStatusData();
158 return oStatus.initFromDbRow(self._oDb.fetchOne());
159
160 def tryFetchStatusAndConfig(self, idTestBox, sTestBoxUuid, sTestBoxAddr):
161 """
162 Tries to fetch the testbox status and current testbox config.
163
164 Returns (TestBoxStatusData, TestBoxData) on success, (None, None) if
165 not found. May throw an exception on database error.
166 """
167 self._oDb.execute('SELECT TestBoxStatuses.*,\n'
168 ' TestBoxesWithStrings.*\n'
169 'FROM TestBoxStatuses,\n'
170 ' TestBoxesWithStrings\n'
171 'WHERE TestBoxStatuses.idTestBox = %s\n'
172 ' AND TestBoxesWithStrings.idTestBox = %s\n'
173 ' AND TestBoxesWithStrings.tsExpire = \'infinity\'::TIMESTAMP\n'
174 ' AND TestBoxesWithStrings.uuidSystem = %s\n'
175 ' AND TestBoxesWithStrings.ip = %s\n'
176 , ( idTestBox,
177 idTestBox,
178 sTestBoxUuid,
179 sTestBoxAddr,) );
180 cRows = self._oDb.getRowCount();
181 if cRows != 1:
182 if cRows != 0:
183 raise TMTooManyRows('tryFetchStatusForCommandReq got %s rows for idTestBox=%s' % (cRows, idTestBox));
184 return (None, None);
185 aoRow = self._oDb.fetchOne();
186 return (TestBoxStatusData().initFromDbRow(aoRow[:TestBoxStatusData.kcDbColumns]),
187 TestBoxData().initFromDbRow(aoRow[TestBoxStatusData.kcDbColumns:]));
188
189
190 def insertIdleStatus(self, idTestBox, idGenTestBox, fCommit = False):
191 """
192 Inserts an idle status for the specified testbox.
193 """
194 self._oDb.execute('INSERT INTO TestBoxStatuses (\n'
195 ' idTestBox,\n'
196 ' idGenTestBox,\n'
197 ' enmState,\n'
198 ' idTestSet,\n'
199 ' iWorkItem)\n'
200 'VALUES ( %s,\n'
201 ' %s,\n'
202 ' \'idle\'::TestBoxState_T,\n'
203 ' NULL,\n'
204 ' 0)\n'
205 , (idTestBox, idGenTestBox) );
206 self._oDb.maybeCommit(fCommit);
207 return True;
208
209 def touchStatus(self, idTestBox, fCommit = False):
210 """
211 Touches the testbox status row, i.e. sets tsUpdated to the current time.
212 """
213 self._oDb.execute('UPDATE TestBoxStatuses\n'
214 'SET tsUpdated = CURRENT_TIMESTAMP\n'
215 'WHERE idTestBox = %s\n'
216 , (idTestBox,));
217 self._oDb.maybeCommit(fCommit);
218 return True;
219
220 def updateState(self, idTestBox, sNewState, idTestSet = None, fCommit = False):
221 """
222 Updates the testbox state.
223 """
224 self._oDb.execute('UPDATE TestBoxStatuses\n'
225 'SET enmState = %s,\n'
226 ' idTestSet = %s,\n'
227 ' tsUpdated = CURRENT_TIMESTAMP\n'
228 'WHERE idTestBox = %s\n',
229 (sNewState, idTestSet, idTestBox));
230 self._oDb.maybeCommit(fCommit);
231 return True;
232
233 def updateGangStatus(self, idTestSetGangLeader, sNewState, fCommit = False):
234 """
235 Update the state of all members of a gang.
236 """
237 self._oDb.execute('UPDATE TestBoxStatuses\n'
238 'SET enmState = %s,\n'
239 ' tsUpdated = CURRENT_TIMESTAMP\n'
240 'WHERE idTestBox IN (SELECT idTestBox\n'
241 ' FROM TestSets\n'
242 ' WHERE idTestSetGangLeader = %s)\n'
243 , (sNewState, idTestSetGangLeader,) );
244 self._oDb.maybeCommit(fCommit);
245 return True;
246
247 def updateWorkItem(self, idTestBox, iWorkItem, fCommit = False):
248 """
249 Updates the testbox state.
250 """
251 self._oDb.execute('UPDATE TestBoxStatuses\n'
252 'SET iWorkItem = %s\n'
253 'WHERE idTestBox = %s\n'
254 , ( iWorkItem, idTestBox,));
255 self._oDb.maybeCommit(fCommit);
256 return True;
257
258 def isWholeGangDoneTesting(self, idTestSetGangLeader):
259 """
260 Checks if the whole gang is done testing.
261 """
262 self._oDb.execute('SELECT COUNT(*)\n'
263 'FROM TestBoxStatuses, TestSets\n'
264 'WHERE TestBoxStatuses.idTestSet = TestSets.idTestSet\n'
265 ' AND TestSets.idTestSetGangLeader = %s\n'
266 ' AND TestBoxStatuses.enmState IN (%s, %s)\n'
267 , ( idTestSetGangLeader,
268 TestBoxStatusData.ksTestBoxState_GangGathering,
269 TestBoxStatusData.ksTestBoxState_GangTesting));
270 return self._oDb.fetchOne()[0] == 0;
271
272 def isTheWholeGangThere(self, idTestSetGangLeader):
273 """
274 Checks if the whole gang is done testing.
275 """
276 self._oDb.execute('SELECT COUNT(*)\n'
277 'FROM TestBoxStatuses, TestSets\n'
278 'WHERE TestBoxStatuses.idTestSet = TestSets.idTestSet\n'
279 ' AND TestSets.idTestSetGangLeader = %s\n'
280 ' AND TestBoxStatuses.enmState IN (%s, %s)\n'
281 , ( idTestSetGangLeader,
282 TestBoxStatusData.ksTestBoxState_GangGathering,
283 TestBoxStatusData.ksTestBoxState_GangTesting));
284 return self._oDb.fetchOne()[0] == 0;
285
286 def timeSinceLastChangeInSecs(self, oStatusData):
287 """
288 Figures the time since the last status change.
289 """
290 tsNow = self._oDb.getCurrentTimestamp();
291 oDelta = tsNow - oStatusData.tsUpdated;
292 return oDelta.seconds + oDelta.days * 24 * 3600;
293
294
295#
296# Unit testing.
297#
298
299# pylint: disable=C0111
300class TestBoxStatusDataTestCase(ModelDataBaseTestCase):
301 def setUp(self):
302 self.aoSamples = [TestBoxStatusData(),];
303
304if __name__ == '__main__':
305 unittest.main();
306 # not reached.
307
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette