Ejemplo n.º 1
0
func TestDecode(t *testing.T) {
	s := shortid.New()
	s.SetWorkerId(0)
	_, worker := s.Decode(s.Generate())
	if worker != 0 {
		t.Errorf("Worker not decoded correctly")
	}

	s.SetWorkerId(1)
	_, worker = s.Decode(s.Generate())
	if worker != 1 {
		t.Errorf("Worker not decoded correctly")
	}

	s.SetWorkerId(15)
	_, worker = s.Decode(s.Generate())
	if worker != 15 {
		t.Errorf("Worker not decoded correctly")
	}
}
Ejemplo n.º 2
0
func TestGenerate(t *testing.T) {
	s := shortid.New()
	s.SetSeed(12345)
	s.SetWorkerId(6)
	i := 5000
	ids := make(map[string]int)

	for i > 0 {
		id := s.Generate()
		t.Logf("Shortened id: %s", id)
		_, exists := ids[id]
		if exists {
			ids[id] = ids[id] + 1
		} else {
			ids[id] = 1
		}

		i = i - 1

		if ids[id] > 1 {
			t.Errorf("Generated non-unique IDs: %s - %d (%d)", id, ids[id], time.Now().Unix())
		}
	}
}