# -*- coding: utf-8 -*-

import sys

import os

import redis

import json

from RLTest import Env

from common import *

from includes import *

from RLTest import Defaults

from functools import reduce

Defaults.decode_responses = True

# ----------------------------------------------------------------------------------------------

# Path to JSON test case files

HERE = os.path.abspath(os.path.dirname(__file__))

ROOT = os.path.abspath(os.path.join(HERE, "../.."))

TESTS_ROOT = os.path.abspath(os.path.join(HERE, ".."))

JSON_PATH = os.path.join(TESTS_ROOT, 'files')

nested_large_key = r'{"jkra":[154,4472,[8567,false,363.84,5276,"ha","rizkzs",93],false],"hh":20.77,"mr":973.217,"ihbe":[68,[true,{"lqe":[486.363,[true,{"mp":{"ory":"rj","qnl":"tyfrju","hf":null},"uooc":7418,"xela":20,"bt":7014,"ia":547,"szec":68.73},null],3622,"iwk",null],"fepi":19.954,"ivu":{"rmnd":65.539,"bk":98,"nc":"bdg","dlb":{"hw":{"upzz":[true,{"nwb":[4259.47],"nbt":"yl"},false,false,65,[[[],629.149,"lvynqh","hsk",[],2011.932,true,[]],null,"ymbc",null],"aj",97.425,"hc",58]},"jq":true,"bi":3333,"hmf":"pl","mrbj":[true,false]}},"hfj":"lwk","utdl":"aku","alqb":[74,534.389,7235,[null,false,null]]},null,{"lbrx":{"vm":"ubdrbb"},"tie":"iok","br":"ojro"},70.558,[{"mmo":null,"dryu":null}]],true,null,false,{"jqun":98,"ivhq":[[[675.936,[520.15,1587.4,false],"jt",true,{"bn":null,"ygn":"cve","zhh":true,"aak":9165,"skx":true,"qqsk":662.28},{"eio":9933.6,"agl":null,"pf":false,"kv":5099.631,"no":null,"shly":58},[null,["uiundu",726.652,false,94.92,259.62,{"ntqu":null,"frv":null,"rvop":"upefj","jvdp":{"nhx":[],"bxnu":{},"gs":null,"mqho":null,"xp":65,"ujj":{}},"ts":false,"kyuk":[false,58,{},"khqqif"]},167,true,"bhlej",53],64,{"eans":"wgzfo","zfgb":431.67,"udy":[{"gnt":[],"zeve":{}},{"pg":{},"vsuc":{},"dw":19,"ffo":"uwsh","spk":"pjdyam","mc":[],"wunb":{},"qcze":2271.15,"mcqx":null},"qob"],"wo":"zy"},{"dok":null,"ygk":null,"afdw":[7848,"ah",null],"foobar":3.141592,"wnuo":{"zpvi":{"stw":true,"bq":{},"zord":true,"omne":3061.73,"bnwm":"wuuyy","tuv":7053,"lepv":null,"xap":94.26},"nuv":false,"hhza":539.615,"rqw":{"dk":2305,"wibo":7512.9,"ytbc":153,"pokp":null,"whzd":null,"judg":[],"zh":null},"bcnu":"ji","yhqu":null,"gwc":true,"smp":{"fxpl":75,"gc":[],"vx":9352.895,"fbzf":4138.27,"tiaq":354.306,"kmfb":{},"fxhy":[],"af":94.46,"wg":{},"fb":null}},"zvym":2921,"hhlh":[45,214.345],"vv":"gqjoz"},["uxlu",null,"utl",64,[2695],[false,null,["cfcrl",[],[],562,1654.9,{},null,"sqzud",934.6],{"hk":true,"ed":"lodube","ye":"ziwddj","ps":null,"ir":{},"heh":false},true,719,50.56,[99,6409,null,4886,"esdtkt",{},null],[false,"bkzqw"]],null,6357],{"asvv":22.873,"vqm":{"drmv":68.12,"tmf":140.495,"le":null,"sanf":[true,[],"vyawd",false,76.496,[],"sdfpr",33.16,"nrxy","antje"],"yrkh":662.426,"vxj":true,"sn":314.382,"eorg":null},"bavq":[21.18,8742.66,{"eq":"urnd"},56.63,"fw",[{},"pjtr",null,"apyemk",[],[],false,{}],{"ho":null,"ir":124,"oevp":159,"xdrv":6705,"ff":[],"sx":false},true,null,true],"zw":"qjqaap","hr":{"xz":32,"mj":8235.32,"yrtv":null,"jcz":"vnemxe","ywai":[null,564,false,"vbr",54.741],"vw":82,"wn":true,"pav":true},"vxa":881},"bgt","vuzk",857]]],null,null,{"xyzl":"nvfff"},true,13],"npd":null,"ha":[["du",[980,{"zdhd":[129.986,["liehns",453,{"fuq":false,"dxpn":{},"hmpx":49,"zb":"gbpt","vdqc":null,"ysjg":false,"gug":7990.66},"evek",[{}],"dfywcu",9686,null]],"gpi":{"gt":{"qe":7460,"nh":"nrn","czj":66.609,"jwd":true,"rb":"azwwe","fj":{"csn":true,"foobar":1.61803398875,"hm":"efsgw","zn":"vbpizt","tjo":138.15,"teo":{},"hecf":[],"ls":false}},"xlc":7916,"jqst":48.166,"zj":"ivctu"},"jl":369.27,"mxkx":null,"sh":[true,373,false,"sdis",6217,{"ernm":null,"srbo":90.798,"py":677,"jgrq":null,"zujl":null,"odsm":{"pfrd":null,"kwz":"kfvjzb","ptkp":false,"pu":null,"xty":null,"ntx":[],"nq":48.19,"lpyx":[]},"ff":null,"rvi":["ych",{},72,9379,7897.383,true,{},999.751,false]},true],"ghe":[24,{"lpr":true,"qrs":true},true,false,7951.94,true,2690.54,[93,null,null,"rlz",true,"ky",true]],"vet":false,"olle":null},"jzm",true],null,null,19.17,7145,"ipsmk"],false,{"du":6550.959,"sps":8783.62,"nblr":{"dko":9856.616,"lz":{"phng":"dj"},"zeu":766,"tn":"dkr"},"xa":"trdw","gn":9875.687,"dl":null,"vuql":null},{"qpjo":null,"das":{"or":{"xfy":null,"xwvs":4181.86,"yj":206.325,"bsr":["qrtsh"],"wndm":{"ve":56,"jyqa":true,"ca":null},"rpd":9906,"ea":"dvzcyt"},"xwnn":9272,"rpx":"zpr","srzg":{"beo":325.6,"sq":null,"yf":null,"nu":[377,"qda",true],"sfz":"zjk"},"kh":"xnpj","rk":null,"hzhn":[null],"uio":6249.12,"nxrv":1931.635,"pd":null},"pxlc":true,"mjer":false,"hdev":"msr","er":null},"ug",null,"yrfoix",503.89,563],"tcy":300,"me":459.17,"tm":[134.761,"jcoels",null],"iig":945.57,"ad":"be"},"ltpdm",null,14.53],"xi":"gxzzs","zfpw":1564.87,"ow":null,"tm":[46,876.85],"xejv":null}'

