1 files changed, 149 insertions, 0 deletions
diff --git a/demoscript/Freeze/library.py b/demoscript/Freeze/library.py
new file mode 100755
index 00000000000..9212c18138f
--- /dev/null
+++ b/demoscript/Freeze/library.py
@@ -0,0 +1,149 @@
+#!/usr/bin/env python
+# **********************************************************************
+#
+# Copyright (c) 2003-2007 ZeroC, Inc. All rights reserved.
+#
+# This copy of Ice is licensed to you under the terms described in the
+# ICE_LICENSE file included in this distribution.
+#
+# **********************************************************************
+
+import pexpect, sys
+
+def dequote(s):
+    cur = 0
+    end = len(s)
+    havequote = False
+    quote = None
+    items = []
+    while cur < len(s):
+        if not quote:
+            if s[cur] == "'" or s[cur] == '"':
+                quote = s[cur]
+                start = cur+1
+        else:
+            if s[cur] == quote:
+                items.append(s[start:cur])
+                quote = None
+        cur = cur + 1
+    return items
+
+def mkregexp(s):
+    s = s.replace('(', '\\(')
+    s = s.replace(')', '\\)')
+    s = s.replace('.', '\\.')
+    s = s.replace('+', '\\+')
+    s = s.replace('*', '\\*')
+    return s
+                
+def run(client, server):
+    print "populating database... ",
+    sys.stdout.flush()
+    f = open("books", "r")
+    books = []
+    for l in f:
+        client.sendline(l)
+        #server.expect('added object')
+        client.expect('added new book')
+        isbn, title, author = dequote(l)
+        books.append((isbn, title, author))
+    print "ok"
+
+    byauthor = {}
+    for b in books:
+        isbn, title, author = b
+        if not byauthor.has_key(author):
+            byauthor[author] = []
+        byauthor[author].append(b)
+
+    print "testing isbn... ",
+    sys.stdout.flush()
+    for b in books:
+        isbn, title, author = b
+        client.sendline('isbn %s' %(isbn))
+        #server.expect('locate found')
+        #server.expect('locate found')
+        client.expect('current book is:')
+        client.expect('isbn: %s' %(isbn))
+        client.expect('title: %s' %(mkregexp(title)))
+        client.expect('authors: %s' %(author))
+    client.sendline('isbn 1000')
+    client.expect('no book with that')
+    print "ok"
+
+    print "testing authors... ",
+    sys.stdout.flush()
+    for a, bl in byauthor.iteritems():
+        client.sendline('authors "%s"' %(a))
+        client.expect('number of books found: ([0-9]+)')
+        n = int(client.match.group(1))
+        assert len(bl) == n
+        for i in range(0, n):
+            #server.expect('locate found')
+            # Consume evicting messages otherwise its possible to get
+            # hangs.
+            #try:
+                #server.expect('evicting', timeout=0)
+            #except pexpect.TIMEOUT:
+                #pass
+            client.expect('current book is:')
+            client.expect('isbn: ([a-zA-Z0-9]+)')
+            findisbn = client.match.group(1)
+            nbl = []
+            for b in bl:
+                isbn, title, author = b
+                if not findisbn:
+                    nbl.append(b)
+                if isbn == findisbn:
+                    findisbn = None
+                    client.expect('title: %s' %(mkregexp(title)))
+                    client.expect('authors: %s' %(author))
+            bl = nbl
+            assert not findisbn
+            client.sendline('next')
+        client.expect('no current book')
+        assert len(bl) == 0
+    client.sendline('authors foo')
+    client.expect('number of books found: 0')
+    client.expect('no current book')
+    print "ok"
+
+    print "testing rent/return... ",
+    sys.stdout.flush()
+    isbn, title, author = books[0]
+    client.sendline('isbn %s' % (isbn))
+    #server.expect('locate')
+    client.expect('current book is:.*isbn.*\r{1,2}\ntitle.*\r{1,2}\nauthors')
+    client.sendline('rent matthew')
+    #server.expect('locate')
+    client.expect("the book is now rented by `matthew'")
+    client.sendline('current')
+    #server.expect('locate')
+    client.expect('rented: matthew')
+    client.sendline('rent john')
+    #server.expect('locate')
+    client.expect('the book has already been rented')
+    client.sendline('return')
+    #server.expect('locate')
+    client.expect('the book has been returned')
+    client.sendline('current')
+    try:
+        client.expect('rented:', timeout=2)
+    except pexpect.TIMEOUT:
+        pass
+    print "ok"
+
+    print "testing remove... ",
+    sys.stdout.flush()
+    isbn, title, author = books[0]
+    client.sendline('isbn %s' % (isbn))
+    #server.expect('locate')
+    client.expect('current book is:.*isbn.*\r{1,2}\ntitle.*\r{1,2}\nauthors')
+    client.sendline('remove')
+    #server.expect('locate')
+    #server.expect('removed')
+    client.expect('removed current book')
+    client.sendline('remove')
+    #server.expect(['locate could not find', 'locate found.*but it was dead or destroyed'])
+    client.expect('current book no longer exists')
+    print "ok"