Function create_nodes

test/test_shock.py:20–46

(FILELIST)

FILELIST = ["AAA.txt", "BBB.txt", "CCC.txt"] 

def create_nodes(FILELIST):
    NODES = []
    TESTURL = "{SHOCK_URL}/node".format(SHOCK_URL=SHOCK_URL)
    TESTHEADERS = {"Authorization": "OAuth {}".format(TOKEN)}
# to get multipart-form correctly, data has to be specified in this strange way
# and passed as the files= parameter to requests.put
    TESTDATA = {"attributes_str": (None, '{"project_id":"TESTPROJECT"}')}
    for FILE in FILELIST:
        FILES = {'upload': open(DATADIR + FILE, 'rb')}
        if DEBUG:
            print("POST", TESTURL, TESTDATA, TESTHEADERS)
        response = requests.post(TESTURL, headers=TESTHEADERS, files=FILES, data=TESTDATA)
        data = json.loads(response.content.decode("utf-8"))
        assert data["status"] == 200
        assert data["data"]["attributes"]["project_id"] == "TESTPROJECT"
        NODES += [data["data"]["id"]]
        if DEBUG:
            print("PUT", SHOCK_URL + "/node/" + NODES[-1], TESTDATA)
        r = requests.put(SHOCK_URL + "/node/" +
                         NODES[-1], files=TESTDATA, headers=TESTHEADERS)
        if DEBUG:
            print(r.content.decode("utf-8"))
        data = json.loads(r.content.decode("utf-8"))
        if DEBUG:
            print(data)
        assert data["data"]["attributes"]["project_id"] == "TESTPROJECT"
    return(NODES)


def confirm_nodes_project(NODES, PROJECT):