This commit is contained in:
Ammar Aijazi 2015-06-30 00:11:44 -07:00
Родитель 24975d342e
Коммит 051c12bfbf
1 изменённых файлов: 15 добавлений и 1 удалений

Просмотреть файл

@ -165,7 +165,7 @@ func TestQueryShard(t *testing.T) {
func TestQueryResult(t *testing.T) {
// We can't do the reflection test because bson
// doesn't do it correctly for embedded fields.
want := "\x90\x01\x00\x00\x03Result\x00\x99\x00\x00\x00\x04Fields\x009\x00\x00\x00\x030\x001\x00\x00\x00\x05Name\x00\x04\x00\x00\x00\x00name\x12Type\x00\x01\x00\x00\x00\x00\x00\x00\x00\x12Flags\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00?RowsAffected\x00\x02\x00\x00\x00\x00\x00\x00\x00?InsertId\x00\x03\x00\x00\x00\x00\x00\x00\x00\x04Rows\x00 \x00\x00\x00\x040\x00\x18\x00\x00\x00\x050\x00\x01\x00\x00\x00\x001\x051\x00\x02\x00\x00\x00\x00aa\x00\x00\nErr\x00\x00\x03Session\x00\xd0\x00\x00\x00\bInTransaction\x00\x01\x04ShardSessions\x00\xac\x00\x00\x00\x030\x00Q\x00\x00\x00\x05Keyspace\x00\x01\x00\x00\x00\x00a\x05Shard\x00\x01\x00\x00\x00\x000\x05TabletType\x00\a\x00\x00\x00\x00replica\x12TransactionId\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x031\x00P\x00\x00\x00\x05Keyspace\x00\x01\x00\x00\x00\x00b\x05Shard\x00\x01\x00\x00\x00\x001\x05TabletType\x00\x06\x00\x00\x00\x00master\x12TransactionId\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x05Error\x00\x05\x00\x00\x00\x00error\x00"
want := "\xc7\x01\x00\x00\x03Result\x00\x99\x00\x00\x00\x04Fields\x009\x00\x00\x00\x030\x001\x00\x00\x00\x05Name\x00\x04\x00\x00\x00\x00name\x12Type\x00\x01\x00\x00\x00\x00\x00\x00\x00\x12Flags\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00?RowsAffected\x00\x02\x00\x00\x00\x00\x00\x00\x00?InsertId\x00\x03\x00\x00\x00\x00\x00\x00\x00\x04Rows\x00 \x00\x00\x00\x040\x00\x18\x00\x00\x00\x050\x00\x01\x00\x00\x00\x001\x051\x00\x02\x00\x00\x00\x00aa\x00\x00\nErr\x00\x00\x03Session\x00\xd0\x00\x00\x00\bInTransaction\x00\x01\x04ShardSessions\x00\xac\x00\x00\x00\x030\x00Q\x00\x00\x00\x05Keyspace\x00\x01\x00\x00\x00\x00a\x05Shard\x00\x01\x00\x00\x00\x000\x05TabletType\x00\a\x00\x00\x00\x00replica\x12TransactionId\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x031\x00P\x00\x00\x00\x05Keyspace\x00\x01\x00\x00\x00\x00b\x05Shard\x00\x01\x00\x00\x00\x001\x05TabletType\x00\x06\x00\x00\x00\x00master\x12TransactionId\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x05Error\x00\x05\x00\x00\x00\x00error\x03Err\x002\x00\x00\x00\x12Code\x00\xd0\a\x00\x00\x00\x00\x00\x00\x05Message\x00\x11\x00\x00\x00\x00failed due to err\x00\x00"
custom := QueryResult{
Result: &mproto.QueryResult{
@ -178,6 +178,10 @@ func TestQueryResult(t *testing.T) {
},
Session: &commonSession,
Error: "error",
Err: &mproto.RPCError{
Code: 2000,
Message: "failed due to err",
},
}
encoded, err := bson.Marshal(&custom)
if err != nil {
@ -312,6 +316,7 @@ type reflectQueryResultList struct {
List []mproto.QueryResult
Session *Session
Error string
Err *mproto.RPCError
}
type extraQueryResultList struct {
@ -319,6 +324,7 @@ type extraQueryResultList struct {
List []mproto.QueryResult
Session *Session
Error string
Err *mproto.RPCError
}
func TestQueryResultList(t *testing.T) {
@ -333,6 +339,10 @@ func TestQueryResultList(t *testing.T) {
}},
Session: &commonSession,
Error: "error",
Err: &mproto.RPCError{
Code: 2000,
Message: "failed due to err",
},
})
if err != nil {
t.Error(err)
@ -350,6 +360,10 @@ func TestQueryResultList(t *testing.T) {
}},
Session: &commonSession,
Error: "error",
Err: &mproto.RPCError{
Code: 2000,
Message: "failed due to err",
},
}
encoded, err := bson.Marshal(&custom)
if err != nil {