jython test scripts

1 files changed, 278 insertions, 0 deletions

diff --git a/tests/scripts/incr.py b/tests/scripts/incr.py
new file mode 100644
index 000000000..9aada6beb
--- /dev/null
+++ b/tests/scripts/incr.py
@@ -0,0 +1,278 @@
+import os, sys

+

+from org.aspectj.util import FileUtil

+from java.io import File

+

+sourcedir = "incr_test_scratch_sources"

+outdir = "incr_test_scratch_classes"

+errorList = []

+VERBOSE = 0

+

+def createEmpty(dir):

+	if os.path.exists(dir):

+		FileUtil.deleteContents(File(dir))

+	else:

+		os.mkdir(dir)

+

+def makeFile(name, contents):

+	fullname = os.path.join(sourcedir, name)

+	dirname = os.path.dirname(fullname)

+	if not os.path.exists(dirname):	

+		os.makedirs(dirname)

+		

+	fp = open(fullname, 'w')

+	fp.write(contents)

+	fp.close()

+

+def deleteFile(name):

+	os.remove(os.path.join(sourcedir, name))

+

+def snapshot(dir, map=None):

+	if map is None: map = {}

+	for file in os.listdir(dir):

+		filename = os.path.join(dir, file)

+		if os.path.isdir(filename):

+			snapshot(filename, map)

+		else:

+			stats = os.stat(filename)

+			map[filename] = stats[8]

+	return map

+

+def diffSnapshots(old, new):

+	unchanged = []

+	changed = []

+	for name, mtime in new.items():

+		if old.has_key(name):

+			oldTime = old[name]

+			if oldTime == mtime:

+				unchanged.append(name)

+			else:

+				changed.append(name)

+			del old[name]

+		else:

+			changed.append(name)

+	

+	deleted = old.keys()

+	

+	return unchanged, changed, deleted

+

+

+

+

+def error(m):

+	errorList.append(m)

+	print m

+

+def suffixInList(suffix, list):

+	for i in list:

+		if i.endswith(suffix): return 1

+	return 0

+

+def checkClasses(kind, filelist, names):

+	filenames = []

+	for o in filelist:

+		name = os.path.basename(o)

+		filenames.append(name[:-6])

+	checkSets(names, filenames, kind)

+

+"""

+	#print names, repr(names)

+	if repr(names).startswith("\'"): names = [names]

+

+	for c in names:

+		classname = c+".class"

+		if not suffixInList(classname, filelist):

+			error("%s expected %s not found in %s" % (name, classname, filelist))

+"""

+

+def findAndRemove(l, item):

+	for i in range(len(l)):

+		if l[i] == item:

+			del l[i]

+			return 1

+	return 0

+	

+def makeList(l):

+	if repr(l).startswith("\'"): return [l]

+	return l

+	

+

+def checkSets(expected, found, kind="error"):

+	expected = makeList(expected)

+	for e in expected:

+		if not findAndRemove(found, e):

+			error("expected %s %s not found in %s" % (kind, e, found))

+	

+	for f in found:

+		error("unexpected %s %s" % (kind, f))

+

+

+

+from org.aspectj.ajdt.ajc import AjdtCommand

+from org.aspectj.bridge import IMessageHandler, IMessage

+

+def makeSet(errors):

+	ret = {}

+	for e in errors:

+		loc = e.getISourceLocation()

+		s = "%s:%i" % (loc.sourceFile.name[:-5], loc.line)

+		ret[s] = s

+	return ret.keys()

+

+

+class Handler (IMessageHandler):

+	def __init__(self):

+		self.errors = []

+		

+	def handleMessage(self, message):

+		if message.kind == IMessage.ERROR:

+			self.errors.append(message)

+		if VERBOSE: print message

+	def isIgnoring(self, kind):

+		return 0

+

+

+createEmpty(sourcedir)

+createEmpty(outdir)

+

+handler = Handler()

+cmd = AjdtCommand()

+

+

+TEMPLATE = """\

+%(package)s

+%(modifiers)s %(kind)s %(classname)s %(parents)s {

+    %(body)s

+    public static void main(String[] args) {

+        %(stmts)s

+    }

+}

+"""

