def test_create(self):
c = amf3.Context()
self.assertEqual(c.strings, [])
self.assertEqual(c.classes, {})
self.assertEqual(len(c.strings), 0)
self.assertEqual(len(c.classes), 0)
def test_get_reference(self):
x = amf3.Context()
y = [1, 2, 3]
z = {'spam': 'eggs'}
spam_alias = miniamf.register_class(Spam, 'spam.eggs')
class Foo:
pass
foo_alias = miniamf.register_class(Foo, 'foo.bar')
a = amf3.ClassDefinition(spam_alias)
b = amf3.ClassDefinition(foo_alias)
ref1 = x.addObject(y)
ref2 = x.addObject(z)
x.addString(b'abc')
x.addString(b'def')
x.addClass(a, Spam)
x.addClass(b, Foo)
self.assertEqual(x.getObjectReference(y), ref1)
self.assertEqual(x.getObjectReference(z), ref2)
self.assertEqual(x.getObjectReference({}), -1)
self.assertEqual(x.getStringReference(b'abc'), 0)
self.assertEqual(x.getStringReference(b'def'), 1)
self.assertEqual(x.getStringReference(b'asdfas'), -1)
self.assertEqual(x.getClass(Spam), a)
self.assertEqual(x.getClass(Foo), b)
self.assertEqual(x.getClass(object()), None)
def test_complex_encode_decode_dict(self):
complex = {'element': 'ignore', 'objects': self.build_complex()}
self.encoder.writeElement(complex)
encoded = self.encoder.stream.getvalue()
context = amf3.Context()
decoded = amf3.Decoder(encoded, context).readElement()
self.complex_encode_decode_test(decoded['objects'])
def test_add_class(self):
x = amf3.Context()
alias = miniamf.register_class(Spam, 'spam.eggs')
y = amf3.ClassDefinition(alias)
self.assertEqual(x.addClass(y, Spam), 0)
self.assertEqual(x.classes, {Spam: y})
self.assertEqual(x.class_ref, {0: y})
self.assertEqual(len(x.class_ref), 1)
def test_add_string(self):
x = amf3.Context()
y = b'abc'
self.assertEqual(x.addString(y), 0)
self.assertTrue(y in x.strings)
self.assertEqual(len(x.strings), 1)
self.assertEqual(x.addString(b''), -1)
self.assertRaises(TypeError, x.addString, 132)
def test_clear(self):
x = amf3.Context()
y = [1, 2, 3]
x.addObject(y)
x.addString(b'spameggs')
x.clear()
self.assertEqual(x.strings, [])
self.assertEqual(len(x.strings), 0)
self.assertFalse('spameggs' in x.strings)
def test_get_by_reference(self):
x = amf3.Context()
y = [1, 2, 3]
z = {'spam': 'eggs'}
alias_spam = miniamf.register_class(Spam, 'spam.eggs')
class Foo:
pass
class Bar:
pass
alias_foo = miniamf.register_class(Foo, 'foo.bar')
a = amf3.ClassDefinition(alias_spam)
b = amf3.ClassDefinition(alias_foo)
x.addObject(y)
x.addObject(z)
x.addString(b'abc')
x.addString(b'def')
x.addClass(a, Foo)
x.addClass(b, Bar)
self.assertEqual(x.getObject(0), y)
self.assertEqual(x.getObject(1), z)
self.assertEqual(x.getObject(2), None)
self.assertRaises(TypeError, x.getObject, '')
self.assertRaises(TypeError, x.getObject, 2.2323)
self.assertEqual(x.getString(0), b'abc')
self.assertEqual(x.getString(1), b'def')
self.assertEqual(x.getString(2), None)
self.assertRaises(TypeError, x.getString, b'')
self.assertRaises(TypeError, x.getString, 2.2323)
self.assertEqual(x.getClass(Foo), a)
self.assertEqual(x.getClass(Bar), b)
self.assertEqual(x.getClass(2), None)
self.assertEqual(x.getClassByReference(0), a)
self.assertEqual(x.getClassByReference(1), b)
self.assertEqual(x.getClassByReference(2), None)
self.assertEqual(x.getObject(2), None)
self.assertEqual(x.getString(2), None)
self.assertEqual(x.getClass(2), None)
self.assertEqual(x.getClassByReference(2), None)