Пример #1
0
  @Test
  @SuppressWarnings("unchecked")
  public void testCustomEncoders() throws IOException {
    // If clearEncodingHooks isn't working the first test will fail.
    Transformer tf = new TestDateTransformer();
    BSON.addEncodingHook(TestDate.class, tf);
    BSON.clearEncodingHooks();
    TestDate td = new TestDate(2009, 01, 23, 10, 53, 42);
    BSONObject o = new BasicBSONObject("date", td);
    BSONEncoder e = new BasicBSONEncoder();
    BSONDecoder d = new BasicBSONDecoder();
    BSONCallback cb = new BasicBSONCallback();
    OutputBuffer buf = new BasicOutputBuffer();
    e.set(buf);
    boolean encodeFailed = false;
    try {
      e.putObject(o);
    } catch (IllegalArgumentException ieE) {
      encodeFailed = true;
    }
    assertTrue(encodeFailed);
    // Reset the buffer
    buf.seekStart();
    assertTrue(tf.transform(td) instanceof java.util.Date);

    BSON.addEncodingHook(TestDate.class, tf);
    e.putObject(o);
    e.done();

    d.decode(new ByteArrayInputStream(buf.toByteArray()), cb);
    Object result = cb.get();
    assertTrue(result instanceof BSONObject);
    BSONObject bson = (BSONObject) result;
    assertNotNull(bson.get("date"));
    assertTrue(bson.get("date") instanceof java.util.Date);

    // Check that the hooks registered
    assertNotNull(BSON.getEncodingHooks(TestDate.class));
    Vector expect = new Vector(1);
    expect.add(tf);
    assertEquals(BSON.getEncodingHooks(TestDate.class), expect);
    assertTrue(BSON.getEncodingHooks(TestDate.class).contains(tf));
    BSON.removeEncodingHook(TestDate.class, tf);
    assertFalse(BSON.getEncodingHooks(TestDate.class).contains(tf));
  }
Пример #2
0
  @Test
  @SuppressWarnings({"deprecation", "unchecked"})
  public void testCustomDecoders() throws IOException {
    // If clearDecodingHooks isn't working this whole test will fail.
    Transformer tf = new TestDateTransformer();
    BSON.addDecodingHook(Date.class, tf);
    BSON.clearDecodingHooks();
    TestDate td = new TestDate(2009, 01, 23, 10, 53, 42);
    Date dt = new Date(2009, 01, 23, 10, 53, 42);
    BSONObject o = new BasicBSONObject("date", dt);
    BSONDecoder d = new BasicBSONDecoder();
    BSONEncoder e = new BasicBSONEncoder();
    BSONCallback cb = new BasicBSONCallback();
    OutputBuffer buf = new BasicOutputBuffer();
    e.set(buf);
    e.putObject(o);
    e.done();

    d.decode(new ByteArrayInputStream(buf.toByteArray()), cb);
    Object result = cb.get();
    assertTrue(result instanceof BSONObject);
    BSONObject bson = (BSONObject) result;
    assertNotNull(bson.get("date"));
    assertTrue(bson.get("date") instanceof java.util.Date);

    BSON.addDecodingHook(Date.class, tf);

    d.decode(new ByteArrayInputStream(buf.toByteArray()), cb);
    bson = (BSONObject) cb.get();
    assertNotNull(bson.get("date"));
    assertTrue(bson.get("date") instanceof TestDate);
    assertEquals(bson.get("date"), td);

    // Check that the hooks registered
    assertNotNull(BSON.getDecodingHooks(Date.class));
    Vector expect = new Vector(1);
    expect.add(tf);
    assertEquals(BSON.getDecodingHooks(Date.class), expect);
    assertTrue(BSON.getDecodingHooks(Date.class).contains(tf));
    BSON.removeDecodingHook(Date.class, tf);
    assertFalse(BSON.getDecodingHooks(Date.class).contains(tf));
  }
Пример #3
0
  void _test(BSONObject o, int size, String hash) throws IOException {
    BSONEncoder e = new BasicBSONEncoder();
    OutputBuffer buf = new BasicOutputBuffer();
    e.set(buf);
    e.putObject(o);
    assertEquals(size, buf.size());
    assertEquals(hash, buf.md5());
    e.done();

    BSONDecoder d = new BasicBSONDecoder();
    BSONCallback cb = new BasicBSONCallback();
    int s = d.decode(new ByteArrayInputStream(buf.toByteArray()), cb);
    assertEquals(size, s);

    OutputBuffer buf2 = new BasicOutputBuffer();
    e.set(buf2);
    e.putObject((BSONObject) cb.get());
    assertEquals(size, buf2.size());
    assertEquals(hash, buf2.md5());
  }