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()