Identity accessors

client/config.go

@@ -35,9 +35,9 @@ type Config struct {
 	DbSuffix string `json:"db_suffix,omitempty"`
 
 	// Inner
-	memoryPassword      string    `json:"memory_password,omitempty"`
-	additionalPasswords []string  `json:"additional_passwords,omitempty"`
-	me                  *Identity `json:"me,omitempty"`
+	memoryPassword      string
+	additionalPasswords []string
+	me                  *Identity
 }
 
 var instance *Config
@@ -82,6 +82,14 @@ func (c *Config) GetMemPass() string {
 	return c.memoryPassword
 }
 
+func (c *Config) GetIdentity() *Identity {
+	return c.me
+}
+
+func (c *Config) SetIdentity(id *Identity) {
+	c.me = id
+}
+
 func (c *Config) SaveIdentity() error {
 	return c.me.Save()
 }

client/storage_test.go

@@ -38,9 +38,35 @@ func TestStoreMessage(t *testing.T) {
 			panic(err)
 		}
 	}
-
 }
 
-func TestGetLastMessages(t *testing.T) {
-
+func TestManyStoreMessage(t *testing.T) {
+	id := createId()
+	for i := 1; i < 100; i++ {
+		var um meowlib.UserMessage
+		um.Data = []byte(randomLenString(20, 200))
+		err := StoreMessage(&id.Peers[0], &um, GetConfig().memoryPassword)
+		if err != nil {
+			log.Fatal(err)
+		}
+	}
+	messages, err := GetLastMessages(&id.Peers[0], 0, 0, 10, GetConfig().memoryPassword)
+	if err != nil {
+		log.Fatal(err)
+	}
+	// Checks
+	assert.Equal(t, len(messages), 10, "not 10 message")
+	// Cleanup
+	if exists("test.id") {
+		os.Remove("test.id")
+	}
+	files, err := filepath.Glob("*.sqlite")
+	if err != nil {
+		panic(err)
+	}
+	for _, f := range files {
+		if err := os.Remove(f); err != nil {
+			panic(err)
+		}
+	}
 }