mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-28 17:50:56 +01:00
test: updated test cases
This commit is contained in:
@@ -1,89 +1,10 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"github.com/emirpasic/gods/sets/hashset"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
"reflect"
|
||||
)
|
||||
|
||||
func BsonMEqual(v1, v2 bson.M) (ok bool) {
|
||||
//ok = reflect.DeepEqual(v1, v2)
|
||||
ok = bsonMEqual(v1, v2)
|
||||
return ok
|
||||
}
|
||||
|
||||
func bsonMEqual(v1, v2 bson.M) (ok bool) {
|
||||
// all keys
|
||||
allKeys := hashset.New()
|
||||
for key := range v1 {
|
||||
allKeys.Add(key)
|
||||
}
|
||||
for key := range v2 {
|
||||
allKeys.Add(key)
|
||||
}
|
||||
|
||||
for _, keyRes := range allKeys.Values() {
|
||||
key := keyRes.(string)
|
||||
v1Value, ok := v1[key]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
v2Value, ok := v2[key]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
mode := 0
|
||||
|
||||
var v1ValueBsonM bson.M
|
||||
var v1ValueBsonA bson.A
|
||||
switch v1Value.(type) {
|
||||
case bson.M:
|
||||
mode = 1
|
||||
v1ValueBsonM = v1Value.(bson.M)
|
||||
case bson.A:
|
||||
mode = 2
|
||||
v1ValueBsonA = v1Value.(bson.A)
|
||||
}
|
||||
|
||||
var v2ValueBsonM bson.M
|
||||
var v2ValueBsonA bson.A
|
||||
switch v2Value.(type) {
|
||||
case bson.M:
|
||||
if mode != 1 {
|
||||
return false
|
||||
}
|
||||
v2ValueBsonM = v2Value.(bson.M)
|
||||
case bson.A:
|
||||
if mode != 2 {
|
||||
return false
|
||||
}
|
||||
v2ValueBsonA = v2Value.(bson.A)
|
||||
}
|
||||
|
||||
switch mode {
|
||||
case 0:
|
||||
if v1Value != v2Value {
|
||||
return false
|
||||
}
|
||||
case 1:
|
||||
if !bsonMEqual(v1ValueBsonM, v2ValueBsonM) {
|
||||
return false
|
||||
}
|
||||
case 2:
|
||||
if !reflect.DeepEqual(v1ValueBsonA, v2ValueBsonA) {
|
||||
return false
|
||||
}
|
||||
default:
|
||||
// not reachable
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func NormalizeBsonMObjectId(m bson.M) (res bson.M) {
|
||||
for k, v := range m {
|
||||
switch v.(type) {
|
||||
@@ -99,18 +20,6 @@ func NormalizeBsonMObjectId(m bson.M) (res bson.M) {
|
||||
return m
|
||||
}
|
||||
|
||||
func DenormalizeBsonMObjectId(m bson.M) (res bson.M) {
|
||||
for k, v := range m {
|
||||
switch v.(type) {
|
||||
case primitive.ObjectID:
|
||||
m[k] = v.(primitive.ObjectID).Hex()
|
||||
case bson.M:
|
||||
m[k] = NormalizeBsonMObjectId(v.(bson.M))
|
||||
}
|
||||
}
|
||||
return m
|
||||
}
|
||||
|
||||
func NormalizeObjectId(v interface{}) (res interface{}) {
|
||||
switch v.(type) {
|
||||
case string:
|
||||
|
||||
Reference in New Issue
Block a user