test_files.py

在线编辑器

    assert not files
    
def test_top_level_deletion():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "foo", "data")
    fm.delete(macgyver, bigmac, "foo")
    flist = fm.list_files(macgyver, bigmac)
    assert flist == []
    
def test_directory_deletion():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "whiz/bang", "stillmore")
    starting_used = macgyver.amount_used
    fm.save_file(macgyver, bigmac, "foo/bar", "data")
    fm.save_file(macgyver, bigmac, "foo/blorg", "moredata")
    fm.delete(macgyver, bigmac, "foo/")
    fm.session.clear()
    flist = fm.list_files(macgyver, bigmac)
    assert len(flist) == 1
    assert flist[0].name == 'whiz/'
    file = fm.session.query(File).filter_by(name="foo/bar") \
                        .filter_by(project=bigmac).first()
    assert file is None
    assert macgyver.amount_used == starting_used
    
def test_basic_edit_functions():
    fm = _get_fm()
    s = fm.session
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    bigmac2 = fm.get_project(someone_else, someone_else, "bigmac", create=True)
    fm.save_edit(macgyver, bigmac, "foo/bar/baz", "['edit', 'thinger']")
    fm.save_edit(someone_else, bigmac2, "foo/bar/baz", "['some', 'thing']")
    file_obj = s.query(File).filter_by(name="foo/bar/baz") \
                .filter_by(project=bigmac).one()
    assert len(file_obj.edits) == 1
    
    try:
        content = fm.get_file(macgyver, bigmac, "foo/bar/baz")
        assert False, "Files are not retrievable until the edits are saved"
    except model.FileNotFound:
        pass
        
    files = fm.list_open(macgyver)
    info = files['bigmac']['foo/bar/baz']
    assert info['mode'] == "rw"
    
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz")
    assert edits == ["['edit', 'thinger']"]
    
    fm.save_edit(macgyver, bigmac, "foo/bar/baz", "['second', 'edit']")
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz")
    assert edits == ["['edit', 'thinger']", "['second', 'edit']"]
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz", 1)
    assert edits == ["['second', 'edit']"]
    
    try:
        edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz", 2)
        assert False, "Expected FSException for out-of-bounds start point"
    except model.FSException:
        pass
    
def test_reset_edits():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_edit(macgyver, bigmac, "foo/bar/baz", "['edit', 'thinger']")
    fm.reset_edits(macgyver, bigmac, "foo/bar/baz")
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz")
    assert edits == []
    files = fm.list_open(macgyver)
    assert files == {}
    
    fm.save_edit(macgyver, bigmac, "foo/bar/baz", "['edit', 'thinger']")
    fm.save_edit(macgyver, bigmac, "foo/bar/blork", "['edit', 'thinger']")
    files = fm.list_open(macgyver)
    bigmac_files = files['bigmac']
    assert len(bigmac_files) == 2
    fm.reset_edits(macgyver)
    files = fm.list_open(macgyver)
    assert files == {}
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz")
    assert edits == []
    
def test_edits_cleared_after_save():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_edit(macgyver, bigmac, "foo/bar/baz", "['edit', 'thinger']")
    fm.save_file(macgyver, bigmac, "foo/bar/baz", "macaroni")
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz")
    assert edits == []

def test_edits_cleared_after_close():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "foo/bar/baz", "macaroni")
    fm.get_file(macgyver, bigmac, "foo/bar/baz")
    fm.save_edit(macgyver, bigmac, "foo/bar/baz", "['edit', 'thinger']")
    fm.close(macgyver, bigmac, "foo/bar/baz")
    edits = fm.list_edits(macgyver, bigmac, "foo/bar/baz")
    assert edits == []
    
def test_list_top_level():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "readme.txt", "Hi there!")
    result = fm.list_files(macgyver, bigmac)
    result_names = [file.name for file in result]
    assert result_names == ["readme.txt"]
    result = fm.list_files(macgyver)
    result_names = [proj.name for proj in result]
    assert result_names == ["BespinSettings",
                            "SampleProject", "bigmac"]
    
    
