pymemstore/test.py at master · optimuspaul/pymemstore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
if __name__ == "__main__":
	"""
	Simple tests.  TODO - write up real unit tests and make this thing sing
	"""
	import resource
	import msgpack
	from pymemstore import MemStoreClient, MemStore
	store = MemStore("main")
	store.start()

	client = MemStoreClient("main")

	print client.info()
	print client.restore("test-data.msgpack")
	print client.info()

	for x in range(20):
		client.put("table_one", "{0:06d}".format(x), "test value {0}")

	print client.info()

	for x in range(10):
		client.put("table_two", "{0:04d}".format(x), "best value {0}".format(x))

	print "store", client.info()
	print "client", resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024.0 / 1024.0, "mb"
	print client.store("test-data.msgpack")
	print client.get("table_two", "0045")
	full_table = client.all("table_two")
	print "{0} records found".format(len(full_table.keys()))
	print "store", client.info()
	print "client", resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024.0 / 1024.0, "mb"

	print "=" * 80
	print "push tests"
	print "=" * 80
	print "push", client.push("test-set", "0045"), "0045"
	print "push", client.push("test-set", "1045"), "1045"
	print "pull", client.pull("test-set", "1045"), "1045"
	print "in_set", client.in_set("test-set", "0045"), "0045"
	print "in_set", client.in_set("test-set", "1045"), "1045"

	print client.store("test-data.msgpack")

	client.send_message(msgpack.dumps(["done"]))
	client.stop()
	print "client", resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024.0 / 1024.0, "mb"