+

+import string, time

+

+def splitClassName(className):

+	dot = className.rfind('.')

+	if dot == -1:

+		return None, className, className +".java"

+	else:

+		packageName = className[:dot]

+		className = className[dot+1:]

+		l = packageName.split('.')

+		l.append(className + ".java")

+		path = apply(os.path.join, l)

+		

+		return packageName, className, path

+

+

+def makeType(className, stmts="""System.out.println("hello");""", body="", kind="class", parents=""):

+	packageName, className, path = splitClassName(className)

+	if packageName is None: packageDecl = ""

+	else: packageDecl = "package %s;" % packageName

+

+	contents = TEMPLATE % {'package':packageDecl, 'modifiers':'public', 

+							'classname':className, 'body':body, 

+							'stmts':stmts, 'kind':kind, 'parents':parents}

+	makeFile(path, contents)

+

+def deleteType(className):

+	packageName, className, path = splitClassName(className)

+	deleteFile(path)

+

+

+def test(batch=0, couldChange=[], changed=[], deleted=[], errors=[]):

+	print ">>>>test changed=%s, couldChange=%s, deleted=%s, errors=%s<<<<" % (changed, couldChange, deleted, errors)

+	

+	start = snapshot(outdir)

+	handler.errors = []

+	

+	time.sleep(0.1)

+	

+	if batch: cmd.runCommand(["-d", outdir, "-sourceroots", sourcedir], handler)

+	else: cmd.repeatCommand(handler)

+

+	checkSets(errors, makeSet(handler.errors))

+	if len(handler.errors) > 0: return

+

+	end = snapshot(outdir)

+	u, c, d = diffSnapshots(start, end)

+	checkClasses("changed", c, makeList(changed) + makeList(couldChange)) 

+	checkClasses("deleted", d, deleted) 

+

+"""

+Stress testing

+"""

+N = 2000

+l = []

+for i in range(N):

+	name = "p1.Hello" + str(i)

+	makeType(name)

+	l.append("Hello" + str(i))

+

+test(batch=1, changed=l)

+

+print "done", errorList

+sys.exit(0)

+

+

+"""

+Simple tests with aspects

+"""

+

+makeType("p1.Hello")

+test(batch=1, changed="Hello")

+

+makeType("p1.A", kind="aspect", body="before(): within(String) { }")

+test(changed=["A"], couldChange=["Hello"])

+

+makeType("p1.A", kind="aspect", body="before(): within(Hello) { }")

+test(changed=["A", "Hello"])

+

+makeType("p1.Target")

+test(changed="Target")

+

+makeType("p1.Hello", stmts="new Target().m();")

+test(errors=["Hello:5"])

+

+makeType("p1.ATypes", kind="aspect", body="int Target.m() { return 10; }")

+test(changed=["Hello", "ATypes", "Target"], couldChange=["A"])

+

+makeType("p1.ATypes", kind="aspect", body="int Target.m(int x) { return x + 10; }")

+test(errors=["Hello:5"])

+

+makeType("p1.Hello", stmts="new Target().m(2);")

+test(changed="Hello")

+

+print "done", errorList

+sys.exit(0)

+

+

+

+"""

+Pure Java tests

+"""

+

+makeType("p1.Hello")

+test(batch=1, changed="Hello")

+

+test()

+

+makeType("p1.Hello", stmts="Target.staticM();")

+test(errors="Hello:5")

+

+makeType("p1.Target", body="static void staticM() {}")

+test(changed=["Hello", "Target"])

+

+makeType("p1.Target", body="""static void staticM() { System.out.println("foo"); }""")

+test(changed=["Target"])

+

+makeType("p1.Target", body="static int staticM() { return 2; }")

+test(changed=["Hello", "Target"])

+

+makeType("p1.Hello", body="static class Inner {}")

+test(changed=["Hello", "Hello$Inner"])

+

+deleteType("p1.Hello")

+test(deleted=["Hello", "Hello$Inner"])

+

+print "done", errorList
\ No newline at end of file