x/terracotta
Code Issues Proposed changes

Added hostname to the state, added submitting underload requetss

Browse Source
This commit is contained in:
Anton Beloglazov 2012-10-05 11:29:29 +10:00
parent 8f2595528a
commit 2ad9b18b49
2 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
neat/locals/manager.py
View File

@ -104,6 +104,8 @@ from contracts import contract
from neat.contracts_extra import *
import itertools
import requests
from hashlib import sha1
import neat.common as common
from neat.config import *
@ -158,7 +160,8 @@ def init_state(config):
return {'previous_time': 0.,
'vir_connection': vir_connection,
'db': init_db(config['sql_connection']),
'physical_cpu_mhz_total': physical_cpu_mhz_total}
'physical_cpu_mhz_total': physical_cpu_mhz_total,
'hostname': vir_connection.getHostname()}
@contract
@ -263,6 +266,17 @@ def execute(config, state):
if underload:
log.info('Underload detected')
host = 'http://' + config['global_manager_host'] + \
':' + config['global_manager_port']
username = sha1(config['os_admin_user']).hexdigest()
password = sha1(config['os_admin_password']).hexdigest()
log.info('Request: host %s, username %s, password %s',
host, username, password)
requests.put(host,
{'username': username,
'password': password,
'reason': 0,
'host': state['hostname']})
# Send a request to the global manager with the host name
pass
else:

2
tests/locals/test_manager.py
View File

@ -62,12 +62,14 @@ class LocalManager(TestCase):
and_return(db).once()
expect(common).physical_cpu_mhz_total(vir_connection). \
and_return(mhz)
expect(vir_connection).getHostname().and_return('host').once()
config = {'sql_connection': 'db'}
state = manager.init_state(config)
assert state['previous_time'] == 0
assert state['vir_connection'] == vir_connection
assert state['db'] == db
assert state['physical_cpu_mhz_total'] == mhz
assert state['hostname'] == 'host'
@qc(1)
def get_local_data(