|
25 | 25 |
|
26 | 26 | class TestAgent(compute_fakes.TestComputev2): |
27 | 27 |
|
28 | | - fake_agent = compute_fakes.FakeAgent.create_one_agent() |
| 28 | + attr = {} |
| 29 | + attr['agent_id'] = 1 |
| 30 | + fake_agent = compute_fakes.FakeAgent.create_one_agent(attr) |
29 | 31 |
|
30 | 32 | columns = ( |
31 | 33 | 'agent_id', |
@@ -238,28 +240,81 @@ def setUp(self): |
238 | 240 | super(TestAgentSet, self).setUp() |
239 | 241 |
|
240 | 242 | self.agents_mock.update.return_value = self.fake_agent |
| 243 | + self.agents_mock.list.return_value = [self.fake_agent] |
241 | 244 | self.cmd = agent.SetAgent(self.app, None) |
242 | 245 |
|
243 | | - def test_agent_set(self): |
| 246 | + def test_agent_set_nothing(self): |
244 | 247 | arglist = [ |
245 | | - 'id', |
246 | | - 'new-version', |
247 | | - 'new-url', |
248 | | - 'new-md5hash', |
| 248 | + '1', |
| 249 | + ] |
| 250 | + verifylist = [ |
| 251 | + ('id', '1'), |
| 252 | + ] |
| 253 | + parsed_args = self.check_parser(self.cmd, arglist, verifylist) |
| 254 | + result = self.cmd.take_action(parsed_args) |
| 255 | + |
| 256 | + self.agents_mock.update.assert_called_with(parsed_args.id, |
| 257 | + self.fake_agent.version, |
| 258 | + self.fake_agent.url, |
| 259 | + self.fake_agent.md5hash) |
| 260 | + self.assertIsNone(result) |
| 261 | + |
| 262 | + def test_agent_set_version(self): |
| 263 | + arglist = [ |
| 264 | + '1', |
| 265 | + '--agent-version', 'new-version', |
249 | 266 | ] |
250 | 267 |
|
251 | 268 | verifylist = [ |
252 | | - ('id', 'id'), |
| 269 | + ('id', '1'), |
253 | 270 | ('version', 'new-version'), |
254 | | - ('url', 'new-url'), |
255 | | - ('md5hash', 'new-md5hash'), |
256 | 271 | ] |
257 | 272 |
|
258 | 273 | parsed_args = self.check_parser(self.cmd, arglist, verifylist) |
259 | 274 | result = self.cmd.take_action(parsed_args) |
260 | 275 |
|
261 | 276 | self.agents_mock.update.assert_called_with(parsed_args.id, |
262 | 277 | parsed_args.version, |
| 278 | + self.fake_agent.url, |
| 279 | + self.fake_agent.md5hash) |
| 280 | + self.assertIsNone(result) |
| 281 | + |
| 282 | + def test_agent_set_url(self): |
| 283 | + arglist = [ |
| 284 | + '1', |
| 285 | + '--url', 'new-url', |
| 286 | + ] |
| 287 | + |
| 288 | + verifylist = [ |
| 289 | + ('id', '1'), |
| 290 | + ('url', 'new-url'), |
| 291 | + ] |
| 292 | + |
| 293 | + parsed_args = self.check_parser(self.cmd, arglist, verifylist) |
| 294 | + result = self.cmd.take_action(parsed_args) |
| 295 | + |
| 296 | + self.agents_mock.update.assert_called_with(parsed_args.id, |
| 297 | + self.fake_agent.version, |
263 | 298 | parsed_args.url, |
| 299 | + self.fake_agent.md5hash) |
| 300 | + self.assertIsNone(result) |
| 301 | + |
| 302 | + def test_agent_set_md5hash(self): |
| 303 | + arglist = [ |
| 304 | + '1', |
| 305 | + '--md5hash', 'new-md5hash', |
| 306 | + ] |
| 307 | + |
| 308 | + verifylist = [ |
| 309 | + ('id', '1'), |
| 310 | + ('md5hash', 'new-md5hash'), |
| 311 | + ] |
| 312 | + |
| 313 | + parsed_args = self.check_parser(self.cmd, arglist, verifylist) |
| 314 | + result = self.cmd.take_action(parsed_args) |
| 315 | + |
| 316 | + self.agents_mock.update.assert_called_with(parsed_args.id, |
| 317 | + self.fake_agent.version, |
| 318 | + self.fake_agent.url, |
264 | 319 | parsed_args.md5hash) |
265 | 320 | self.assertIsNone(result) |
0 commit comments