def test_secondary_objects_are_saved_when_creating_new_file():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "foo/bar", "Data")
    project_names = [proj.name for proj in macgyver.projects]
    assert "bigmac" in project_names
    bigmac_obj = fm.session.query(Directory).filter_by(name="")\
                    .filter_by(project=bigmac).one()
    assert bigmac_obj.subdirs[0].name == "foo/"
    
def test_delete_does_not_affect_other_projects():
    global macgyver, someone_else
    fm = _get_fm()
    bigmac1 = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac1, "bar/foo", "bar!")
    fm.save_file(macgyver, bigmac1, "bar/foobar", "yo")
    bigmac2 = fm.get_project(someone_else, someone_else, "bigmac", create=True)
    fm.save_file(someone_else, bigmac2, "other/myfoo", "double bar!")
    fm.save_file(someone_else, bigmac2, "bar/foo", "this one is mine!")
    assert len(macgyver.projects) == 3
    assert len(someone_else.projects) == 3
    
    fm.delete(macgyver, bigmac1, "bar/foo")
    flist = fm.list_files(macgyver, bigmac1, "bar/")
    assert len(flist) == 1
    flist = fm.list_files(someone_else, bigmac2, "bar/")
    assert len(flist) == 1
    
    fm.delete(macgyver, bigmac1)
    
    s = fm.session
    s.expire(macgyver)
    s.expire(someone_else)
    macgyver = s.query(User).filter_by(username='MacGyver').one()
    someone_else = s.query(User).filter_by(username='SomeoneElse').one()
    assert len(macgyver.projects) == 2
    assert len(someone_else.projects) == 3
    flist = fm.list_files(someone_else, bigmac2)
    assert len(flist) == 2
    assert flist[0].name == "bar/"
    file_obj = fm.get_file_object(someone_else, bigmac2, "other/myfoo")
    assert str(file_obj.data) == "double bar!"
    try:
        file_obj = fm.get_file_object(macgyver, bigmac1, "bar/foo")
        assert False, "File should be gone"
    except model.FileNotFound:
        pass
        
def test_save_file_can_create_directory():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "foo/bar/")
    flist = fm.list_files(macgyver, bigmac)
    assert len(flist) == 1
    assert flist[0].name == "foo/"
    flist = fm.list_files(macgyver, bigmac, "foo/")
    assert len(flist) == 1
    assert flist[0].name == "foo/bar/"
    s = fm.session
    file_obj = s.query(File).filter_by(name='foo/bar/').first()
    print file_obj
    assert file_obj is None, "No file should have been created"

# -------
# Web tests
# -------
    
def test_good_file_operations_from_web():
    fm = _get_fm()
    s = fm.session
    app.put("/file/at/bigmac/reqs", "Chewing gum wrapper")
    fileobj = s.query(File).filter_by(name="reqs").one()
    fileobj.created = datetime(2009, 2, 3, 10, 0, 0)
    fileobj.modified = datetime(2009, 2, 4, 11, 30, 30)
    s.commit()
    contents = str(fileobj.data)
    assert contents == "Chewing gum wrapper"
    
    resp = app.get("/file/at/bigmac/reqs")
    assert resp.body == "Chewing gum wrapper"
    resp = app.get("/file/listopen/")
    assert resp.content_type == "application/json"
    data = simplejson.loads(resp.body)
    bigmac_data = data['bigmac']
    assert len(bigmac_data) == 1
    assert bigmac_data['reqs']['mode'] == "rw"
    app.post("/file/close/bigmac/reqs")
    
    resp = app.get("/file/at/bigmac/reqs?mode=r")
    assert resp.body == "Chewing gum wrapper"
    
    resp = app.get("/file/list/bigmac/")
    data = simplejson.loads(resp.body)
    assert data[0]['openedBy'] == ['MacGyver']
    
    resp = app.get("/file/listopen/")
    data = simplejson.loads(resp.body)
    bigmac_data = data['bigmac']
    assert len(bigmac_data) == 1
    assert bigmac_data['reqs']['mode'] == "r"
    app.post("/file/close/bigmac/reqs")
    resp = app.get("/file/listopen/")
    data = simplejson.loads(resp.body)
    assert data == {}
    
    resp = app.get("/file/list/")
    data = simplejson.loads(resp.body)
    assert data == [{'name' : 'BespinSettings/'},
                    {'name' : 'SampleProject/'}, 
                    {'name' : 'bigmac/'}]
    
    resp = app.get("/file/list/SampleProject/")
    data = simplejson.loads(resp.body)
    assert data[1]['name'] == 'index.html'
    
    resp = app.get("/file/list/bigmac/")
    assert resp.content_type == "application/json"
    data = simplejson.loads(resp.body)
    assert len(data) == 1
    data = data[0]
    assert data['name'] == 'reqs'
    assert data['size'] == 19
    assert data['created'] == "20090203T100000"
    assert data['modified'] == '20090204T113030'
    
    app.delete("/file/at/bigmac/reqs")
    resp = app.get("/file/list/bigmac/")
    data = simplejson.loads(resp.body)
    assert data == []
    
    
