Fixing parsing of device Ids (#5580)

model/push_notification_test.go

@@ -17,3 +17,78 @@ func TestPushNotification(t *testing.T) {
 		t.Fatal("Ids do not match")
 	}
 }
+
+func TestPushNotificationDeviceId(t *testing.T) {
+
+	msg := PushNotification{Platform: "test"}
+
+	msg.SetDeviceIdAndPlatform("android:12345")
+	if msg.Platform != "android" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != "12345" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+
+	msg.SetDeviceIdAndPlatform("android:12:345")
+	if msg.Platform != "android" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != "12:345" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+
+	msg.SetDeviceIdAndPlatform("android::12345")
+	if msg.Platform != "android" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != ":12345" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+
+	msg.SetDeviceIdAndPlatform(":12345")
+	if msg.Platform != "" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != "12345" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+
+	msg.SetDeviceIdAndPlatform("android:")
+	if msg.Platform != "android" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != "" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+
+	msg.SetDeviceIdAndPlatform("")
+	if msg.Platform != "" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != "" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+
+	msg.SetDeviceIdAndPlatform(":")
+	if msg.Platform != "" {
+		t.Fatal(msg.Platform)
+	}
+	if msg.DeviceId != "" {
+		t.Fatal(msg.DeviceId)
+	}
+	msg.Platform = ""
+	msg.DeviceId = ""
+}