# FIXME: Test all multi-path options (dot notation and bracket notation):

# Recursive descent, e.g., $..leaf_val

# Wildcard (in key and in index), e.g., $.*[*]

# Array slice [start:end:step], e.g., $.arr[2:4]

# Union, e.g., $.arr[1,2,4] and $.[field1, field5]

# Boolean filter, e.g., $.arr[?(@.field>3 && @.id==null)]

def testDelCommand(env):

r.assertEqual(json.loads(res), [[True, {"answer": 42}]])

def testDelCommand_issue529(env):

r.assertEqual(res, [3, 3, 3])

def testDelCommand_issue754(env):

r.assertEqual(json.loads(res), [{"a":[[2,3,[5,[7,8]]],[{"11":11}],["13"]],"b":[{"a":[4,5]}]}])

def testForgetCommand(env):

r.assertEqual(res, 0)

def testSetAndGetCommands(env):

r.assertEqual(res, '"inizio"')

def testMGetCommand(env):

r.assertEqual(res, [None, None, None])

def testNumByCommands(env):

r.assertEqual(res, '16')

def testStrAppendCommand(env):

r.assertEqual(res, 7)

def testStrLenCommand(env):

r.assertEqual(res, 7)

def testArrAppendCommand(env):

r.expect('JSON.ARRAPPEND', 'non_existing_doc', '..a').raiseError()

def testArrInsertCommand(env):

r.expect('JSON.ARRINSERT', 'non_existing_doc', '..a').raiseError()

def testArrLenCommand(env):

r.assertEqual(res, 5)

def testArrPopCommand(env):

r.assertEqual(res, '0')

def testArrTrimCommand(env):

r.expect('JSON.ARRTRIM', 'non_existing_doc', '..a').raiseError()

def testObjKeysCommand(env):

r.assertEqual(res, ["nested1", "a", "nested2"])

def testObjLenCommand(env):

r.assertEqual(res, 3)

def load_types_data(nested_key_name):

return jdata, types

def testTypeCommand(env):

r.assertEqual(res, None)

def testClearCommand(env):

r.expect('JSON.CLEAR', 'non_existing_doc', '$..a').raiseError()

def testToggleCommand(env):

r.expect('JSON.TOGGLE', 'non_existing_doc', '$..a').raiseError()

# TODO adapt test to run on Mac

# @no_san

# def testMemoryUsage(env):

# """

# Test MEMORY USAGE key

# """

# if env.moduleArgs is not None and ['JSON_BACKEND SERDE_JSON'] in env.moduleArgs:

# env.skip()

# env

# r = env

# jdata, jtypes = load_types_data('a')

# r.assertOk(r.execute_command('JSON.SET', 'doc1', '$', json.dumps(jdata)))

# res = r.execute_command('MEMORY', 'USAGE', 'doc1')

# r.assertEqual(res, 211)

# jdata, jtypes = load_types_data('verylongfieldname')

# r.assertOk(r.execute_command('JSON.SET', 'doc2', '$', json.dumps(jdata)))

# res = r.execute_command('MEMORY', 'USAGE', 'doc2')

# r.assertEqual(res, 323)

@no_san

def testDebugCommand(env):

r.expect('JSON.DEBUG', 'non_existing_doc', '$..a').raiseError()

def testRespCommand(env):

r.assertEqual(res, ['[', ['[', 1], ['[', 2]])

def testArrIndexCommand(env):