def test_error_conditions_from_web():
    fm = _get_fm()
    app.get("/file/at/bigmac/UNKNOWN", status=404)
    app.put("/file/at/bigmac/bar/baz", "A file in bar")
    app.put("/file/at/bigmac/bar", "A file to replace bar", status=409)
    app.get("/file/at/bigmac/bar/baz")
    app.get("/file/at/bigmac", status=400)
    app.get("/file/at/bigmac/", status=404)
    app.get("/file/at/", status=400)

def test_edit_interface():
    fm = _get_fm()
    app.put("/edit/at/bigmac/bar/baz", "Starting a file")
    app.put("/edit/at/bigmac/bar/baz", "Second edit")
    resp = app.get("/edit/list/bigmac/bar/baz")
    assert resp.content_type == "application/json"
    data = simplejson.loads(resp.body)
    assert data == ["Starting a file", "Second edit"]
    
    resp = app.get("/edit/recent/1/bigmac/bar/baz")
    assert resp.content_type == "application/json"
    data = simplejson.loads(resp.body)
    assert data == ["Second edit"]
    
    resp = app.get("/file/listopen/")
    data = simplejson.loads(resp.body)
    bigmac_data = data['bigmac']
    assert len(bigmac_data) == 1
    assert bigmac_data['bar/baz']['mode'] == 'rw'
    
    app.post("/edit/reset/")
    resp = app.get("/edit/list/bigmac/bar/baz")
    data = simplejson.loads(resp.body)
    assert data == []
    
    app.put("/edit/at/bigmac/bar/baz", "Starting a file")
    app.post("/edit/reset/bigmac/bar/baz")
    resp = app.get("/edit/list/bigmac/bar/baz")
    data = simplejson.loads(resp.body)
    assert data == []
    
def test_get_file_stats_from_web():
    fm = _get_fm()
    s = fm.session
    app.put("/file/at/bigmac/reqs", "Chewing gum wrapper")
    resp = app.get("/file/stats/bigmac/reqs")
    assert resp.content_type == "application/json"
    data = simplejson.loads(resp.body)
    assert data['size'] == 19
    
def test_preview_mode():
    fm = _get_fm()
    bigmac = fm.get_project(macgyver, macgyver, "bigmac", create=True)
    fm.save_file(macgyver, bigmac, "README.txt", 
        "This is the readme file.")
    fm.session.commit()
    
    resp = app.get("/preview/at/bigmac/", status=404)
    resp = app.get("/preview/at/bigmac/README.txt")
    assert resp.body == "This is the readme file."
    assert resp.content_type == "text/plain"
    
    fm.save_file(macgyver, bigmac, "index.html",
        "<html><body>Simple HTML file</body></html>")
    fm.session.commit()
    resp = app.get("/preview/at/bigmac/index.html")
    assert resp.body == "<html><body>Simple HTML file</body></html>"
    assert resp.content_type == "text/html"
    
def test_quota_limits_on_the_web():
    fm = _get_fm()
    old_units = model.QUOTA_UNITS
    model.QUOTA_UNITS = 10
    try:
        resp = app.put("/file/at/bigmac/foo", "x" * 11, status=400)
        assert resp.body == "Over quota"
    finally:
        model.QUOTA_UNITS = old_units