123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553 |
- // Copyright (C) 2019-2022 Nicola Murino
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published
- // by the Free Software Foundation, version 3.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- // Package httpdtest provides utilities for testing the exposed REST API.
- package httpdtest
- import (
- "bytes"
- "encoding/hex"
- "encoding/json"
- "errors"
- "fmt"
- "io"
- "net/http"
- "net/url"
- "path"
- "strconv"
- "strings"
- "github.com/go-chi/render"
- "github.com/sftpgo/sdk"
- "github.com/drakkan/sftpgo/v2/internal/common"
- "github.com/drakkan/sftpgo/v2/internal/dataprovider"
- "github.com/drakkan/sftpgo/v2/internal/httpclient"
- "github.com/drakkan/sftpgo/v2/internal/httpd"
- "github.com/drakkan/sftpgo/v2/internal/kms"
- "github.com/drakkan/sftpgo/v2/internal/util"
- "github.com/drakkan/sftpgo/v2/internal/version"
- "github.com/drakkan/sftpgo/v2/internal/vfs"
- )
- const (
- tokenPath = "/api/v2/token"
- activeConnectionsPath = "/api/v2/connections"
- quotasBasePath = "/api/v2/quotas"
- quotaScanPath = "/api/v2/quotas/users/scans"
- quotaScanVFolderPath = "/api/v2/quotas/folders/scans"
- userPath = "/api/v2/users"
- groupPath = "/api/v2/groups"
- versionPath = "/api/v2/version"
- folderPath = "/api/v2/folders"
- serverStatusPath = "/api/v2/status"
- dumpDataPath = "/api/v2/dumpdata"
- loadDataPath = "/api/v2/loaddata"
- defenderHosts = "/api/v2/defender/hosts"
- adminPath = "/api/v2/admins"
- adminPwdPath = "/api/v2/admin/changepwd"
- apiKeysPath = "/api/v2/apikeys"
- retentionBasePath = "/api/v2/retention/users"
- retentionChecksPath = "/api/v2/retention/users/checks"
- eventActionsPath = "/api/v2/eventactions"
- eventRulesPath = "/api/v2/eventrules"
- )
- const (
- defaultTokenAuthUser = "admin"
- defaultTokenAuthPass = "password"
- )
- var (
- httpBaseURL = "http://127.0.0.1:8080"
- jwtToken = ""
- )
- // SetBaseURL sets the base url to use for HTTP requests.
- // Default URL is "http://127.0.0.1:8080"
- func SetBaseURL(url string) {
- httpBaseURL = url
- }
- // SetJWTToken sets the JWT token to use
- func SetJWTToken(token string) {
- jwtToken = token
- }
- func sendHTTPRequest(method, url string, body io.Reader, contentType, token string) (*http.Response, error) {
- req, err := http.NewRequest(method, url, body)
- if err != nil {
- return nil, err
- }
- if contentType != "" {
- req.Header.Set("Content-Type", "application/json")
- }
- if token != "" {
- req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", token))
- }
- return httpclient.GetHTTPClient().Do(req)
- }
- func buildURLRelativeToBase(paths ...string) string {
- // we need to use path.Join and not filepath.Join
- // since filepath.Join will use backslash separator on Windows
- p := path.Join(paths...)
- return fmt.Sprintf("%s/%s", strings.TrimRight(httpBaseURL, "/"), strings.TrimLeft(p, "/"))
- }
- // GetToken tries to return a JWT token
- func GetToken(username, password string) (string, map[string]any, error) {
- req, err := http.NewRequest(http.MethodGet, buildURLRelativeToBase(tokenPath), nil)
- if err != nil {
- return "", nil, err
- }
- req.SetBasicAuth(username, password)
- resp, err := httpclient.GetHTTPClient().Do(req)
- if err != nil {
- return "", nil, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, http.StatusOK)
- if err != nil {
- return "", nil, err
- }
- responseHolder := make(map[string]any)
- err = render.DecodeJSON(resp.Body, &responseHolder)
- if err != nil {
- return "", nil, err
- }
- return responseHolder["access_token"].(string), responseHolder, nil
- }
- func getDefaultToken() string {
- if jwtToken != "" {
- return jwtToken
- }
- token, _, err := GetToken(defaultTokenAuthUser, defaultTokenAuthPass)
- if err != nil {
- return ""
- }
- return token
- }
- // AddUser adds a new user and checks the received HTTP Status code against expectedStatusCode.
- func AddUser(user dataprovider.User, expectedStatusCode int) (dataprovider.User, []byte, error) {
- var newUser dataprovider.User
- var body []byte
- userAsJSON, _ := json.Marshal(user)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(userPath), bytes.NewBuffer(userAsJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newUser, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newUser, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newUser)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkUser(&user, &newUser)
- }
- return newUser, body, err
- }
- // UpdateUserWithJSON update a user using the provided JSON as POST body
- func UpdateUserWithJSON(user dataprovider.User, expectedStatusCode int, disconnect string, userAsJSON []byte) (dataprovider.User, []byte, error) {
- var newUser dataprovider.User
- var body []byte
- url, err := addDisconnectQueryParam(buildURLRelativeToBase(userPath, url.PathEscape(user.Username)), disconnect)
- if err != nil {
- return user, body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "application/json",
- getDefaultToken())
- if err != nil {
- return user, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newUser, body, err
- }
- if err == nil {
- newUser, body, err = GetUserByUsername(user.Username, expectedStatusCode)
- }
- if err == nil {
- err = checkUser(&user, &newUser)
- }
- return newUser, body, err
- }
- // UpdateUser updates an existing user and checks the received HTTP Status code against expectedStatusCode.
- func UpdateUser(user dataprovider.User, expectedStatusCode int, disconnect string) (dataprovider.User, []byte, error) {
- userAsJSON, _ := json.Marshal(user)
- return UpdateUserWithJSON(user, expectedStatusCode, disconnect, userAsJSON)
- }
- // RemoveUser removes an existing user and checks the received HTTP Status code against expectedStatusCode.
- func RemoveUser(user dataprovider.User, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(userPath, url.PathEscape(user.Username)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetUserByUsername gets a user by username and checks the received HTTP Status code against expectedStatusCode.
- func GetUserByUsername(username string, expectedStatusCode int) (dataprovider.User, []byte, error) {
- var user dataprovider.User
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(userPath, url.PathEscape(username)),
- nil, "", getDefaultToken())
- if err != nil {
- return user, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &user)
- } else {
- body, _ = getResponseBody(resp)
- }
- return user, body, err
- }
- // GetUsers returns a list of users and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetUsers(limit, offset int64, expectedStatusCode int) ([]dataprovider.User, []byte, error) {
- var users []dataprovider.User
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(userPath), limit, offset)
- if err != nil {
- return users, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return users, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &users)
- } else {
- body, _ = getResponseBody(resp)
- }
- return users, body, err
- }
- // AddGroup adds a new group and checks the received HTTP Status code against expectedStatusCode.
- func AddGroup(group dataprovider.Group, expectedStatusCode int) (dataprovider.Group, []byte, error) {
- var newGroup dataprovider.Group
- var body []byte
- asJSON, _ := json.Marshal(group)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(groupPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newGroup, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newGroup, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newGroup)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkGroup(group, newGroup)
- }
- return newGroup, body, err
- }
- // UpdateGroup updates an existing group and checks the received HTTP Status code against expectedStatusCode
- func UpdateGroup(group dataprovider.Group, expectedStatusCode int) (dataprovider.Group, []byte, error) {
- var newGroup dataprovider.Group
- var body []byte
- asJSON, _ := json.Marshal(group)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(groupPath, url.PathEscape(group.Name)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newGroup, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newGroup, body, err
- }
- if err == nil {
- newGroup, body, err = GetGroupByName(group.Name, expectedStatusCode)
- }
- if err == nil {
- err = checkGroup(group, newGroup)
- }
- return newGroup, body, err
- }
- // RemoveGroup removes an existing group and checks the received HTTP Status code against expectedStatusCode.
- func RemoveGroup(group dataprovider.Group, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(groupPath, url.PathEscape(group.Name)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetGroupByName gets a group by name and checks the received HTTP Status code against expectedStatusCode.
- func GetGroupByName(name string, expectedStatusCode int) (dataprovider.Group, []byte, error) {
- var group dataprovider.Group
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(groupPath, url.PathEscape(name)),
- nil, "", getDefaultToken())
- if err != nil {
- return group, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &group)
- } else {
- body, _ = getResponseBody(resp)
- }
- return group, body, err
- }
- // GetGroups returns a list of groups and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetGroups(limit, offset int64, expectedStatusCode int) ([]dataprovider.Group, []byte, error) {
- var groups []dataprovider.Group
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(groupPath), limit, offset)
- if err != nil {
- return groups, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return groups, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &groups)
- } else {
- body, _ = getResponseBody(resp)
- }
- return groups, body, err
- }
- // AddAdmin adds a new admin and checks the received HTTP Status code against expectedStatusCode.
- func AddAdmin(admin dataprovider.Admin, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
- var newAdmin dataprovider.Admin
- var body []byte
- asJSON, _ := json.Marshal(admin)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(adminPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newAdmin, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newAdmin, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newAdmin)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkAdmin(&admin, &newAdmin)
- }
- return newAdmin, body, err
- }
- // UpdateAdmin updates an existing admin and checks the received HTTP Status code against expectedStatusCode
- func UpdateAdmin(admin dataprovider.Admin, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
- var newAdmin dataprovider.Admin
- var body []byte
- asJSON, _ := json.Marshal(admin)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(adminPath, url.PathEscape(admin.Username)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newAdmin, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newAdmin, body, err
- }
- if err == nil {
- newAdmin, body, err = GetAdminByUsername(admin.Username, expectedStatusCode)
- }
- if err == nil {
- err = checkAdmin(&admin, &newAdmin)
- }
- return newAdmin, body, err
- }
- // RemoveAdmin removes an existing admin and checks the received HTTP Status code against expectedStatusCode.
- func RemoveAdmin(admin dataprovider.Admin, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(adminPath, url.PathEscape(admin.Username)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetAdminByUsername gets an admin by username and checks the received HTTP Status code against expectedStatusCode.
- func GetAdminByUsername(username string, expectedStatusCode int) (dataprovider.Admin, []byte, error) {
- var admin dataprovider.Admin
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(adminPath, url.PathEscape(username)),
- nil, "", getDefaultToken())
- if err != nil {
- return admin, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &admin)
- } else {
- body, _ = getResponseBody(resp)
- }
- return admin, body, err
- }
- // GetAdmins returns a list of admins and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetAdmins(limit, offset int64, expectedStatusCode int) ([]dataprovider.Admin, []byte, error) {
- var admins []dataprovider.Admin
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(adminPath), limit, offset)
- if err != nil {
- return admins, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return admins, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &admins)
- } else {
- body, _ = getResponseBody(resp)
- }
- return admins, body, err
- }
- // ChangeAdminPassword changes the password for an existing admin
- func ChangeAdminPassword(currentPassword, newPassword string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- pwdChange := make(map[string]string)
- pwdChange["current_password"] = currentPassword
- pwdChange["new_password"] = newPassword
- asJSON, _ := json.Marshal(&pwdChange)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(adminPwdPath),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- body, _ = getResponseBody(resp)
- return body, err
- }
- // GetAPIKeys returns a list of API keys and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetAPIKeys(limit, offset int64, expectedStatusCode int) ([]dataprovider.APIKey, []byte, error) {
- var apiKeys []dataprovider.APIKey
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(apiKeysPath), limit, offset)
- if err != nil {
- return apiKeys, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return apiKeys, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &apiKeys)
- } else {
- body, _ = getResponseBody(resp)
- }
- return apiKeys, body, err
- }
- // AddAPIKey adds a new API key and checks the received HTTP Status code against expectedStatusCode.
- func AddAPIKey(apiKey dataprovider.APIKey, expectedStatusCode int) (dataprovider.APIKey, []byte, error) {
- var newAPIKey dataprovider.APIKey
- var body []byte
- asJSON, _ := json.Marshal(apiKey)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(apiKeysPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newAPIKey, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newAPIKey, body, err
- }
- if err != nil {
- body, _ = getResponseBody(resp)
- return newAPIKey, body, err
- }
- response := make(map[string]string)
- err = render.DecodeJSON(resp.Body, &response)
- if err == nil {
- newAPIKey, body, err = GetAPIKeyByID(resp.Header.Get("X-Object-ID"), http.StatusOK)
- }
- if err == nil {
- err = checkAPIKey(&apiKey, &newAPIKey)
- }
- newAPIKey.Key = response["key"]
- return newAPIKey, body, err
- }
- // UpdateAPIKey updates an existing API key and checks the received HTTP Status code against expectedStatusCode
- func UpdateAPIKey(apiKey dataprovider.APIKey, expectedStatusCode int) (dataprovider.APIKey, []byte, error) {
- var newAPIKey dataprovider.APIKey
- var body []byte
- asJSON, _ := json.Marshal(apiKey)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(apiKeysPath, url.PathEscape(apiKey.KeyID)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newAPIKey, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newAPIKey, body, err
- }
- if err == nil {
- newAPIKey, body, err = GetAPIKeyByID(apiKey.KeyID, expectedStatusCode)
- }
- if err == nil {
- err = checkAPIKey(&apiKey, &newAPIKey)
- }
- return newAPIKey, body, err
- }
- // RemoveAPIKey removes an existing API key and checks the received HTTP Status code against expectedStatusCode.
- func RemoveAPIKey(apiKey dataprovider.APIKey, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(apiKeysPath, url.PathEscape(apiKey.KeyID)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetAPIKeyByID gets a API key by ID and checks the received HTTP Status code against expectedStatusCode.
- func GetAPIKeyByID(keyID string, expectedStatusCode int) (dataprovider.APIKey, []byte, error) {
- var apiKey dataprovider.APIKey
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(apiKeysPath, url.PathEscape(keyID)),
- nil, "", getDefaultToken())
- if err != nil {
- return apiKey, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &apiKey)
- } else {
- body, _ = getResponseBody(resp)
- }
- return apiKey, body, err
- }
- // AddEventAction adds a new event action
- func AddEventAction(action dataprovider.BaseEventAction, expectedStatusCode int) (dataprovider.BaseEventAction, []byte, error) {
- var newAction dataprovider.BaseEventAction
- var body []byte
- asJSON, _ := json.Marshal(action)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(eventActionsPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newAction, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newAction, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newAction)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkEventAction(action, newAction)
- }
- return newAction, body, err
- }
- // UpdateEventAction updates an existing event action
- func UpdateEventAction(action dataprovider.BaseEventAction, expectedStatusCode int) (dataprovider.BaseEventAction, []byte, error) {
- var newAction dataprovider.BaseEventAction
- var body []byte
- asJSON, _ := json.Marshal(action)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(eventActionsPath, url.PathEscape(action.Name)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newAction, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newAction, body, err
- }
- if err == nil {
- newAction, body, err = GetEventActionByName(action.Name, expectedStatusCode)
- }
- if err == nil {
- err = checkEventAction(action, newAction)
- }
- return newAction, body, err
- }
- // RemoveEventAction removes an existing action and checks the received HTTP Status code against expectedStatusCode.
- func RemoveEventAction(action dataprovider.BaseEventAction, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(eventActionsPath, url.PathEscape(action.Name)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetEventActionByName gets an event action by name and checks the received HTTP Status code against expectedStatusCode.
- func GetEventActionByName(name string, expectedStatusCode int) (dataprovider.BaseEventAction, []byte, error) {
- var action dataprovider.BaseEventAction
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(eventActionsPath, url.PathEscape(name)),
- nil, "", getDefaultToken())
- if err != nil {
- return action, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &action)
- } else {
- body, _ = getResponseBody(resp)
- }
- return action, body, err
- }
- // GetEventActions returns a list of event actions and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetEventActions(limit, offset int64, expectedStatusCode int) ([]dataprovider.BaseEventAction, []byte, error) {
- var actions []dataprovider.BaseEventAction
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(eventActionsPath), limit, offset)
- if err != nil {
- return actions, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return actions, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &actions)
- } else {
- body, _ = getResponseBody(resp)
- }
- return actions, body, err
- }
- // AddEventRule adds a new event rule
- func AddEventRule(rule dataprovider.EventRule, expectedStatusCode int) (dataprovider.EventRule, []byte, error) {
- var newRule dataprovider.EventRule
- var body []byte
- asJSON, _ := json.Marshal(rule)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(eventRulesPath), bytes.NewBuffer(asJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newRule, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newRule, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newRule)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkEventRule(rule, newRule)
- }
- return newRule, body, err
- }
- // UpdateEventRule updates an existing event rule
- func UpdateEventRule(rule dataprovider.EventRule, expectedStatusCode int) (dataprovider.EventRule, []byte, error) {
- var newRule dataprovider.EventRule
- var body []byte
- asJSON, _ := json.Marshal(rule)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(eventRulesPath, url.PathEscape(rule.Name)),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return newRule, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return newRule, body, err
- }
- if err == nil {
- newRule, body, err = GetEventRuleByName(rule.Name, expectedStatusCode)
- }
- if err == nil {
- err = checkEventRule(rule, newRule)
- }
- return newRule, body, err
- }
- // RemoveEventRule removes an existing rule and checks the received HTTP Status code against expectedStatusCode.
- func RemoveEventRule(rule dataprovider.EventRule, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(eventRulesPath, url.PathEscape(rule.Name)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetEventRuleByName gets an event rule by name and checks the received HTTP Status code against expectedStatusCode.
- func GetEventRuleByName(name string, expectedStatusCode int) (dataprovider.EventRule, []byte, error) {
- var rule dataprovider.EventRule
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(eventRulesPath, url.PathEscape(name)),
- nil, "", getDefaultToken())
- if err != nil {
- return rule, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &rule)
- } else {
- body, _ = getResponseBody(resp)
- }
- return rule, body, err
- }
- // GetEventRules returns a list of event rules and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- func GetEventRules(limit, offset int64, expectedStatusCode int) ([]dataprovider.EventRule, []byte, error) {
- var rules []dataprovider.EventRule
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(eventRulesPath), limit, offset)
- if err != nil {
- return rules, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return rules, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &rules)
- } else {
- body, _ = getResponseBody(resp)
- }
- return rules, body, err
- }
- // GetQuotaScans gets active quota scans for users and checks the received HTTP Status code against expectedStatusCode.
- func GetQuotaScans(expectedStatusCode int) ([]common.ActiveQuotaScan, []byte, error) {
- var quotaScans []common.ActiveQuotaScan
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(quotaScanPath), nil, "", getDefaultToken())
- if err != nil {
- return quotaScans, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, "aScans)
- } else {
- body, _ = getResponseBody(resp)
- }
- return quotaScans, body, err
- }
- // StartQuotaScan starts a new quota scan for the given user and checks the received HTTP Status code against expectedStatusCode.
- func StartQuotaScan(user dataprovider.User, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(quotasBasePath, "users", user.Username, "scan"),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // UpdateQuotaUsage updates the user used quota limits and checks the received
- // HTTP Status code against expectedStatusCode.
- func UpdateQuotaUsage(user dataprovider.User, mode string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- userAsJSON, _ := json.Marshal(user)
- url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "users", user.Username, "usage"), mode)
- if err != nil {
- return body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "application/json",
- getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // UpdateTransferQuotaUsage updates the user used transfer quota limits and checks the received
- // HTTP Status code against expectedStatusCode.
- func UpdateTransferQuotaUsage(user dataprovider.User, mode string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- userAsJSON, _ := json.Marshal(user)
- url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "users", user.Username, "transfer-usage"), mode)
- if err != nil {
- return body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(userAsJSON), "application/json",
- getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetRetentionChecks returns the active retention checks
- func GetRetentionChecks(expectedStatusCode int) ([]common.ActiveRetentionChecks, []byte, error) {
- var checks []common.ActiveRetentionChecks
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(retentionChecksPath), nil, "", getDefaultToken())
- if err != nil {
- return checks, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &checks)
- } else {
- body, _ = getResponseBody(resp)
- }
- return checks, body, err
- }
- // StartRetentionCheck starts a new retention check
- func StartRetentionCheck(username string, retention []dataprovider.FolderRetention, expectedStatusCode int) ([]byte, error) {
- var body []byte
- asJSON, _ := json.Marshal(retention)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(retentionBasePath, username, "check"),
- bytes.NewBuffer(asJSON), "application/json", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetConnections returns status and stats for active SFTP/SCP connections
- func GetConnections(expectedStatusCode int) ([]common.ConnectionStatus, []byte, error) {
- var connections []common.ConnectionStatus
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(activeConnectionsPath), nil, "", getDefaultToken())
- if err != nil {
- return connections, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &connections)
- } else {
- body, _ = getResponseBody(resp)
- }
- return connections, body, err
- }
- // CloseConnection closes an active connection identified by connectionID
- func CloseConnection(connectionID string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(activeConnectionsPath, connectionID),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- body, _ = getResponseBody(resp)
- return body, err
- }
- // AddFolder adds a new folder and checks the received HTTP Status code against expectedStatusCode
- func AddFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
- var newFolder vfs.BaseVirtualFolder
- var body []byte
- folderAsJSON, _ := json.Marshal(folder)
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(folderPath), bytes.NewBuffer(folderAsJSON),
- "application/json", getDefaultToken())
- if err != nil {
- return newFolder, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusCreated {
- body, _ = getResponseBody(resp)
- return newFolder, body, err
- }
- if err == nil {
- err = render.DecodeJSON(resp.Body, &newFolder)
- } else {
- body, _ = getResponseBody(resp)
- }
- if err == nil {
- err = checkFolder(&folder, &newFolder)
- }
- return newFolder, body, err
- }
- // UpdateFolder updates an existing folder and checks the received HTTP Status code against expectedStatusCode.
- func UpdateFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
- var updatedFolder vfs.BaseVirtualFolder
- var body []byte
- folderAsJSON, _ := json.Marshal(folder)
- resp, err := sendHTTPRequest(http.MethodPut, buildURLRelativeToBase(folderPath, url.PathEscape(folder.Name)),
- bytes.NewBuffer(folderAsJSON), "application/json", getDefaultToken())
- if err != nil {
- return updatedFolder, body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if expectedStatusCode != http.StatusOK {
- return updatedFolder, body, err
- }
- if err == nil {
- updatedFolder, body, err = GetFolderByName(folder.Name, expectedStatusCode)
- }
- if err == nil {
- err = checkFolder(&folder, &updatedFolder)
- }
- return updatedFolder, body, err
- }
- // RemoveFolder removes an existing user and checks the received HTTP Status code against expectedStatusCode.
- func RemoveFolder(folder vfs.BaseVirtualFolder, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(folderPath, url.PathEscape(folder.Name)),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetFolderByName gets a folder by name and checks the received HTTP Status code against expectedStatusCode.
- func GetFolderByName(name string, expectedStatusCode int) (vfs.BaseVirtualFolder, []byte, error) {
- var folder vfs.BaseVirtualFolder
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(folderPath, url.PathEscape(name)),
- nil, "", getDefaultToken())
- if err != nil {
- return folder, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &folder)
- } else {
- body, _ = getResponseBody(resp)
- }
- return folder, body, err
- }
- // GetFolders returns a list of folders and checks the received HTTP Status code against expectedStatusCode.
- // The number of results can be limited specifying a limit.
- // Some results can be skipped specifying an offset.
- // The results can be filtered specifying a folder path, the folder path filter is an exact match
- func GetFolders(limit int64, offset int64, expectedStatusCode int) ([]vfs.BaseVirtualFolder, []byte, error) {
- var folders []vfs.BaseVirtualFolder
- var body []byte
- url, err := addLimitAndOffsetQueryParams(buildURLRelativeToBase(folderPath), limit, offset)
- if err != nil {
- return folders, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return folders, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &folders)
- } else {
- body, _ = getResponseBody(resp)
- }
- return folders, body, err
- }
- // GetFoldersQuotaScans gets active quota scans for folders and checks the received HTTP Status code against expectedStatusCode.
- func GetFoldersQuotaScans(expectedStatusCode int) ([]common.ActiveVirtualFolderQuotaScan, []byte, error) {
- var quotaScans []common.ActiveVirtualFolderQuotaScan
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(quotaScanVFolderPath), nil, "", getDefaultToken())
- if err != nil {
- return quotaScans, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, "aScans)
- } else {
- body, _ = getResponseBody(resp)
- }
- return quotaScans, body, err
- }
- // StartFolderQuotaScan start a new quota scan for the given folder and checks the received HTTP Status code against expectedStatusCode.
- func StartFolderQuotaScan(folder vfs.BaseVirtualFolder, expectedStatusCode int) ([]byte, error) {
- var body []byte
- resp, err := sendHTTPRequest(http.MethodPost, buildURLRelativeToBase(quotasBasePath, "folders", folder.Name, "scan"),
- nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // UpdateFolderQuotaUsage updates the folder used quota limits and checks the received HTTP Status code against expectedStatusCode.
- func UpdateFolderQuotaUsage(folder vfs.BaseVirtualFolder, mode string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- folderAsJSON, _ := json.Marshal(folder)
- url, err := addModeQueryParam(buildURLRelativeToBase(quotasBasePath, "folders", folder.Name, "usage"), mode)
- if err != nil {
- return body, err
- }
- resp, err := sendHTTPRequest(http.MethodPut, url.String(), bytes.NewBuffer(folderAsJSON), "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // GetVersion returns version details
- func GetVersion(expectedStatusCode int) (version.Info, []byte, error) {
- var appVersion version.Info
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(versionPath), nil, "", getDefaultToken())
- if err != nil {
- return appVersion, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &appVersion)
- } else {
- body, _ = getResponseBody(resp)
- }
- return appVersion, body, err
- }
- // GetStatus returns the server status
- func GetStatus(expectedStatusCode int) (httpd.ServicesStatus, []byte, error) {
- var response httpd.ServicesStatus
- var body []byte
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(serverStatusPath), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && (expectedStatusCode == http.StatusOK) {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // GetDefenderHosts returns hosts that are banned or for which some violations have been detected
- func GetDefenderHosts(expectedStatusCode int) ([]dataprovider.DefenderEntry, []byte, error) {
- var response []dataprovider.DefenderEntry
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(defenderHosts))
- if err != nil {
- return response, body, err
- }
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // GetDefenderHostByIP returns the host with the given IP, if it exists
- func GetDefenderHostByIP(ip string, expectedStatusCode int) (dataprovider.DefenderEntry, []byte, error) {
- var host dataprovider.DefenderEntry
- var body []byte
- id := hex.EncodeToString([]byte(ip))
- resp, err := sendHTTPRequest(http.MethodGet, buildURLRelativeToBase(defenderHosts, id),
- nil, "", getDefaultToken())
- if err != nil {
- return host, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &host)
- } else {
- body, _ = getResponseBody(resp)
- }
- return host, body, err
- }
- // RemoveDefenderHostByIP removes the host with the given IP from the defender list
- func RemoveDefenderHostByIP(ip string, expectedStatusCode int) ([]byte, error) {
- var body []byte
- id := hex.EncodeToString([]byte(ip))
- resp, err := sendHTTPRequest(http.MethodDelete, buildURLRelativeToBase(defenderHosts, id), nil, "", getDefaultToken())
- if err != nil {
- return body, err
- }
- defer resp.Body.Close()
- body, _ = getResponseBody(resp)
- return body, checkResponse(resp.StatusCode, expectedStatusCode)
- }
- // Dumpdata requests a backup to outputFile.
- // outputFile is relative to the configured backups_path
- func Dumpdata(outputFile, outputData, indent string, expectedStatusCode int) (map[string]any, []byte, error) {
- var response map[string]any
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(dumpDataPath))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- if outputData != "" {
- q.Add("output-data", outputData)
- }
- if outputFile != "" {
- q.Add("output-file", outputFile)
- }
- if indent != "" {
- q.Add("indent", indent)
- }
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // Loaddata restores a backup.
- func Loaddata(inputFile, scanQuota, mode string, expectedStatusCode int) (map[string]any, []byte, error) {
- var response map[string]any
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(loadDataPath))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- q.Add("input-file", inputFile)
- if scanQuota != "" {
- q.Add("scan-quota", scanQuota)
- }
- if mode != "" {
- q.Add("mode", mode)
- }
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodGet, url.String(), nil, "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- // LoaddataFromPostBody restores a backup
- func LoaddataFromPostBody(data []byte, scanQuota, mode string, expectedStatusCode int) (map[string]any, []byte, error) {
- var response map[string]any
- var body []byte
- url, err := url.Parse(buildURLRelativeToBase(loadDataPath))
- if err != nil {
- return response, body, err
- }
- q := url.Query()
- if scanQuota != "" {
- q.Add("scan-quota", scanQuota)
- }
- if mode != "" {
- q.Add("mode", mode)
- }
- url.RawQuery = q.Encode()
- resp, err := sendHTTPRequest(http.MethodPost, url.String(), bytes.NewReader(data), "", getDefaultToken())
- if err != nil {
- return response, body, err
- }
- defer resp.Body.Close()
- err = checkResponse(resp.StatusCode, expectedStatusCode)
- if err == nil && expectedStatusCode == http.StatusOK {
- err = render.DecodeJSON(resp.Body, &response)
- } else {
- body, _ = getResponseBody(resp)
- }
- return response, body, err
- }
- func checkResponse(actual int, expected int) error {
- if expected != actual {
- return fmt.Errorf("wrong status code: got %v want %v", actual, expected)
- }
- return nil
- }
- func getResponseBody(resp *http.Response) ([]byte, error) {
- return io.ReadAll(resp.Body)
- }
- func checkEventAction(expected, actual dataprovider.BaseEventAction) error {
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual action ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("action ID mismatch")
- }
- }
- if dataprovider.ConvertName(expected.Name) != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if expected.Type != actual.Type {
- return errors.New("type mismatch")
- }
- if err := compareEventActionCmdConfigFields(expected.Options.CmdConfig, actual.Options.CmdConfig); err != nil {
- return err
- }
- if err := compareEventActionEmailConfigFields(expected.Options.EmailConfig, actual.Options.EmailConfig); err != nil {
- return err
- }
- if err := compareEventActionDataRetentionFields(expected.Options.RetentionConfig, actual.Options.RetentionConfig); err != nil {
- return err
- }
- if err := compareEventActionFsConfigFields(expected.Options.FsConfig, actual.Options.FsConfig); err != nil {
- return err
- }
- return compareEventActionHTTPConfigFields(expected.Options.HTTPConfig, actual.Options.HTTPConfig)
- }
- func checkEventSchedules(expected, actual []dataprovider.Schedule) error {
- if len(expected) != len(actual) {
- return errors.New("schedules mismatch")
- }
- for _, ex := range expected {
- found := false
- for _, ac := range actual {
- if ac.DayOfMonth == ex.DayOfMonth && ac.DayOfWeek == ex.DayOfWeek && ac.Hours == ex.Hours && ac.Month == ex.Month {
- found = true
- break
- }
- }
- if !found {
- return errors.New("schedules content mismatch")
- }
- }
- return nil
- }
- func compareConditionPatternOptions(expected, actual []dataprovider.ConditionPattern) error {
- if len(expected) != len(actual) {
- return errors.New("condition pattern mismatch")
- }
- for _, ex := range expected {
- found := false
- for _, ac := range actual {
- if ac.Pattern == ex.Pattern && ac.InverseMatch == ex.InverseMatch {
- found = true
- break
- }
- }
- if !found {
- return errors.New("condition pattern content mismatch")
- }
- }
- return nil
- }
- func checkEventConditionOptions(expected, actual dataprovider.ConditionOptions) error {
- if err := compareConditionPatternOptions(expected.Names, actual.Names); err != nil {
- return errors.New("condition names mismatch")
- }
- if err := compareConditionPatternOptions(expected.GroupNames, actual.GroupNames); err != nil {
- return errors.New("condition group names mismatch")
- }
- if err := compareConditionPatternOptions(expected.FsPaths, actual.FsPaths); err != nil {
- return errors.New("condition fs_paths mismatch")
- }
- if len(expected.Protocols) != len(actual.Protocols) {
- return errors.New("condition protocols mismatch")
- }
- for _, v := range expected.Protocols {
- if !util.Contains(actual.Protocols, v) {
- return errors.New("condition protocols content mismatch")
- }
- }
- if len(expected.ProviderObjects) != len(actual.ProviderObjects) {
- return errors.New("condition provider objects mismatch")
- }
- for _, v := range expected.ProviderObjects {
- if !util.Contains(actual.ProviderObjects, v) {
- return errors.New("condition provider objects content mismatch")
- }
- }
- if expected.MinFileSize != actual.MinFileSize {
- return errors.New("condition min file size mismatch")
- }
- if expected.MaxFileSize != actual.MaxFileSize {
- return errors.New("condition max file size mismatch")
- }
- return nil
- }
- func checkEventConditions(expected, actual dataprovider.EventConditions) error {
- if len(expected.FsEvents) != len(actual.FsEvents) {
- return errors.New("fs events mismatch")
- }
- for _, v := range expected.FsEvents {
- if !util.Contains(actual.FsEvents, v) {
- return errors.New("fs events content mismatch")
- }
- }
- if len(expected.ProviderEvents) != len(actual.ProviderEvents) {
- return errors.New("provider events mismatch")
- }
- for _, v := range expected.ProviderEvents {
- if !util.Contains(actual.ProviderEvents, v) {
- return errors.New("provider events content mismatch")
- }
- }
- if err := checkEventConditionOptions(expected.Options, actual.Options); err != nil {
- return err
- }
- return checkEventSchedules(expected.Schedules, actual.Schedules)
- }
- func checkEventRuleActions(expected, actual []dataprovider.EventAction) error {
- if len(expected) != len(actual) {
- return errors.New("actions mismatch")
- }
- for _, ex := range expected {
- found := false
- for _, ac := range actual {
- if ex.Name == ac.Name && ex.Order == ac.Order && ex.Options.ExecuteSync == ac.Options.ExecuteSync &&
- ex.Options.IsFailureAction == ac.Options.IsFailureAction && ex.Options.StopOnFailure == ac.Options.StopOnFailure {
- found = true
- break
- }
- }
- if !found {
- return errors.New("actions contents mismatch")
- }
- }
- return nil
- }
- func checkEventRule(expected, actual dataprovider.EventRule) error {
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual group ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("group ID mismatch")
- }
- }
- if dataprovider.ConvertName(expected.Name) != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if actual.CreatedAt == 0 {
- return errors.New("created_at unset")
- }
- if actual.UpdatedAt == 0 {
- return errors.New("updated_at unset")
- }
- if expected.Trigger != actual.Trigger {
- return errors.New("trigger mismatch")
- }
- if err := checkEventConditions(expected.Conditions, actual.Conditions); err != nil {
- return err
- }
- return checkEventRuleActions(expected.Actions, actual.Actions)
- }
- func checkGroup(expected, actual dataprovider.Group) error {
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual group ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("group ID mismatch")
- }
- }
- if dataprovider.ConvertName(expected.Name) != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if actual.CreatedAt == 0 {
- return errors.New("created_at unset")
- }
- if actual.UpdatedAt == 0 {
- return errors.New("updated_at unset")
- }
- if err := compareEqualGroupSettingsFields(expected.UserSettings.BaseGroupUserSettings,
- actual.UserSettings.BaseGroupUserSettings); err != nil {
- return err
- }
- if err := compareVirtualFolders(expected.VirtualFolders, actual.VirtualFolders); err != nil {
- return err
- }
- if err := compareUserFilters(expected.UserSettings.Filters, actual.UserSettings.Filters); err != nil {
- return err
- }
- return compareFsConfig(&expected.UserSettings.FsConfig, &actual.UserSettings.FsConfig)
- }
- func checkFolder(expected *vfs.BaseVirtualFolder, actual *vfs.BaseVirtualFolder) error {
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual folder ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("folder ID mismatch")
- }
- }
- if dataprovider.ConvertName(expected.Name) != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.MappedPath != actual.MappedPath {
- return errors.New("mapped path mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- return compareFsConfig(&expected.FsConfig, &actual.FsConfig)
- }
- func checkAPIKey(expected, actual *dataprovider.APIKey) error {
- if actual.Key != "" {
- return errors.New("key must not be visible")
- }
- if actual.KeyID == "" {
- return errors.New("actual key_id cannot be empty")
- }
- if expected.Name != actual.Name {
- return errors.New("name mismatch")
- }
- if expected.Scope != actual.Scope {
- return errors.New("scope mismatch")
- }
- if actual.CreatedAt == 0 {
- return errors.New("created_at cannot be 0")
- }
- if actual.UpdatedAt == 0 {
- return errors.New("updated_at cannot be 0")
- }
- if expected.ExpiresAt != actual.ExpiresAt {
- return errors.New("expires_at mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if expected.User != actual.User {
- return errors.New("user mismatch")
- }
- if expected.Admin != actual.Admin {
- return errors.New("admin mismatch")
- }
- return nil
- }
- func checkAdmin(expected, actual *dataprovider.Admin) error {
- if actual.Password != "" {
- return errors.New("admin password must not be visible")
- }
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual admin ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("admin ID mismatch")
- }
- }
- if expected.CreatedAt > 0 {
- if expected.CreatedAt != actual.CreatedAt {
- return fmt.Errorf("created_at mismatch %v != %v", expected.CreatedAt, actual.CreatedAt)
- }
- }
- if err := compareAdminEqualFields(expected, actual); err != nil {
- return err
- }
- if len(expected.Permissions) != len(actual.Permissions) {
- return errors.New("permissions mismatch")
- }
- for _, p := range expected.Permissions {
- if !util.Contains(actual.Permissions, p) {
- return errors.New("permissions content mismatch")
- }
- }
- if err := compareAdminFilters(expected.Filters, actual.Filters); err != nil {
- return err
- }
- return compareAdminGroups(expected, actual)
- }
- func compareAdminFilters(expected, actual dataprovider.AdminFilters) error {
- if expected.AllowAPIKeyAuth != actual.AllowAPIKeyAuth {
- return errors.New("allow_api_key_auth mismatch")
- }
- if len(expected.AllowList) != len(actual.AllowList) {
- return errors.New("allow list mismatch")
- }
- for _, v := range expected.AllowList {
- if !util.Contains(actual.AllowList, v) {
- return errors.New("allow list content mismatch")
- }
- }
- if expected.Preferences.HideUserPageSections != actual.Preferences.HideUserPageSections {
- return errors.New("hide user page sections mismatch")
- }
- return nil
- }
- func compareAdminEqualFields(expected *dataprovider.Admin, actual *dataprovider.Admin) error {
- if dataprovider.ConvertName(expected.Username) != actual.Username {
- return errors.New("sername mismatch")
- }
- if expected.Email != actual.Email {
- return errors.New("email mismatch")
- }
- if expected.Status != actual.Status {
- return errors.New("status mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- if expected.AdditionalInfo != actual.AdditionalInfo {
- return errors.New("additional info mismatch")
- }
- return nil
- }
- func checkUser(expected *dataprovider.User, actual *dataprovider.User) error {
- if actual.Password != "" {
- return errors.New("user password must not be visible")
- }
- if expected.ID <= 0 {
- if actual.ID <= 0 {
- return errors.New("actual user ID must be > 0")
- }
- } else {
- if actual.ID != expected.ID {
- return errors.New("user ID mismatch")
- }
- }
- if expected.CreatedAt > 0 {
- if expected.CreatedAt != actual.CreatedAt {
- return fmt.Errorf("created_at mismatch %v != %v", expected.CreatedAt, actual.CreatedAt)
- }
- }
- if expected.Email != actual.Email {
- return errors.New("email mismatch")
- }
- if err := compareUserPermissions(expected.Permissions, actual.Permissions); err != nil {
- return err
- }
- if err := compareUserFilters(expected.Filters.BaseUserFilters, actual.Filters.BaseUserFilters); err != nil {
- return err
- }
- if err := compareFsConfig(&expected.FsConfig, &actual.FsConfig); err != nil {
- return err
- }
- if err := compareUserGroups(expected, actual); err != nil {
- return err
- }
- if err := compareVirtualFolders(expected.VirtualFolders, actual.VirtualFolders); err != nil {
- return err
- }
- return compareEqualsUserFields(expected, actual)
- }
- func compareUserPermissions(expected map[string][]string, actual map[string][]string) error {
- if len(expected) != len(actual) {
- return errors.New("permissions mismatch")
- }
- for dir, perms := range expected {
- if actualPerms, ok := actual[dir]; ok {
- for _, v := range actualPerms {
- if !util.Contains(perms, v) {
- return errors.New("permissions contents mismatch")
- }
- }
- } else {
- return errors.New("permissions directories mismatch")
- }
- }
- return nil
- }
- func compareAdminGroups(expected *dataprovider.Admin, actual *dataprovider.Admin) error {
- if len(actual.Groups) != len(expected.Groups) {
- return errors.New("groups len mismatch")
- }
- for _, g := range actual.Groups {
- found := false
- for _, g1 := range expected.Groups {
- if g1.Name == g.Name {
- found = true
- if g1.Options.AddToUsersAs != g.Options.AddToUsersAs {
- return fmt.Errorf("add to users as field mismatch for group %s", g.Name)
- }
- }
- }
- if !found {
- return errors.New("groups mismatch")
- }
- }
- return nil
- }
- func compareUserGroups(expected *dataprovider.User, actual *dataprovider.User) error {
- if len(actual.Groups) != len(expected.Groups) {
- return errors.New("groups len mismatch")
- }
- for _, g := range actual.Groups {
- found := false
- for _, g1 := range expected.Groups {
- if g1.Name == g.Name {
- found = true
- if g1.Type != g.Type {
- return fmt.Errorf("type mismatch for group %s", g.Name)
- }
- }
- }
- if !found {
- return errors.New("groups mismatch")
- }
- }
- return nil
- }
- func compareVirtualFolders(expected []vfs.VirtualFolder, actual []vfs.VirtualFolder) error {
- if len(actual) != len(expected) {
- return errors.New("virtual folders len mismatch")
- }
- for _, v := range actual {
- found := false
- for _, v1 := range expected {
- if path.Clean(v.VirtualPath) == path.Clean(v1.VirtualPath) {
- if err := checkFolder(&v1.BaseVirtualFolder, &v.BaseVirtualFolder); err != nil {
- return err
- }
- if v.QuotaSize != v1.QuotaSize {
- return errors.New("vfolder quota size mismatch")
- }
- if (v.QuotaFiles) != (v1.QuotaFiles) {
- return errors.New("vfolder quota files mismatch")
- }
- found = true
- break
- }
- }
- if !found {
- return errors.New("virtual folders mismatch")
- }
- }
- return nil
- }
- func compareFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.Provider != actual.Provider {
- return errors.New("fs provider mismatch")
- }
- if err := compareS3Config(expected, actual); err != nil {
- return err
- }
- if err := compareGCSConfig(expected, actual); err != nil {
- return err
- }
- if err := compareAzBlobConfig(expected, actual); err != nil {
- return err
- }
- if err := checkEncryptedSecret(expected.CryptConfig.Passphrase, actual.CryptConfig.Passphrase); err != nil {
- return err
- }
- if err := compareSFTPFsConfig(expected, actual); err != nil {
- return err
- }
- return compareHTTPFsConfig(expected, actual)
- }
- func compareS3Config(expected *vfs.Filesystem, actual *vfs.Filesystem) error { //nolint:gocyclo
- if expected.S3Config.Bucket != actual.S3Config.Bucket {
- return errors.New("fs S3 bucket mismatch")
- }
- if expected.S3Config.Region != actual.S3Config.Region {
- return errors.New("fs S3 region mismatch")
- }
- if expected.S3Config.AccessKey != actual.S3Config.AccessKey {
- return errors.New("fs S3 access key mismatch")
- }
- if expected.S3Config.RoleARN != actual.S3Config.RoleARN {
- return errors.New("fs S3 role ARN mismatch")
- }
- if err := checkEncryptedSecret(expected.S3Config.AccessSecret, actual.S3Config.AccessSecret); err != nil {
- return fmt.Errorf("fs S3 access secret mismatch: %v", err)
- }
- if expected.S3Config.Endpoint != actual.S3Config.Endpoint {
- return errors.New("fs S3 endpoint mismatch")
- }
- if expected.S3Config.StorageClass != actual.S3Config.StorageClass {
- return errors.New("fs S3 storage class mismatch")
- }
- if expected.S3Config.ACL != actual.S3Config.ACL {
- return errors.New("fs S3 ACL mismatch")
- }
- if expected.S3Config.UploadPartSize != actual.S3Config.UploadPartSize {
- return errors.New("fs S3 upload part size mismatch")
- }
- if expected.S3Config.UploadConcurrency != actual.S3Config.UploadConcurrency {
- return errors.New("fs S3 upload concurrency mismatch")
- }
- if expected.S3Config.DownloadPartSize != actual.S3Config.DownloadPartSize {
- return errors.New("fs S3 download part size mismatch")
- }
- if expected.S3Config.DownloadConcurrency != actual.S3Config.DownloadConcurrency {
- return errors.New("fs S3 download concurrency mismatch")
- }
- if expected.S3Config.ForcePathStyle != actual.S3Config.ForcePathStyle {
- return errors.New("fs S3 force path style mismatch")
- }
- if expected.S3Config.DownloadPartMaxTime != actual.S3Config.DownloadPartMaxTime {
- return errors.New("fs S3 download part max time mismatch")
- }
- if expected.S3Config.UploadPartMaxTime != actual.S3Config.UploadPartMaxTime {
- return errors.New("fs S3 upload part max time mismatch")
- }
- if expected.S3Config.KeyPrefix != actual.S3Config.KeyPrefix &&
- expected.S3Config.KeyPrefix+"/" != actual.S3Config.KeyPrefix {
- return errors.New("fs S3 key prefix mismatch")
- }
- return nil
- }
- func compareGCSConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.GCSConfig.Bucket != actual.GCSConfig.Bucket {
- return errors.New("GCS bucket mismatch")
- }
- if expected.GCSConfig.StorageClass != actual.GCSConfig.StorageClass {
- return errors.New("GCS storage class mismatch")
- }
- if expected.GCSConfig.ACL != actual.GCSConfig.ACL {
- return errors.New("GCS ACL mismatch")
- }
- if expected.GCSConfig.KeyPrefix != actual.GCSConfig.KeyPrefix &&
- expected.GCSConfig.KeyPrefix+"/" != actual.GCSConfig.KeyPrefix {
- return errors.New("GCS key prefix mismatch")
- }
- if expected.GCSConfig.AutomaticCredentials != actual.GCSConfig.AutomaticCredentials {
- return errors.New("GCS automatic credentials mismatch")
- }
- return nil
- }
- func compareHTTPFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.HTTPConfig.Endpoint != actual.HTTPConfig.Endpoint {
- return errors.New("HTTPFs endpoint mismatch")
- }
- if expected.HTTPConfig.Username != actual.HTTPConfig.Username {
- return errors.New("HTTPFs username mismatch")
- }
- if expected.HTTPConfig.SkipTLSVerify != actual.HTTPConfig.SkipTLSVerify {
- return errors.New("HTTPFs skip_tls_verify mismatch")
- }
- if expected.SFTPConfig.EqualityCheckMode != actual.SFTPConfig.EqualityCheckMode {
- return errors.New("HTTPFs equality_check_mode mismatch")
- }
- if err := checkEncryptedSecret(expected.HTTPConfig.Password, actual.HTTPConfig.Password); err != nil {
- return fmt.Errorf("HTTPFs password mismatch: %v", err)
- }
- if err := checkEncryptedSecret(expected.HTTPConfig.APIKey, actual.HTTPConfig.APIKey); err != nil {
- return fmt.Errorf("HTTPFs API key mismatch: %v", err)
- }
- return nil
- }
- func compareSFTPFsConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.SFTPConfig.Endpoint != actual.SFTPConfig.Endpoint {
- return errors.New("SFTPFs endpoint mismatch")
- }
- if expected.SFTPConfig.Username != actual.SFTPConfig.Username {
- return errors.New("SFTPFs username mismatch")
- }
- if expected.SFTPConfig.DisableCouncurrentReads != actual.SFTPConfig.DisableCouncurrentReads {
- return errors.New("SFTPFs disable_concurrent_reads mismatch")
- }
- if expected.SFTPConfig.BufferSize != actual.SFTPConfig.BufferSize {
- return errors.New("SFTPFs buffer_size mismatch")
- }
- if expected.SFTPConfig.EqualityCheckMode != actual.SFTPConfig.EqualityCheckMode {
- return errors.New("SFTPFs equality_check_mode mismatch")
- }
- if err := checkEncryptedSecret(expected.SFTPConfig.Password, actual.SFTPConfig.Password); err != nil {
- return fmt.Errorf("SFTPFs password mismatch: %v", err)
- }
- if err := checkEncryptedSecret(expected.SFTPConfig.PrivateKey, actual.SFTPConfig.PrivateKey); err != nil {
- return fmt.Errorf("SFTPFs private key mismatch: %v", err)
- }
- if err := checkEncryptedSecret(expected.SFTPConfig.KeyPassphrase, actual.SFTPConfig.KeyPassphrase); err != nil {
- return fmt.Errorf("SFTPFs private key passphrase mismatch: %v", err)
- }
- if expected.SFTPConfig.Prefix != actual.SFTPConfig.Prefix {
- if expected.SFTPConfig.Prefix != "" && actual.SFTPConfig.Prefix != "/" {
- return errors.New("SFTPFs prefix mismatch")
- }
- }
- if len(expected.SFTPConfig.Fingerprints) != len(actual.SFTPConfig.Fingerprints) {
- return errors.New("SFTPFs fingerprints mismatch")
- }
- for _, value := range actual.SFTPConfig.Fingerprints {
- if !util.Contains(expected.SFTPConfig.Fingerprints, value) {
- return errors.New("SFTPFs fingerprints mismatch")
- }
- }
- return nil
- }
- func compareAzBlobConfig(expected *vfs.Filesystem, actual *vfs.Filesystem) error {
- if expected.AzBlobConfig.Container != actual.AzBlobConfig.Container {
- return errors.New("azure Blob container mismatch")
- }
- if expected.AzBlobConfig.AccountName != actual.AzBlobConfig.AccountName {
- return errors.New("azure Blob account name mismatch")
- }
- if err := checkEncryptedSecret(expected.AzBlobConfig.AccountKey, actual.AzBlobConfig.AccountKey); err != nil {
- return fmt.Errorf("azure Blob account key mismatch: %v", err)
- }
- if expected.AzBlobConfig.Endpoint != actual.AzBlobConfig.Endpoint {
- return errors.New("azure Blob endpoint mismatch")
- }
- if err := checkEncryptedSecret(expected.AzBlobConfig.SASURL, actual.AzBlobConfig.SASURL); err != nil {
- return fmt.Errorf("azure Blob SAS URL mismatch: %v", err)
- }
- if expected.AzBlobConfig.UploadPartSize != actual.AzBlobConfig.UploadPartSize {
- return errors.New("azure Blob upload part size mismatch")
- }
- if expected.AzBlobConfig.UploadConcurrency != actual.AzBlobConfig.UploadConcurrency {
- return errors.New("azure Blob upload concurrency mismatch")
- }
- if expected.AzBlobConfig.DownloadPartSize != actual.AzBlobConfig.DownloadPartSize {
- return errors.New("azure Blob download part size mismatch")
- }
- if expected.AzBlobConfig.DownloadConcurrency != actual.AzBlobConfig.DownloadConcurrency {
- return errors.New("azure Blob download concurrency mismatch")
- }
- if expected.AzBlobConfig.KeyPrefix != actual.AzBlobConfig.KeyPrefix &&
- expected.AzBlobConfig.KeyPrefix+"/" != actual.AzBlobConfig.KeyPrefix {
- return errors.New("azure Blob key prefix mismatch")
- }
- if expected.AzBlobConfig.UseEmulator != actual.AzBlobConfig.UseEmulator {
- return errors.New("azure Blob use emulator mismatch")
- }
- if expected.AzBlobConfig.AccessTier != actual.AzBlobConfig.AccessTier {
- return errors.New("azure Blob access tier mismatch")
- }
- return nil
- }
- func areSecretEquals(expected, actual *kms.Secret) bool {
- if expected == nil && actual == nil {
- return true
- }
- if expected != nil && expected.IsEmpty() && actual == nil {
- return true
- }
- if actual != nil && actual.IsEmpty() && expected == nil {
- return true
- }
- return false
- }
- func checkEncryptedSecret(expected, actual *kms.Secret) error {
- if areSecretEquals(expected, actual) {
- return nil
- }
- if expected == nil && actual != nil && !actual.IsEmpty() {
- return errors.New("secret mismatch")
- }
- if actual == nil && expected != nil && !expected.IsEmpty() {
- return errors.New("secret mismatch")
- }
- if expected.IsPlain() && actual.IsEncrypted() {
- if actual.GetPayload() == "" {
- return errors.New("invalid secret payload")
- }
- if actual.GetAdditionalData() != "" {
- return errors.New("invalid secret additional data")
- }
- if actual.GetKey() != "" {
- return errors.New("invalid secret key")
- }
- } else {
- if expected.GetStatus() != actual.GetStatus() || expected.GetPayload() != actual.GetPayload() {
- return errors.New("secret mismatch")
- }
- }
- return nil
- }
- func compareUserFilterSubStructs(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- for _, IPMask := range expected.AllowedIP {
- if !util.Contains(actual.AllowedIP, IPMask) {
- return errors.New("allowed IP contents mismatch")
- }
- }
- for _, IPMask := range expected.DeniedIP {
- if !util.Contains(actual.DeniedIP, IPMask) {
- return errors.New("denied IP contents mismatch")
- }
- }
- for _, method := range expected.DeniedLoginMethods {
- if !util.Contains(actual.DeniedLoginMethods, method) {
- return errors.New("denied login methods contents mismatch")
- }
- }
- for _, protocol := range expected.DeniedProtocols {
- if !util.Contains(actual.DeniedProtocols, protocol) {
- return errors.New("denied protocols contents mismatch")
- }
- }
- for _, options := range expected.WebClient {
- if !util.Contains(actual.WebClient, options) {
- return errors.New("web client options contents mismatch")
- }
- }
- return compareUserFiltersEqualFields(expected, actual)
- }
- func compareUserFiltersEqualFields(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- if expected.Hooks.ExternalAuthDisabled != actual.Hooks.ExternalAuthDisabled {
- return errors.New("external_auth_disabled hook mismatch")
- }
- if expected.Hooks.PreLoginDisabled != actual.Hooks.PreLoginDisabled {
- return errors.New("pre_login_disabled hook mismatch")
- }
- if expected.Hooks.CheckPasswordDisabled != actual.Hooks.CheckPasswordDisabled {
- return errors.New("check_password_disabled hook mismatch")
- }
- if expected.DisableFsChecks != actual.DisableFsChecks {
- return errors.New("disable_fs_checks mismatch")
- }
- if expected.StartDirectory != actual.StartDirectory {
- return errors.New("start_directory mismatch")
- }
- return nil
- }
- func compareBaseUserFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- if len(expected.AllowedIP) != len(actual.AllowedIP) {
- return errors.New("allowed IP mismatch")
- }
- if len(expected.DeniedIP) != len(actual.DeniedIP) {
- return errors.New("denied IP mismatch")
- }
- if len(expected.DeniedLoginMethods) != len(actual.DeniedLoginMethods) {
- return errors.New("denied login methods mismatch")
- }
- if len(expected.DeniedProtocols) != len(actual.DeniedProtocols) {
- return errors.New("denied protocols mismatch")
- }
- if expected.MaxUploadFileSize != actual.MaxUploadFileSize {
- return errors.New("max upload file size mismatch")
- }
- if expected.TLSUsername != actual.TLSUsername {
- return errors.New("TLSUsername mismatch")
- }
- if len(expected.WebClient) != len(actual.WebClient) {
- return errors.New("WebClient filter mismatch")
- }
- if expected.AllowAPIKeyAuth != actual.AllowAPIKeyAuth {
- return errors.New("allow_api_key_auth mismatch")
- }
- if expected.ExternalAuthCacheTime != actual.ExternalAuthCacheTime {
- return errors.New("external_auth_cache_time mismatch")
- }
- if expected.FTPSecurity != actual.FTPSecurity {
- return errors.New("ftp_security mismatch")
- }
- if expected.IsAnonymous != actual.IsAnonymous {
- return errors.New("is_anonymous mismatch")
- }
- if expected.DefaultSharesExpiration != actual.DefaultSharesExpiration {
- return errors.New("default_shares_expiration mismatch")
- }
- return nil
- }
- func compareUserFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- if err := compareBaseUserFilters(expected, actual); err != nil {
- return err
- }
- if err := compareUserFilterSubStructs(expected, actual); err != nil {
- return err
- }
- if err := compareUserBandwidthLimitFilters(expected, actual); err != nil {
- return err
- }
- if err := compareUserDataTransferLimitFilters(expected, actual); err != nil {
- return err
- }
- return compareUserFilePatternsFilters(expected, actual)
- }
- func checkFilterMatch(expected []string, actual []string) bool {
- if len(expected) != len(actual) {
- return false
- }
- for _, e := range expected {
- if !util.Contains(actual, strings.ToLower(e)) {
- return false
- }
- }
- return true
- }
- func compareUserDataTransferLimitFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- if len(expected.DataTransferLimits) != len(actual.DataTransferLimits) {
- return errors.New("data transfer limits filters mismatch")
- }
- for idx, l := range expected.DataTransferLimits {
- if actual.DataTransferLimits[idx].UploadDataTransfer != l.UploadDataTransfer {
- return errors.New("data transfer limit upload_data_transfer mismatch")
- }
- if actual.DataTransferLimits[idx].DownloadDataTransfer != l.DownloadDataTransfer {
- return errors.New("data transfer limit download_data_transfer mismatch")
- }
- if actual.DataTransferLimits[idx].TotalDataTransfer != l.TotalDataTransfer {
- return errors.New("data transfer limit total_data_transfer mismatch")
- }
- for _, source := range actual.DataTransferLimits[idx].Sources {
- if !util.Contains(l.Sources, source) {
- return errors.New("data transfer limit source mismatch")
- }
- }
- }
- return nil
- }
- func compareUserBandwidthLimitFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- if len(expected.BandwidthLimits) != len(actual.BandwidthLimits) {
- return errors.New("bandwidth limits filters mismatch")
- }
- for idx, l := range expected.BandwidthLimits {
- if actual.BandwidthLimits[idx].UploadBandwidth != l.UploadBandwidth {
- return errors.New("bandwidth filters upload_bandwidth mismatch")
- }
- if actual.BandwidthLimits[idx].DownloadBandwidth != l.DownloadBandwidth {
- return errors.New("bandwidth filters download_bandwidth mismatch")
- }
- if len(actual.BandwidthLimits[idx].Sources) != len(l.Sources) {
- return errors.New("bandwidth filters sources mismatch")
- }
- for _, source := range actual.BandwidthLimits[idx].Sources {
- if !util.Contains(l.Sources, source) {
- return errors.New("bandwidth filters source mismatch")
- }
- }
- }
- return nil
- }
- func compareUserFilePatternsFilters(expected sdk.BaseUserFilters, actual sdk.BaseUserFilters) error {
- if len(expected.FilePatterns) != len(actual.FilePatterns) {
- return errors.New("file patterns mismatch")
- }
- for _, f := range expected.FilePatterns {
- found := false
- for _, f1 := range actual.FilePatterns {
- if path.Clean(f.Path) == path.Clean(f1.Path) && f.DenyPolicy == f1.DenyPolicy {
- if !checkFilterMatch(f.AllowedPatterns, f1.AllowedPatterns) ||
- !checkFilterMatch(f.DeniedPatterns, f1.DeniedPatterns) {
- return errors.New("file patterns contents mismatch")
- }
- found = true
- }
- }
- if !found {
- return errors.New("file patterns contents mismatch")
- }
- }
- return nil
- }
- func compareKeyValues(expected, actual []dataprovider.KeyValue) error {
- if len(expected) != len(actual) {
- return errors.New("kay values mismatch")
- }
- for _, ex := range expected {
- found := false
- for _, ac := range actual {
- if ac.Key == ex.Key && ac.Value == ex.Value {
- found = true
- break
- }
- }
- if !found {
- return errors.New("kay values mismatch")
- }
- }
- return nil
- }
- func compareHTTPparts(expected, actual []dataprovider.HTTPPart) error {
- for _, p1 := range expected {
- found := false
- for _, p2 := range actual {
- if p1.Name == p2.Name {
- found = true
- if err := compareKeyValues(p1.Headers, p2.Headers); err != nil {
- return fmt.Errorf("http headers mismatch for part %q", p1.Name)
- }
- if p1.Body != p2.Body || p1.Filepath != p2.Filepath {
- return fmt.Errorf("http part %q mismatch", p1.Name)
- }
- }
- }
- if !found {
- return fmt.Errorf("expected http part %q not found", p1.Name)
- }
- }
- return nil
- }
- func compareEventActionHTTPConfigFields(expected, actual dataprovider.EventActionHTTPConfig) error {
- if expected.Endpoint != actual.Endpoint {
- return errors.New("http endpoint mismatch")
- }
- if expected.Username != actual.Username {
- return errors.New("http username mismatch")
- }
- if err := checkEncryptedSecret(expected.Password, actual.Password); err != nil {
- return err
- }
- if err := compareKeyValues(expected.Headers, actual.Headers); err != nil {
- return errors.New("http headers mismatch")
- }
- if expected.Timeout != actual.Timeout {
- return errors.New("http timeout mismatch")
- }
- if expected.SkipTLSVerify != actual.SkipTLSVerify {
- return errors.New("http skip TLS verify mismatch")
- }
- if expected.Method != actual.Method {
- return errors.New("http method mismatch")
- }
- if err := compareKeyValues(expected.QueryParameters, actual.QueryParameters); err != nil {
- return errors.New("http query parameters mismatch")
- }
- if expected.Body != actual.Body {
- return errors.New("http body mismatch")
- }
- if len(expected.Parts) != len(actual.Parts) {
- return errors.New("http parts mismatch")
- }
- return compareHTTPparts(expected.Parts, actual.Parts)
- }
- func compareEventActionEmailConfigFields(expected, actual dataprovider.EventActionEmailConfig) error {
- if len(expected.Recipients) != len(actual.Recipients) {
- return errors.New("email recipients mismatch")
- }
- for _, v := range expected.Recipients {
- if !util.Contains(actual.Recipients, v) {
- return errors.New("email recipients content mismatch")
- }
- }
- if expected.Subject != actual.Subject {
- return errors.New("email subject mismatch")
- }
- if expected.Body != actual.Body {
- return errors.New("email body mismatch")
- }
- if len(expected.Attachments) != len(actual.Attachments) {
- return errors.New("email attachments mismatch")
- }
- for _, v := range expected.Attachments {
- if !util.Contains(actual.Attachments, v) {
- return errors.New("email attachments content mismatch")
- }
- }
- return nil
- }
- func compareEventActionFsCompressFields(expected, actual dataprovider.EventActionFsCompress) error {
- if expected.Name != actual.Name {
- return errors.New("fs compress name mismatch")
- }
- if len(expected.Paths) != len(actual.Paths) {
- return errors.New("fs compress paths mismatch")
- }
- for _, v := range expected.Paths {
- if !util.Contains(actual.Paths, v) {
- return errors.New("fs compress paths content mismatch")
- }
- }
- return nil
- }
- func compareEventActionFsConfigFields(expected, actual dataprovider.EventActionFilesystemConfig) error {
- if expected.Type != actual.Type {
- return errors.New("fs type mismatch")
- }
- if err := compareKeyValues(expected.Renames, actual.Renames); err != nil {
- return errors.New("fs renames mismatch")
- }
- if len(expected.Deletes) != len(actual.Deletes) {
- return errors.New("fs deletes mismatch")
- }
- for _, v := range expected.Deletes {
- if !util.Contains(actual.Deletes, v) {
- return errors.New("fs deletes content mismatch")
- }
- }
- if len(expected.MkDirs) != len(actual.MkDirs) {
- return errors.New("fs mkdirs mismatch")
- }
- for _, v := range expected.MkDirs {
- if !util.Contains(actual.MkDirs, v) {
- return errors.New("fs mkdir content mismatch")
- }
- }
- if len(expected.Exist) != len(actual.Exist) {
- return errors.New("fs exist mismatch")
- }
- for _, v := range expected.Exist {
- if !util.Contains(actual.Exist, v) {
- return errors.New("fs exist content mismatch")
- }
- }
- return compareEventActionFsCompressFields(expected.Compress, actual.Compress)
- }
- func compareEventActionCmdConfigFields(expected, actual dataprovider.EventActionCommandConfig) error {
- if expected.Cmd != actual.Cmd {
- return errors.New("command mismatch")
- }
- if expected.Timeout != actual.Timeout {
- return errors.New("cmd timeout mismatch")
- }
- if len(expected.Args) != len(actual.Args) {
- return errors.New("cmd args mismatch")
- }
- for _, v := range expected.Args {
- if !util.Contains(actual.Args, v) {
- return errors.New("cmd args content mismatch")
- }
- }
- if err := compareKeyValues(expected.EnvVars, actual.EnvVars); err != nil {
- return errors.New("cmd env vars mismatch")
- }
- return nil
- }
- func compareEventActionDataRetentionFields(expected, actual dataprovider.EventActionDataRetentionConfig) error {
- if len(expected.Folders) != len(actual.Folders) {
- return errors.New("retention folders mismatch")
- }
- for _, f1 := range expected.Folders {
- found := false
- for _, f2 := range actual.Folders {
- if f1.Path == f2.Path {
- found = true
- if f1.Retention != f2.Retention {
- return fmt.Errorf("retention mismatch for folder %s", f1.Path)
- }
- if f1.DeleteEmptyDirs != f2.DeleteEmptyDirs {
- return fmt.Errorf("delete_empty_dirs mismatch for folder %s", f1.Path)
- }
- if f1.IgnoreUserPermissions != f2.IgnoreUserPermissions {
- return fmt.Errorf("ignore_user_permissions mismatch for folder %s", f1.Path)
- }
- break
- }
- }
- if !found {
- return errors.New("retention folders mismatch")
- }
- }
- return nil
- }
- func compareEqualGroupSettingsFields(expected sdk.BaseGroupUserSettings, actual sdk.BaseGroupUserSettings) error {
- if expected.HomeDir != actual.HomeDir {
- return errors.New("home dir mismatch")
- }
- if expected.MaxSessions != actual.MaxSessions {
- return errors.New("MaxSessions mismatch")
- }
- if expected.QuotaSize != actual.QuotaSize {
- return errors.New("QuotaSize mismatch")
- }
- if expected.QuotaFiles != actual.QuotaFiles {
- return errors.New("QuotaFiles mismatch")
- }
- if expected.UploadBandwidth != actual.UploadBandwidth {
- return errors.New("UploadBandwidth mismatch")
- }
- if expected.DownloadBandwidth != actual.DownloadBandwidth {
- return errors.New("DownloadBandwidth mismatch")
- }
- if expected.UploadDataTransfer != actual.UploadDataTransfer {
- return errors.New("upload_data_transfer mismatch")
- }
- if expected.DownloadDataTransfer != actual.DownloadDataTransfer {
- return errors.New("download_data_transfer mismatch")
- }
- if expected.TotalDataTransfer != actual.TotalDataTransfer {
- return errors.New("total_data_transfer mismatch")
- }
- return compareUserPermissions(expected.Permissions, actual.Permissions)
- }
- func compareEqualsUserFields(expected *dataprovider.User, actual *dataprovider.User) error {
- if dataprovider.ConvertName(expected.Username) != actual.Username {
- return errors.New("username mismatch")
- }
- if expected.HomeDir != actual.HomeDir {
- return errors.New("home dir mismatch")
- }
- if expected.UID != actual.UID {
- return errors.New("UID mismatch")
- }
- if expected.GID != actual.GID {
- return errors.New("GID mismatch")
- }
- if expected.MaxSessions != actual.MaxSessions {
- return errors.New("MaxSessions mismatch")
- }
- if len(expected.Permissions) != len(actual.Permissions) {
- return errors.New("permissions mismatch")
- }
- if expected.UploadBandwidth != actual.UploadBandwidth {
- return errors.New("UploadBandwidth mismatch")
- }
- if expected.DownloadBandwidth != actual.DownloadBandwidth {
- return errors.New("DownloadBandwidth mismatch")
- }
- if expected.Status != actual.Status {
- return errors.New("status mismatch")
- }
- if expected.ExpirationDate != actual.ExpirationDate {
- return errors.New("ExpirationDate mismatch")
- }
- if expected.AdditionalInfo != actual.AdditionalInfo {
- return errors.New("AdditionalInfo mismatch")
- }
- if expected.Description != actual.Description {
- return errors.New("description mismatch")
- }
- return compareQuotaUserFields(expected, actual)
- }
- func compareQuotaUserFields(expected *dataprovider.User, actual *dataprovider.User) error {
- if expected.QuotaSize != actual.QuotaSize {
- return errors.New("QuotaSize mismatch")
- }
- if expected.QuotaFiles != actual.QuotaFiles {
- return errors.New("QuotaFiles mismatch")
- }
- if expected.UploadDataTransfer != actual.UploadDataTransfer {
- return errors.New("upload_data_transfer mismatch")
- }
- if expected.DownloadDataTransfer != actual.DownloadDataTransfer {
- return errors.New("download_data_transfer mismatch")
- }
- if expected.TotalDataTransfer != actual.TotalDataTransfer {
- return errors.New("total_data_transfer mismatch")
- }
- return nil
- }
- func addLimitAndOffsetQueryParams(rawurl string, limit, offset int64) (*url.URL, error) {
- url, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
- q := url.Query()
- if limit > 0 {
- q.Add("limit", strconv.FormatInt(limit, 10))
- }
- if offset > 0 {
- q.Add("offset", strconv.FormatInt(offset, 10))
- }
- url.RawQuery = q.Encode()
- return url, err
- }
- func addModeQueryParam(rawurl, mode string) (*url.URL, error) {
- url, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
- q := url.Query()
- if len(mode) > 0 {
- q.Add("mode", mode)
- }
- url.RawQuery = q.Encode()
- return url, err
- }
- func addDisconnectQueryParam(rawurl, disconnect string) (*url.URL, error) {
- url, err := url.Parse(rawurl)
- if err != nil {
- return nil, err
- }
- q := url.Query()
- if len(disconnect) > 0 {
- q.Add("disconnect", disconnect)
- }
- url.RawQuery = q.Encode()
- return url, err
- }
|