Understanding the genetic make-up of fruit flies could be a key to treating addiction, according to research.