func TestDeleteOperation(t *testing.T) {
op := bsonToOperation(t, &bson.M{
"ts": bson.MongoTimestamp(5982836443431567364),
"h": int64(2778471229732698240),
"v": 2,
"op": "d",
"ns": "test_api.dashboards",
"b": true,
"o": map[string]interface{}{
"_id": bson.ObjectIdHex("52e7e160f4eb2740dda12844"),
},
})
if op.Op != Delete {
t.Error("Expected operation to be", Delete)
}
if id, ok := op.Object["_id"]; !ok {
t.Error("Expected _id to be set")
} else {
if h := id.(bson.ObjectId).Hex(); h != "52e7e160f4eb2740dda12844" {
t.Error("Invalid object id", h)
}
}
if ns := op.Namespace; ns != "test_api.dashboards" {
t.Error("Invalid namespace", ns)
}
if ts, valid := *op.Timestamp.Time(), time.Date(2014, time.February, 21, 13, 01, 00, 0, time.UTC); ts != valid {
t.Error("Invalid time", ts, "Expected", valid)
}
}
func TestEsOperation(t *testing.T) {
// Update with full object
op := bsonToOperation(t, &bson.M{
// 2014-01-28 16:23:1q3 +0000 UTC
"ts": bson.MongoTimestamp(5973982510084788956),
"h": int64(-9122761770815979503),
"v": 2,
"op": "u",
"ns": "test.conversations",
"o": map[string]interface{}{
"_cls": "Conversation",
"_id": bson.ObjectIdHex("50eadae392cd864e50cd0dbc"),
"created_at": time.Date(2013, time.January, 07, 14, 25, 39, 941e6, time.UTC),
// ... snip snip ...
},
})
esOp := getEsOp(op)
if a, _ := esOp.Action(); a != "update" {
t.Error("Expected update not", a)
}
if doc, err := esOp.Document(); err != nil {
t.Error(err)
} else {
tr := NewBsonTraverser(doc)
if v, ok := tr.Next("_cls").Value().(string); !ok || v != "Conversation" {
t.Error("Unexpected value", v)
}
}
}
func TestUpdateOperationSet(t *testing.T) {
op := bsonToOperation(t, &bson.M{
"ts": bson.MongoTimestamp(5973982510084789348),
"h": int64(-7293803671238204358),
"v": 2,
"op": "u",
"ns": "test_api.users",
"o2": map[string]interface{}{
"_id": bson.ObjectIdHex("52e7db73f4eb27371874b289"),
},
"o": bson.M{
"$set": map[string]interface{}{
"photo_tally": map[string]interface{}{
"total": 1,
},
},
},
})
if op.Op != Update {
t.Error("Expected operation to be", Update)
}
if id, err := op.ObjectId(); err != nil {
t.Error(err)
} else {
if id.Hex() != "52e7db73f4eb27371874b289" {
t.Error("Invalid Id", id)
}
}
if ns := op.Namespace; ns != "test_api.users" {
t.Error("Invalid namespace", ns)
}
if ts, valid := *op.Timestamp.Time(), time.Date(2014, time.January, 28, 16, 23, 13, 0, time.UTC); ts != valid {
t.Error("Invalid time", ts, "Expected", valid)
}
if _, ok := op.Object["$set"]; !ok {
t.Error("Expected $set to be available")
}
if c, err := (&EsOperation{Operation: op}).Document(); err != nil {
t.Error(err)
} else {
if c["photo_tally"].(bson.M)["total"].(int) != 1 {
t.Error("Incorrect changeset:", c)
}
}
}
func TestInsertOperation(t *testing.T) {
op := bsonToOperation(t, &bson.M{
// 2014-01-28 16:23:1q3 +0000 UTC
"ts": bson.MongoTimestamp(5973982510084788956),
"h": int64(-9122761770815979503),
"v": 2,
"op": "i",
"ns": "api.conversations",
"o": map[string]interface{}{
"_cls": "Conversation",
"_id": bson.ObjectIdHex("50eadae392cd864e50cd0dbc"),
"created_at": time.Date(2013, time.January, 07, 14, 25, 39, 941e6, time.UTC),
// ... snip snip ...
},
})
if op.Op != Insert {
t.Error("Expected operation to be", Insert)
}
if id, err := op.ObjectId(); err != nil {
t.Error(err)
} else {
if id.Hex() != "50eadae392cd864e50cd0dbc" {
t.Error("Invalid Id", id)
}
}
if ns := op.Namespace; ns != "api.conversations" {
t.Error("Invalid namespace", ns)
}
if createdAt, ok := op.Object["created_at"]; !ok {
t.Error("Expected to find created_at field")
} else {
if ok := createdAt.(time.Time).Equal(time.Date(2013, time.January, 7, 14, 25, 39, 941e6, time.UTC)); !ok {
t.Error("Invalid created at time", createdAt)
}
}
if c, err := (&EsOperation{Operation: op}).Document(); err != nil {
t.Error(err)
} else {
if c["_cls"].(string) != "Conversation" {
t.Error("Incorrect changeset:", c)
}
}
}
func TestEsOperationUnsetSet(t *testing.T) {
op := bsonToOperation(t, &bson.M{
// 2014-01-28 16:23:1q3 +0000 UTC
"ts": bson.MongoTimestamp(5973982510084788956),
"h": int64(-9122761770815979503),
"v": 2,
"op": "u",
"ns": "test.conversations",
"o": map[string]interface{}{
"$set": map[string]interface{}{
"alias": "Hello",
},
"$unset": map[string]interface{}{
"foo": 1,
},
},
"o2": map[string]interface{}{
"_id": bson.ObjectIdHex("50eadae392cd864e50cd0dbc"),
},
})
esOp := getEsOp(op)
if a, _ := esOp.Action(); a != "update" {
t.Error("Unexpected action", a)
}
d, err := esOp.Document()
if err != nil {
t.Fatal(err)
}
if len(d) != 2 {
t.Error("Incorrect number of fields in:", d)
}
// Since we got an $unset we will treat that one as a $set with null value
if v, ok := d["foo"]; !ok || v != nil {
t.Error("Expected foo to be nil")
}
}
{bson.M{"_": time.Unix(0, 258e6)}, // Note the NS <=> MS conversion.
"\x09_\x00\x02\x01\x00\x00\x00\x00\x00\x00"},
{bson.M{"_": nil},
"\x0A_\x00"},
{bson.M{"_": bson.RegEx{"ab", "cd"}},
"\x0B_\x00ab\x00cd\x00"},
{bson.M{"_": bson.JavaScript{"code", nil}},
"\x0D_\x00\x05\x00\x00\x00code\x00"},
{bson.M{"_": bson.Symbol("sym")},
"\x0E_\x00\x04\x00\x00\x00sym\x00"},
{bson.M{"_": bson.JavaScript{"code", bson.M{"": nil}}},
"\x0F_\x00\x14\x00\x00\x00\x05\x00\x00\x00code\x00" +
"\x07\x00\x00\x00\x0A\x00\x00"},
{bson.M{"_": 258},
"\x10_\x00\x02\x01\x00\x00"},
{bson.M{"_": bson.MongoTimestamp(258)},
"\x11_\x00\x02\x01\x00\x00\x00\x00\x00\x00"},
{bson.M{"_": int64(258)},
"\x12_\x00\x02\x01\x00\x00\x00\x00\x00\x00"},
{bson.M{"_": int64(258 << 32)},
"\x12_\x00\x00\x00\x00\x00\x02\x01\x00\x00"},
{bson.M{"_": bson.MaxKey},
"\x7F_\x00"},
{bson.M{"_": bson.MinKey},
"\xFF_\x00"},
}
func (s *S) TestMarshalAllItems(c *C) {
for i, item := range allItems {
data, err := bson.Marshal(item.obj)
c.Assert(err, IsNil)
// GetBSON helps bson marshal understand that we're really a MongoTimestamp
func (t Timestamp) GetBSON() (interface{}, error) {
return bson.MongoTimestamp(t), nil
}